class ArchiveClientTest(unittest.TestCase): def setUp(self): self.result_service = MagicMock() self.report_type = "TEST_REPORT_TYPE" self.scope = "TEST_SCOPE" self.result_dict = { "Success": True } self.result_service.get_deserialized_result.return_value = self.result_dict self.client = ArchiveClient(self.result_service) def test_get_archive(self): args = (ARCHIVE_BASE_URI, self.scope) params = { "type": self.report_type } result = self.client.get_archive(self.scope, self.report_type) self.assertDictEqual(result, self.result_dict) self.result_service.get_deserialized_result.assert_called_once_with(args, params) def test_get_archive_with_no_scope(self): args = (ARCHIVE_BASE_URI,) params = { "type": self.report_type } result = self.client.get_archive(None, self.report_type) self.assertDictEqual(result, self.result_dict) self.result_service.get_deserialized_result.assert_called_once_with(args, params)
def setUp(self): self.result_service = MagicMock() self.report_type = "TEST_REPORT_TYPE" self.scope = "TEST_SCOPE" self.result_dict = { "Success": True } self.result_service.get_deserialized_result.return_value = self.result_dict self.client = ArchiveClient(self.result_service)