def test_get_download_dir_exception(self, transmission_client): rpc_client = transmission_client.return_value rpc_client.get_session.side_effect = transmissionrpc.TransmissionError plugin = TransmissionClientPlugin() settings = { 'host': 'localhost', 'username': '******', 'password': '******' } plugin.set_settings(settings) with pytest.raises(transmissionrpc.TransmissionError) as e: plugin.get_download_dir() rpc_client.get_session.assert_called_once()
def test_get_download_dir_success(self, transmission_client): rpc_client = transmission_client.return_value rpc_client.get_session.return_value = transmissionrpc.Session( fields={'download_dir': '/mnt/media/downloads'}) plugin = TransmissionClientPlugin() assert plugin.get_download_dir() is None settings = { 'host': 'localhost', 'username': '******', 'password': '******' } plugin.set_settings(settings) assert plugin.get_download_dir() == u'/mnt/media/downloads' rpc_client.get_session.assert_called_once()