Example #1
0
    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()