class TestSalesApi(unittest.TestCase): def setUp(self): self.base_url = "http://example.com" self.result_service = MagicMock() self.sales_client = SalesClient(self.result_service) def test_report_by_country_and_product(self): result_dict = { 'test': 'test' } expected_params = { 'paramOne': 'test2', 'paramTwoTwo': 'test2' } startdate = datetime.date(2011, 10, 1) enddate = datetime.date(2012, 4, 22) expected_args = (SALES_BASE_URI, REPORT_BY_COUNTRY_AND_PRODUCT, startdate, enddate) self.result_service.get_deserialized_result.return_value = result_dict result = self.sales_client.get_sales_report(REPORT_BY_COUNTRY_AND_PRODUCT, startdate, enddate, **expected_params) self.assertDictEqual(result, result_dict) self.result_service.get_deserialized_result.assert_called_once_with(expected_args, expected_params)
def setUp(self): self.base_url = "http://example.com" self.result_service = MagicMock() self.sales_client = SalesClient(self.result_service)