mirror of
https://pagure.io/fm-orchestrator.git
synced 2026-02-06 06:43:18 +08:00
f24cd4222f
Please note that this patch does not change the use of database session in MBS. So, in the frontend, the database session is still managed by Flask-SQLAlchemy, that is the db.session. And the backend, running event handlers, has its own database session created from SQLAclehmy session API directly. This patch aims to reduce the number of scoped_session created when call original function make_db_session. For technical detailed information, please refer to SQLAlchemy documentation Contextual/Thread-local Sessions. As a result, a global scoped_session is accessible from the code running inside backend, both the event handlers and functions called from handlers. The library code shared by frontend and backend, like resolvers, has no change. Similarly, db.session is only used to recreate database for every test. Signed-off-by: Chenxiong Qi <cqi@redhat.com>
92 lines
2.9 KiB
Python
92 lines
2.9 KiB
Python
# -*- coding: utf-8 -*-
|
|
# SPDX-License-Identifier: MIT
|
|
import os
|
|
import pytest
|
|
import requests
|
|
import mock
|
|
import module_build_service.config as mbs_config
|
|
import module_build_service.monitor
|
|
from module_build_service import models
|
|
from module_build_service.db_session import db_session
|
|
from conf.config import TestConfiguration
|
|
|
|
from six.moves import reload_module
|
|
from tests import app, init_data, make_module_in_db
|
|
|
|
num_of_metrics = 18
|
|
|
|
|
|
class TestViews:
|
|
def setup_method(self, test_method):
|
|
self.client = app.test_client()
|
|
init_data(2)
|
|
|
|
def test_metrics(self):
|
|
rv = self.client.get("/module-build-service/1/monitor/metrics")
|
|
|
|
count = len([
|
|
l for l in rv.get_data(as_text=True).splitlines()
|
|
if (l.startswith("# TYPE") and "_created " not in l)
|
|
])
|
|
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():
|
|
os.environ["MONITOR_STANDALONE_METRICS_SERVER_ENABLE"] = "true"
|
|
reload_module(module_build_service.monitor)
|
|
|
|
r = requests.get("http://127.0.0.1:10040/metrics")
|
|
count = len([
|
|
l for l in r.text.splitlines()
|
|
if (l.startswith("# TYPE") and "_created " not in l)
|
|
])
|
|
assert count == num_of_metrics
|
|
|
|
|
|
@mock.patch("module_build_service.monitor.builder_failed_counter.labels")
|
|
@mock.patch("module_build_service.monitor.builder_success_counter.inc")
|
|
def test_monitor_state_changing_success(succ_cnt, failed_cnt):
|
|
conf = mbs_config.Config(TestConfiguration)
|
|
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.monitor.builder_failed_counter.labels")
|
|
@mock.patch("module_build_service.monitor.builder_success_counter.inc")
|
|
def test_monitor_state_changing_failure(succ_cnt, failed_cnt):
|
|
failure_type = "user"
|
|
conf = mbs_config.Config(TestConfiguration)
|
|
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)
|