diff options
| -rw-r--r-- | testsuite/Testsrc/Testlib/TestServer/TestPlugin/Testinterfaces.py | 10 |
1 files changed, 6 insertions, 4 deletions
diff --git a/testsuite/Testsrc/Testlib/TestServer/TestPlugin/Testinterfaces.py b/testsuite/Testsrc/Testlib/TestServer/TestPlugin/Testinterfaces.py index ace509057..9a064663e 100644 --- a/testsuite/Testsrc/Testlib/TestServer/TestPlugin/Testinterfaces.py +++ b/testsuite/Testsrc/Testlib/TestServer/TestPlugin/Testinterfaces.py @@ -326,11 +326,13 @@ class TestGoalValidator(Bcfg2TestCase): gv.validate_goals, None, None) -class TestVersion(Bcfg2TestCase): +class TestVersion(TestPlugin): test_obj = Version - - def get_obj(self): - return self.test_obj(datastore) + + def get_obj(self, core=None): + if core is None: + core = Mock() + return self.test_obj(core, datastore) def test_get_revision(self): d = self.get_obj() |