def test_ga_query_hook_is_called(self):
service = Mock()
ga_hook = Mock()
client = QueryClient(service, ga_hook)
self.mock_get('short-query', service)
client.get('12345', date(2012, 1, 1), date(2012, 1, 2), 'metric')
kwargs = {
'ids': 'ga:12345',
'metrics': 'ga:metric',
'start_date': '2012-01-01',
'end_date': '2012-01-02',
}
ga_hook.assert_called_once_with(kwargs)
def setUp(self):
self.service = Mock()
self.client = QueryClient(self.service)
