class RanksClietTest(unittest.TestCase):
def setUp(self):
self.result_service = MagicMock()
self.ranks_client = RanksClient(self.result_service)
def test_get_ranks(self):
result_dict = { 'test': 'test' }
expected_params = { 'param_one': ['test1', 1234], 'param_two_two': 'test2' }
products = [1235, 44444, 1111]
granularity = "Bi-quarterly"
startdate = date(2012, 04, 1)
enddate = date(2012, 05, 30)
expected_args = (RANKS_BASE_URI, products, granularity, startdate, enddate)
self.result_service.get_deserialized_result.return_value = result_dict
result = self.ranks_client.get_ranks(products, granularity, 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.result_service = MagicMock()
self.ranks_client = RanksClient(self.result_service)
