def _test_get_metric_history_as_columns(self, slugs, granularity):
"""Test that R.get_metric_history_as_columns makes calls to the
following functions:
* ``R.r.mget``
* ``R.get_metric_history``
* ``templatetags.metric_slug``
* ``templatetags.strip_metric_prefix``
"""
keys = self._metric_history_keys(slugs, granularity=granularity)
self.r.get_metric_history_as_columns(slugs, granularity=granularity)
# Verifies the correct call to redis
self.redis.assert_has_calls([call.mget(keys)])
# Verify that the method gets called correctly
with patch('redis_metrics.models.R') as mock_r:
r = mock_r.return_value # Get an instance of our Mocked R class
r.get_metric_history_as_columns(slugs, granularity=granularity)
mock_r.assert_has_calls([
call().get_metric_history_as_columns(
slugs, granularity=granularity
)
])
def _test_get_metric_history(self, slugs, granularity):
"""actual test code for ``R.get_metric_history``."""
keys = self._metric_history_keys(slugs, granularity=granularity)
self.r.get_metric_history(slugs, granularity=granularity)
self.redis.assert_has_calls([call.mget(keys)])
