def test_empty_file_list(self, resolwe_mock, os_mock): resolwe_mock.configure_mock(**self.config) 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.isdir.return_value = False message = "Download directory does not exist: .*" with self.assertRaisesRegex(ValueError, message): Resolwe._download_files(resolwe_mock, self.file_list)
def test_fail_if_bad_dir(self, resolwe_mock, os_mock): resolwe_mock.configure_mock(**self.config) os_mock.path.isdir.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, os_mock): resolwe_mock.configure_mock(**self.config) os_mock.path.isfile.return_value = True response = {"raise_for_status.side_effect": Exception("abc")} resolwe_mock.session.get.return_value = MagicMock(ok=False, **response) with self.assertRaisesRegex(Exception, "abc"): Resolwe._download_files(resolwe_mock, self.file_list[:1]) self.assertEqual(resolwe_mock.logger.info.call_count, 2)
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 self.assertRaisesRegex(Exception, "abc"): Resolwe._download_files(resolwe_mock, self.file_list[:1]) self.assertEqual(resolwe_mock.logger.info.call_count, 2)
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)
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)
def test_good_response(self, resolwe_mock, os_mock, open_mock): resolwe_mock.configure_mock(**self.config) os_mock.path.isfile.return_value = True resolwe_mock.session.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 )
def test_empty_file_list(self, resolwe_mock, os_mock): resolwe_mock.configure_mock(**self.config) Resolwe._download_files(resolwe_mock, []) resolwe_mock.logger.info.assert_called_once_with("No files to download.")