diff --git a/tests/test_graphql/test_api_monitoring.py b/tests/test_graphql/test_api_monitoring.py index f1fdd46..965af3e 100644 --- a/tests/test_graphql/test_api_monitoring.py +++ b/tests/test_graphql/test_api_monitoring.py @@ -2,252 +2,252 @@ # pylint: disable=unused-argument # pylint: disable=missing-function-docstring -from dataclasses import dataclass -from datetime import datetime -from typing import List, Dict -import pytest +# from dataclasses import dataclass +# from datetime import datetime +# from typing import List, Dict +# import pytest -from tests.test_graphql.common import ( - assert_empty, - get_data, -) +# from tests.test_graphql.common import ( +# assert_empty, +# get_data, +# ) -MOCK_VALUES = [ - [1720135748, "3.75"], - [1720135808, "4.525000000139698"], - [1720135868, "4.541666666433841"], - [1720135928, "4.574999999798209"], - [1720135988, "4.579166666759804"], - [1720136048, "3.8791666664959195"], - [1720136108, "4.5458333333954215"], - [1720136168, "4.566666666651145"], - [1720136228, "4.791666666666671"], - [1720136288, "4.720833333364382"], - [1720136348, "3.9624999999068677"], - [1720136408, "4.6875"], - [1720136468, "4.404166666790843"], - [1720136528, "4.31666666680637"], - [1720136588, "4.358333333317816"], - [1720136648, "3.7083333334885538"], - [1720136708, "4.558333333116025"], - [1720136768, "4.729166666511446"], - [1720136828, "4.75416666672875"], - [1720136888, "4.624999999844775"], - [1720136948, "3.9041666667132375"], -] +# MOCK_VALUES = [ +# [1720135748, "3.75"], +# [1720135808, "4.525000000139698"], +# [1720135868, "4.541666666433841"], +# [1720135928, "4.574999999798209"], +# [1720135988, "4.579166666759804"], +# [1720136048, "3.8791666664959195"], +# [1720136108, "4.5458333333954215"], +# [1720136168, "4.566666666651145"], +# [1720136228, "4.791666666666671"], +# [1720136288, "4.720833333364382"], +# [1720136348, "3.9624999999068677"], +# [1720136408, "4.6875"], +# [1720136468, "4.404166666790843"], +# [1720136528, "4.31666666680637"], +# [1720136588, "4.358333333317816"], +# [1720136648, "3.7083333334885538"], +# [1720136708, "4.558333333116025"], +# [1720136768, "4.729166666511446"], +# [1720136828, "4.75416666672875"], +# [1720136888, "4.624999999844775"], +# [1720136948, "3.9041666667132375"], +# ] -@dataclass -class DumbResponse: - status_code: int - json_data: dict +# @dataclass +# class DumbResponse: +# status_code: int +# json_data: dict - def json(self): - return self.json_data +# def json(self): +# return self.json_data -def generate_prometheus_response(result_type: str, result: List[Dict]): - return DumbResponse( - status_code=200, - json_data={"data": {"resultType": result_type, "result": result}}, - ) +# def generate_prometheus_response(result_type: str, result: List[Dict]): +# return DumbResponse( +# status_code=200, +# json_data={"data": {"resultType": result_type, "result": result}}, +# ) -MOCK_SINGLE_METRIC_PROMETHEUS_RESPONSE = generate_prometheus_response( - "matrix", [{"values": MOCK_VALUES}] -) -MOCK_MULTIPLE_METRIC_DEVICE_PROMETHEUS_RESPONSE = generate_prometheus_response( - "matrix", - [ - {"metric": {"device": "a"}, "values": MOCK_VALUES}, - {"metric": {"device": "b"}, "values": MOCK_VALUES}, - {"metric": {"device": "c"}, "values": MOCK_VALUES}, - ], -) +# MOCK_SINGLE_METRIC_PROMETHEUS_RESPONSE = generate_prometheus_response( +# "matrix", [{"values": MOCK_VALUES}] +# ) +# MOCK_MULTIPLE_METRIC_DEVICE_PROMETHEUS_RESPONSE = generate_prometheus_response( +# "matrix", +# [ +# {"metric": {"device": "a"}, "values": MOCK_VALUES}, +# {"metric": {"device": "b"}, "values": MOCK_VALUES}, +# {"metric": {"device": "c"}, "values": MOCK_VALUES}, +# ], +# ) -# def generate_mock_metrics(name: str): -# return { -# "data": { -# "monitoring": { -# f"{name}": { -# "resultType": "matrix", -# "result": [ -# { -# "metric": {"instance": "127.0.0.1:9002"}, -# "values": , -# } -# ], -# } -# } -# } -# } +# # def generate_mock_metrics(name: str): +# # return { +# # "data": { +# # "monitoring": { +# # f"{name}": { +# # "resultType": "matrix", +# # "result": [ +# # { +# # "metric": {"instance": "127.0.0.1:9002"}, +# # "values": , +# # } +# # ], +# # } +# # } +# # } +# # } -# MOCK_CPU_USAGE_RESPONSE = generate_mock_metrics("cpuUsage") -# MOCK_DISK_USAGE_RESPONSE = generate_mock_metrics("diskUsage") -# MOCK_MEMORY_USAGE_RESPONSE = generate_mock_metrics("memoryUsage") +# # MOCK_CPU_USAGE_RESPONSE = generate_mock_metrics("cpuUsage") +# # MOCK_DISK_USAGE_RESPONSE = generate_mock_metrics("diskUsage") +# # MOCK_MEMORY_USAGE_RESPONSE = generate_mock_metrics("memoryUsage") -def generate_mock_query(name): - return f""" - query Query {{ - monitoring {{ - {name} {{ resultType, result }} - }} - }} - """ +# def generate_mock_query(name): +# return f""" +# query Query {{ +# monitoring {{ +# {name} {{ resultType, result }} +# }} +# }} +# """ -def generate_mock_query_with_options(name): - return f""" - query Query($start: DateTime, $end: DateTime, $step: Int) {{ - monitoring {{ - {name}(start: $start, end: $end, step: $step) {{ resultType, result }} - }} - }} - """ +# def generate_mock_query_with_options(name): +# return f""" +# query Query($start: DateTime, $end: DateTime, $step: Int) {{ +# monitoring {{ +# {name}(start: $start, end: $end, step: $step) {{ resultType, result }} +# }} +# }} +# """ -def prometheus_result_from_dict(dict): - # return MonitoringQueryResult(result_type=dict["resultType"], result=dict["result"]) - return dict +# def prometheus_result_from_dict(dict): +# # return MonitoringQueryResult(result_type=dict["resultType"], result=dict["result"]) +# return dict -@pytest.fixture -def mock_cpu_usage(mocker): - mock = mocker.patch( - "selfprivacy_api.utils.prometheus.PrometheusQueries._send_query", - return_value=MOCK_CPU_USAGE_RESPONSE["data"]["monitoring"]["cpuUsage"], - ) - return mock +# @pytest.fixture +# def mock_cpu_usage(mocker): +# mock = mocker.patch( +# "selfprivacy_api.utils.prometheus.PrometheusQueries._send_query", +# return_value=MOCK_CPU_USAGE_RESPONSE["data"]["monitoring"]["cpuUsage"], +# ) +# return mock -@pytest.fixture -def mock_memory_usage(mocker): - mock = mocker.patch( - "selfprivacy_api.utils.prometheus.PrometheusQueries._send_query", - return_value=prometheus_result_from_dict( - MOCK_MEMORY_USAGE_RESPONSE["data"]["monitoring"]["memoryUsage"] - ), - ) - return mock +# @pytest.fixture +# def mock_memory_usage(mocker): +# mock = mocker.patch( +# "selfprivacy_api.utils.prometheus.PrometheusQueries._send_query", +# return_value=prometheus_result_from_dict( +# MOCK_MEMORY_USAGE_RESPONSE["data"]["monitoring"]["memoryUsage"] +# ), +# ) +# return mock -@pytest.fixture -def mock_disk_usage(mocker): - mock = mocker.patch( - "selfprivacy_api.utils.prometheus.PrometheusQueries._send_query", - return_value=prometheus_result_from_dict( - MOCK_DISK_USAGE_RESPONSE["data"]["monitoring"]["diskUsage"] - ), - ) - return mock +# @pytest.fixture +# def mock_disk_usage(mocker): +# mock = mocker.patch( +# "selfprivacy_api.utils.prometheus.PrometheusQueries._send_query", +# return_value=prometheus_result_from_dict( +# MOCK_DISK_USAGE_RESPONSE["data"]["monitoring"]["diskUsage"] +# ), +# ) +# return mock -def test_graphql_get_disk_usage(client, authorized_client, mock_disk_usage): - response = authorized_client.post( - "/graphql", - json={"query": generate_mock_query("diskUsage")}, - ) +# def test_graphql_get_disk_usage(client, authorized_client, mock_disk_usage): +# response = authorized_client.post( +# "/graphql", +# json={"query": generate_mock_query("diskUsage")}, +# ) - data = get_data(response) - assert data == MOCK_DISK_USAGE_RESPONSE["data"] +# data = get_data(response) +# assert data == MOCK_DISK_USAGE_RESPONSE["data"] -def test_graphql_get_disk_usage_with_options( - client, authorized_client, mock_disk_usage -): - response = authorized_client.post( - "/graphql", - json={ - "query": generate_mock_query_with_options("diskUsage"), - "variables": { - "start": datetime.fromtimestamp(1720136108).isoformat(), - "end": datetime.fromtimestamp(1720137319).isoformat(), - "step": 90, - }, - }, - ) +# def test_graphql_get_disk_usage_with_options( +# client, authorized_client, mock_disk_usage +# ): +# response = authorized_client.post( +# "/graphql", +# json={ +# "query": generate_mock_query_with_options("diskUsage"), +# "variables": { +# "start": datetime.fromtimestamp(1720136108).isoformat(), +# "end": datetime.fromtimestamp(1720137319).isoformat(), +# "step": 90, +# }, +# }, +# ) - data = get_data(response) - assert data == MOCK_DISK_USAGE_RESPONSE["data"] +# data = get_data(response) +# assert data == MOCK_DISK_USAGE_RESPONSE["data"] -def test_graphql_get_disk_usage_unauthorized(client): - response = client.post( - "/graphql", - json={"query": generate_mock_query("diskUsage")}, - ) - assert_empty(response) +# def test_graphql_get_disk_usage_unauthorized(client): +# response = client.post( +# "/graphql", +# json={"query": generate_mock_query("diskUsage")}, +# ) +# assert_empty(response) -def test_graphql_get_memory_usage(client, authorized_client, mock_memory_usage): - response = authorized_client.post( - "/graphql", - json={"query": generate_mock_query("memoryUsage")}, - ) +# def test_graphql_get_memory_usage(client, authorized_client, mock_memory_usage): +# response = authorized_client.post( +# "/graphql", +# json={"query": generate_mock_query("memoryUsage")}, +# ) - data = get_data(response) - assert data == MOCK_MEMORY_USAGE_RESPONSE["data"] +# data = get_data(response) +# assert data == MOCK_MEMORY_USAGE_RESPONSE["data"] -def test_graphql_get_memory_usage_with_options( - client, authorized_client, mock_memory_usage -): - response = authorized_client.post( - "/graphql", - json={ - "query": generate_mock_query_with_options("memoryUsage"), - "variables": { - "start": datetime.fromtimestamp(1720136108).isoformat(), - "end": datetime.fromtimestamp(1720137319).isoformat(), - "step": 90, - }, - }, - ) +# def test_graphql_get_memory_usage_with_options( +# client, authorized_client, mock_memory_usage +# ): +# response = authorized_client.post( +# "/graphql", +# json={ +# "query": generate_mock_query_with_options("memoryUsage"), +# "variables": { +# "start": datetime.fromtimestamp(1720136108).isoformat(), +# "end": datetime.fromtimestamp(1720137319).isoformat(), +# "step": 90, +# }, +# }, +# ) - data = get_data(response) - assert data == MOCK_MEMORY_USAGE_RESPONSE["data"] +# data = get_data(response) +# assert data == MOCK_MEMORY_USAGE_RESPONSE["data"] -def test_graphql_get_memory_usage_unauthorized(client): - response = client.post( - "/graphql", - json={"query": generate_mock_query("memoryUsage")}, - ) - assert_empty(response) +# def test_graphql_get_memory_usage_unauthorized(client): +# response = client.post( +# "/graphql", +# json={"query": generate_mock_query("memoryUsage")}, +# ) +# assert_empty(response) -def test_graphql_get_cpu_usage(client, authorized_client, mock_cpu_usage): - response = authorized_client.post( - "/graphql", - json={"query": generate_mock_query("cpuUsage")}, - ) +# def test_graphql_get_cpu_usage(client, authorized_client, mock_cpu_usage): +# response = authorized_client.post( +# "/graphql", +# json={"query": generate_mock_query("cpuUsage")}, +# ) - data = get_data(response) - assert data == MOCK_CPU_USAGE_RESPONSE["data"] +# data = get_data(response) +# assert data == MOCK_CPU_USAGE_RESPONSE["data"] -def test_graphql_get_cpu_usage_with_options(client, authorized_client, mock_cpu_usage): - response = authorized_client.post( - "/graphql", - json={ - "query": generate_mock_query_with_options("cpuUsage"), - "variables": { - "start": datetime.fromtimestamp(1720136108).isoformat(), - "end": datetime.fromtimestamp(1720137319).isoformat(), - "step": 90, - }, - }, - ) +# def test_graphql_get_cpu_usage_with_options(client, authorized_client, mock_cpu_usage): +# response = authorized_client.post( +# "/graphql", +# json={ +# "query": generate_mock_query_with_options("cpuUsage"), +# "variables": { +# "start": datetime.fromtimestamp(1720136108).isoformat(), +# "end": datetime.fromtimestamp(1720137319).isoformat(), +# "step": 90, +# }, +# }, +# ) - data = get_data(response) - assert data == MOCK_CPU_USAGE_RESPONSE["data"] +# data = get_data(response) +# assert data == MOCK_CPU_USAGE_RESPONSE["data"] -def test_graphql_get_cpu_usage_unauthorized(client): - response = client.post( - "/graphql", - json={"query": generate_mock_query("cpuUsage")}, - ) - assert_empty(response) +# def test_graphql_get_cpu_usage_unauthorized(client): +# response = client.post( +# "/graphql", +# json={"query": generate_mock_query("cpuUsage")}, +# ) +# assert_empty(response)