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)])