From f4c53348943914accdeb2d3ab21a8079b9d42b04 Mon Sep 17 00:00:00 2001 From: Chenxiong Qi Date: Wed, 17 Jul 2019 13:53:04 +0800 Subject: [PATCH] Reset FakeModuleBuilder.on_get_task_info_cb properly for TestBuild Signed-off-by: Chenxiong Qi --- tests/test_build/test_build.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/tests/test_build/test_build.py b/tests/test_build/test_build.py index 3f42fdc0..3aa69055 100644 --- a/tests/test_build/test_build.py +++ b/tests/test_build/test_build.py @@ -134,6 +134,7 @@ class FakeModuleBuilder(GenericBuilder): FakeModuleBuilder.on_buildroot_add_artifacts_cb = None FakeModuleBuilder.on_tag_artifacts_cb = None FakeModuleBuilder.on_buildroot_add_repos_cb = None + FakeModuleBuilder.on_get_task_info_cb = None FakeModuleBuilder.DEFAULT_GROUPS = None FakeModuleBuilder.backend = "test" @@ -412,6 +413,11 @@ class TestBuild(BaseTestBuild): self.client = app.test_client() clean_database() + def on_get_task_info_cb(cls, task_id): + return {"state": koji.TASK_STATES["CLOSED"]} + + FakeModuleBuilder.on_get_task_info_cb = on_get_task_info_cb + def teardown_method(self, test_method): FakeModuleBuilder.reset() cleanup_moksha() @@ -462,10 +468,6 @@ class TestBuild(BaseTestBuild): def on_tag_artifacts_cb(cls, artifacts, dest_tag=True): assert tag_groups.pop(0) == set(artifacts) - def on_get_task_info_cb(cls, task_id): - return {"state": koji.TASK_STATES["CLOSED"]} - - FakeModuleBuilder.on_get_task_info_cb = on_get_task_info_cb FakeModuleBuilder.on_finalize_cb = on_finalize_cb FakeModuleBuilder.on_tag_artifacts_cb = on_tag_artifacts_cb