mirror of
https://pagure.io/fm-orchestrator.git
synced 2026-02-13 18:15:00 +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>
80 lines
2.8 KiB
Python
80 lines
2.8 KiB
Python
# -*- coding: utf-8 -*-
|
|
# SPDX-License-Identifier: MIT
|
|
from mock import patch, MagicMock
|
|
from module_build_service.scheduler.consumer import MBSConsumer
|
|
from module_build_service.messaging import KojiTagChange, KojiRepoChange
|
|
|
|
|
|
class TestConsumer:
|
|
def test_get_abstracted_msg_fedmsg(self):
|
|
"""
|
|
Test the output of get_abstracted_msg() when using the
|
|
fedmsg backend.
|
|
"""
|
|
hub = MagicMock(config={})
|
|
consumer = MBSConsumer(hub)
|
|
msg = {
|
|
"username": "apache",
|
|
"source_name": "datanommer",
|
|
"i": 1,
|
|
"timestamp": 1505492681.0,
|
|
"msg_id": "2017-0627b798-f241-4230-b365-8a8a111a8ec5",
|
|
"crypto": "x509",
|
|
"topic": "org.fedoraproject.prod.buildsys.tag",
|
|
"headers": {},
|
|
"source_version": "0.8.1",
|
|
"msg": {
|
|
"build_id": 962861,
|
|
"name": "python3-virtualenv",
|
|
"tag_id": 263,
|
|
"instance": "primary",
|
|
"tag": "epel7-pending",
|
|
"user": "bodhi",
|
|
"version": "15.1.0",
|
|
"owner": "orion",
|
|
"release": "1.el7",
|
|
},
|
|
}
|
|
msg_obj = consumer.get_abstracted_msg(msg)
|
|
assert isinstance(msg_obj, KojiTagChange)
|
|
assert msg_obj.msg_id == msg["msg_id"]
|
|
assert msg_obj.tag == msg["msg"]["tag"]
|
|
assert msg_obj.artifact == msg["msg"]["name"]
|
|
|
|
@patch("module_build_service.scheduler.consumer.models")
|
|
@patch.object(MBSConsumer, "process_message")
|
|
def test_consume_fedmsg(self, process_message, models):
|
|
"""
|
|
Test the MBSConsumer.consume() method when using the
|
|
fedmsg backend.
|
|
"""
|
|
hub = MagicMock(config={})
|
|
consumer = MBSConsumer(hub)
|
|
msg = {
|
|
"topic": "org.fedoraproject.prod.buildsys.repo.done",
|
|
"headers": {},
|
|
"body": {
|
|
"username": "apache",
|
|
"source_name": "datanommer",
|
|
"i": 1,
|
|
"timestamp": 1405126329.0,
|
|
"msg_id": "2014-adbc33f6-51b0-4fce-aa0d-3c699a9920e4",
|
|
"crypto": "x509",
|
|
"topic": "org.fedoraproject.prod.buildsys.repo.done",
|
|
"headers": {},
|
|
"source_version": "0.6.4",
|
|
"msg": {
|
|
"instance": "primary",
|
|
"repo_id": 400859,
|
|
"tag": "f22-build",
|
|
"tag_id": 278,
|
|
},
|
|
},
|
|
}
|
|
consumer.consume(msg)
|
|
assert process_message.call_count == 1
|
|
msg_obj = process_message.call_args[0][0]
|
|
assert isinstance(msg_obj, KojiRepoChange)
|
|
assert msg_obj.msg_id == msg["body"]["msg_id"]
|
|
assert msg_obj.repo_tag == msg["body"]["msg"]["tag"]
|