mirror of
https://pagure.io/fm-orchestrator.git
synced 2026-02-03 05:03:43 +08:00
c584d84b76
Replace clean_database calls with cheaper truncate operation. Remove duplicate calls of clean_database. Extract clean_database/init_data into fixtures.
88 lines
2.9 KiB
Python
88 lines
2.9 KiB
Python
# -*- coding: utf-8 -*-
|
|
# SPDX-License-Identifier: MIT
|
|
from __future__ import absolute_import
|
|
import os
|
|
|
|
import mock
|
|
import pytest
|
|
import requests
|
|
from six.moves import reload_module
|
|
|
|
from module_build_service.common import conf, models
|
|
import module_build_service.common.monitor
|
|
from module_build_service.scheduler.db_session import db_session
|
|
from tests import make_module_in_db
|
|
|
|
num_of_metrics = 18
|
|
|
|
|
|
@pytest.mark.usefixtures("provide_test_client")
|
|
class TestViews:
|
|
|
|
def test_metrics(self, provide_test_data):
|
|
rv = self.client.get("/module-build-service/1/monitor/metrics")
|
|
|
|
count = len([
|
|
line for line in rv.get_data(as_text=True).splitlines()
|
|
if (line.startswith("# TYPE") and "_created " not in line)
|
|
])
|
|
assert count == num_of_metrics
|
|
|
|
|
|
def test_standalone_metrics_server_disabled_by_default():
|
|
with pytest.raises(requests.exceptions.ConnectionError):
|
|
requests.get("http://127.0.0.1:10040/metrics")
|
|
|
|
|
|
def test_standalone_metrics_server():
|
|
with mock.patch.dict(os.environ, {"MONITOR_STANDALONE_METRICS_SERVER_ENABLE": "true"}):
|
|
reload_module(module_build_service.common.monitor)
|
|
|
|
r = requests.get("http://127.0.0.1:10040/metrics")
|
|
count = len([
|
|
line for line in r.text.splitlines()
|
|
if (line.startswith("# TYPE") and "_created " not in line)
|
|
])
|
|
assert count == num_of_metrics
|
|
|
|
|
|
@mock.patch("module_build_service.common.monitor.builder_failed_counter.labels")
|
|
@mock.patch("module_build_service.common.monitor.builder_success_counter.inc")
|
|
def test_monitor_state_changing_success(succ_cnt, failed_cnt, require_empty_database):
|
|
b = make_module_in_db(
|
|
"pkg:0.1:1:c1",
|
|
[
|
|
{
|
|
"requires": {"platform": ["el8"]},
|
|
"buildrequires": {"platform": ["el8"]},
|
|
}
|
|
],
|
|
)
|
|
b.transition(db_session, conf, models.BUILD_STATES["wait"])
|
|
b.transition(db_session, conf, models.BUILD_STATES["build"])
|
|
b.transition(db_session, conf, models.BUILD_STATES["done"])
|
|
db_session.commit()
|
|
succ_cnt.assert_called_once()
|
|
failed_cnt.assert_not_called()
|
|
|
|
|
|
@mock.patch("module_build_service.common.monitor.builder_failed_counter.labels")
|
|
@mock.patch("module_build_service.common.monitor.builder_success_counter.inc")
|
|
def test_monitor_state_changing_failure(succ_cnt, failed_cnt, require_empty_database):
|
|
failure_type = "user"
|
|
b = make_module_in_db(
|
|
"pkg:0.1:1:c1",
|
|
[
|
|
{
|
|
"requires": {"platform": ["el8"]},
|
|
"buildrequires": {"platform": ["el8"]},
|
|
}
|
|
],
|
|
)
|
|
b.transition(db_session, conf, models.BUILD_STATES["wait"])
|
|
b.transition(db_session, conf, models.BUILD_STATES["build"])
|
|
b.transition(db_session, conf, models.BUILD_STATES["failed"], failure_type=failure_type)
|
|
db_session.commit()
|
|
succ_cnt.assert_not_called()
|
|
failed_cnt.assert_called_once_with(reason=failure_type)
|