# coding: utf-8

from __future__ import absolute_import

import unittest
import requests
import testutils

class TestMetricsExist(unittest.TestCase):
    golang_basic_metrics = ["go_gc_duration_seconds", "go_goroutines", "go_info", "go_memstats_alloc_bytes"]

    eigen_metrics = {
        'core': golang_basic_metrics + ["harbor_core_http_request", "harbor_core_http_request_duration_seconds",
    "harbor_core_http_request_inflight"],
        'registry': golang_basic_metrics + ["registry_http_in_flight_requests"],
        'exporter': golang_basic_metrics + ["artifact_pulled",
    "harbor_project_artifact_total", "harbor_project_member_total", "harbor_project_quota_byte",
    "harbor_project_repo_total", "harbor_project_total"
    ]
    }

    def get_metrics(self):
        metrics_url = testutils.METRIC_URL+'/metrics'
        exporter_res = requests.get(metrics_url)
        core_res = requests.get(metrics_url, params={'comp': 'core'})
        reg_res = requests.get(metrics_url, params={'comp': 'registry'})
        return [('exporter', exporter_res.text), ('core', core_res.text), ('registry', reg_res.text)]

    def testMetricsExist(self):
        for k, metric_text in self.get_metrics():
            for metric_name in self.eigen_metrics[k]:
                print("Metric {} should exist in {} ".format(metric_name, k))
                self.assertTrue(metric_name in metric_text)

if __name__ == '__main__':
    unittest.main()