def test_get_https_downloader(self, to_nectar, http): url = 'https://martin.com/test' config = Mock() # test downloader = Importer.get_downloader(config, url) # validation to_nectar.assert_called_once_with(config.flatten.return_value) http.assert_called_once_with(to_nectar.return_value) self.assertEqual(downloader, http.return_value)
def test_get_local_downloader(self, to_nectar, local): url = 'file:///martin.com/test' config = Mock() # test downloader = Importer.get_downloader(config, url) # validation to_nectar.assert_called_once_with(config.flatten.return_value, working_dir=None) local.assert_called_once_with(to_nectar.return_value) self.assertEqual(downloader, local.return_value)