class DataClientTest(unittest.TestCase): def setUp(self): self.result_service = MagicMock() self.data_type = "TEST_DATA_TYPE" self.result_dict = {"Success": True} self.result_service.get_deserialized_result.return_value = self.result_dict self.client = DataClient(self.result_service) def test_get_data(self): expected_args = (DATA_BASE_URI, self.data_type) result = self.client.get_data(self.data_type) self.assertDictEqual(result, self.result_dict) self.result_service.get_deserialized_result.assert_called_once_with(expected_args, None) def test_get_data_with_other_args(self): store_name = "android" expected_args = (DATA_BASE_URI, self.data_type, store_name) result = self.client.get_data(self.data_type, store_name) self.assertDictEqual(result, self.result_dict) self.result_service.get_deserialized_result.assert_called_once_with(expected_args, None)
def setUp(self): self.result_service = MagicMock() self.data_type = "TEST_DATA_TYPE" self.result_dict = {"Success": True} self.result_service.get_deserialized_result.return_value = self.result_dict self.client = DataClient(self.result_service)