def test_empty_file_list(self, resolwe_mock, os_mock):
resolwe_mock.configure_mock(**self.config)
os_mock.path.isfile.return_value = True
Resolwe.download_files(resolwe_mock, [])
resolwe_mock.logger.info.assert_called_once_with("No files to download.")
def test_fail_if_bad_dir(self, resolwe_mock, os_mock):
resolwe_mock.configure_mock(**self.config)
os_mock.path.isfile.return_value = False
message = "Download directory does not exist: .*"
with six.assertRaisesRegex(self, ValueError, message):
Resolwe.download_files(resolwe_mock, self.file_list)
def test_bad_response(self, resolwe_mock, requests_mock, os_mock, open_mock):
resolwe_mock.configure_mock(**self.config)
os_mock.path.isfile.return_value = True
mock_open.return_value = MagicMock(spec=io.IOBase)
response = {'raise_for_status.side_effect': Exception("abc")}
requests_mock.get.return_value = MagicMock(ok=False, **response)
with six.assertRaisesRegex(self, Exception, "abc"):
Resolwe.download_files(resolwe_mock, self.file_list[:1])
self.assertEqual(resolwe_mock.logger.info.call_count, 2)
def test_good_response(self, resolwe_mock, requests_mock, os_mock, open_mock):
resolwe_mock.configure_mock(**self.config)
os_mock.path.isfile.return_value = True
# When mocking open one wants it to return a "file-like" mock: (spec=io.IOBase)
mock_open.return_value = MagicMock(spec=io.IOBase)
requests_mock.get.return_value = MagicMock(ok=True, **{'iter_content.return_value': range(3)})
Resolwe.download_files(resolwe_mock, self.file_list)
self.assertEqual(resolwe_mock.logger.info.call_count, 3)
# This asserts may seem wierd. To check what is happening behind the scenes:
# print(open_mock.mock_calls)
self.assertEqual(open_mock.return_value.__enter__.return_value.write.call_count, 6)
