diff --git a/tests/test_hub/test_add_archivetype.py b/tests/test_hub/test_add_archivetype.py index db78c4da..14342246 100644 --- a/tests/test_hub/test_add_archivetype.py +++ b/tests/test_hub/test_add_archivetype.py @@ -24,7 +24,7 @@ class TestAddArchiveType(unittest.TestCase): self.insert_execute = mock.MagicMock() self.verify_name_internal = mock.patch('kojihub.kojihub.verify_name_internal').start() self.get_archive_type = mock.patch('kojihub.kojihub.get_archive_type').start() - self.QueryProcessor = mock.patch('kojihub.QueryProcessor', + self.QueryProcessor = mock.patch('kojihub.kojihub.QueryProcessor', side_effect=self.getQuery).start() self.queries = [] self.query_execute = mock.MagicMock() diff --git a/tests/test_hub/test_create_build_target.py b/tests/test_hub/test_create_build_target.py index dd0c399a..14d8e9ce 100644 --- a/tests/test_hub/test_create_build_target.py +++ b/tests/test_hub/test_create_build_target.py @@ -10,10 +10,10 @@ import kojihub class TestCreateBuildTarget(unittest.TestCase): def setUp(self): - self.get_build_targets = mock.patch('kojihub.get_build_targets').start() - self.get_tag = mock.patch('kojihub.get_tag').start() - self.verify_name_internal = mock.patch('kojihub.verify_name_internal').start() - self.context = mock.patch('kojihub.context').start() + self.get_build_targets = mock.patch('kojihub.kojihub.get_build_targets').start() + self.get_tag = mock.patch('kojihub.kojihub.get_tag').start() + self.verify_name_internal = mock.patch('kojihub.kojihub.verify_name_internal').start() + self.context = mock.patch('kojihub.kojihub.context').start() # It seems MagicMock will not automatically handle attributes that # start with "assert" self.context.session.assertPerm = mock.MagicMock() diff --git a/tests/test_hub/test_create_image_build.py b/tests/test_hub/test_create_image_build.py index 3ccfcd7b..2b881e13 100644 --- a/tests/test_hub/test_create_image_build.py +++ b/tests/test_hub/test_create_image_build.py @@ -11,9 +11,9 @@ IP = kojihub.InsertProcessor class TestCreateImageBuild(unittest.TestCase): def setUp(self): - self.get_build = mock.patch('kojihub.get_build').start() + self.get_build = mock.patch('kojihub.kojihub.get_build').start() self.exports = kojihub.RootExports() - self.InsertProcessor = mock.patch('kojihub.InsertProcessor', + self.InsertProcessor = mock.patch('kojihub.kojihub.InsertProcessor', side_effect=self.getInsert).start() self.inserts = [] self.insert_execute = mock.MagicMock() diff --git a/tests/test_hub/test_create_maven_build.py b/tests/test_hub/test_create_maven_build.py index aa7dcf93..9ed1afeb 100644 --- a/tests/test_hub/test_create_maven_build.py +++ b/tests/test_hub/test_create_maven_build.py @@ -11,13 +11,13 @@ IP = kojihub.InsertProcessor class TestCreateMavenBuild(unittest.TestCase): def setUp(self): - self.get_build = mock.patch('kojihub.get_build').start() + self.get_build = mock.patch('kojihub.kojihub.get_build').start() self.exports = kojihub.RootExports() self.session = mock.MagicMock() - self.context = mock.patch('kojihub.context').start() + self.context = mock.patch('kojihub.kojihub.context').start() self.context_db = mock.patch('koji.db.context').start() self.context.session.assertPerm = mock.MagicMock() - self.InsertProcessor = mock.patch('kojihub.InsertProcessor', + self.InsertProcessor = mock.patch('kojihub.kojihub.InsertProcessor', side_effect=self.getInsert).start() self.inserts = [] self.insert_execute = mock.MagicMock()