Register handlers as Celery tasks

This commit is contained in:
Qixiang Wan
2019-11-18 16:54:18 +08:00
committed by mprahl
parent d6c5f8c13b
commit e07b82e2f9
9 changed files with 52 additions and 8 deletions

View File

@@ -389,8 +389,18 @@ class TestBuild(BaseTestBuild):
return_value=True)
self.mock_check_gating = self.p_check_gating.start()
self.patch_config_broker = patch.object(
module_build_service.config.Config,
"celery_broker_url",
create=True,
new_callable=PropertyMock,
return_value=False,
)
self.patch_config_broker.start()
def teardown_method(self, test_method):
self.p_check_gating.stop()
self.patch_config_broker.stop()
FakeModuleBuilder.reset()
cleanup_moksha()
for i in range(20):

View File

@@ -2,9 +2,10 @@
# SPDX-License-Identifier: MIT
import pytest
from mock import call, patch, Mock
from mock import call, patch, PropertyMock, Mock
from sqlalchemy import func
import module_build_service.config
from module_build_service import conf
from module_build_service.db_session import db_session
from module_build_service.models import BUILD_STATES, ModuleBuild
@@ -72,6 +73,19 @@ class TestGetCorrespondingModuleBuild:
class TestDecisionUpdateHandler:
"""Test handler decision_update"""
def setup_method(self, test_method):
self.patch_config_broker = patch.object(
module_build_service.config.Config,
"celery_broker_url",
create=True,
new_callable=PropertyMock,
return_value=False,
)
self.patch_config_broker.start()
def teardown_method(self, test_method):
self.patch_config_broker.stop()
@patch("module_build_service.scheduler.handlers.greenwave.log")
def test_decision_context_is_not_match(self, log):
decision_update(