def test_query(self):
        with patch.object(KairosDBRestClient, '_make_request') as mock_method:
            client = KairosDBRestClient()
            client.query(100, 200, metrics=[Metric('metric.name')])

        mock_method.assert_called_once_with(
            QueryMetricDataPointsRequest(100, 200, [Metric('metric.name')],
                                         None, 0))
    def test_metric_tags(self):
        with patch.object(KairosDBRestClient, '_make_request') as mock_method:
            client = KairosDBRestClient()
            client.metric_tags(100, 200,
                               [MetricTag('metric.1', host=['azure'])])

        mock_method.assert_called_once_with(
            QueryMetricTagsRequest(100, 200,
                                   [MetricTag('metric.1', host=['azure'])]))
    def test_tag_values(self):
        with patch.object(KairosDBRestClient, '_make_request') as mock_method:
            client = KairosDBRestClient()
            client.tag_values()

        mock_method.assert_called_once_with(QueryTagValuesRequest())
    def test_metric_names(self):
        with patch.object(KairosDBRestClient, '_make_request') as mock_method:
            client = KairosDBRestClient()
            client.metric_names()

        mock_method.assert_called_once_with(QueryMetricNamesRequest())
    def test_version(self):
        with patch.object(KairosDBRestClient, '_make_request') as mock_method:
            client = KairosDBRestClient()
            client.version()

        mock_method.assert_called_once_with(VersionRequest())
 def setUp(self):
     self.client = KairosDBRestClient()
    def test_health_status(self):
        with patch.object(KairosDBRestClient, '_make_request') as mock_method:
            client = KairosDBRestClient()
            client.health_status()

        mock_method.assert_called_once_with(HealthStatusRequest())
    def test_delete_metric(self):
        with patch.object(KairosDBRestClient, '_make_request') as mock_method:
            client = KairosDBRestClient()
            client.delete_metric('metric.name')

        mock_method.assert_called_once_with(DeleteMetricRequest('metric.name'))