diff options
Diffstat (limited to 'testsuite/Testlib/TestServer/TestPlugins/TestProbes.py')
-rw-r--r-- | testsuite/Testlib/TestServer/TestPlugins/TestProbes.py | 7 |
1 files changed, 4 insertions, 3 deletions
diff --git a/testsuite/Testlib/TestServer/TestPlugins/TestProbes.py b/testsuite/Testlib/TestServer/TestPlugins/TestProbes.py index df9248e50..38161215d 100644 --- a/testsuite/Testlib/TestServer/TestPlugins/TestProbes.py +++ b/testsuite/Testlib/TestServer/TestPlugins/TestProbes.py @@ -1,7 +1,6 @@ import os import sys import time -import unittest import lxml.etree from mock import Mock, MagicMock, patch from ....common import * @@ -228,9 +227,11 @@ text # test__init(), which relies on being able to check the calls # of load_data(), and thus on load_data() being consistently # mocked) - with patch("Bcfg2.Server.Plugins.Probes.Probes.load_data", - new=load_data): + @patch("Bcfg2.Server.Plugins.Probes.Probes.load_data", new=load_data) + def inner(): return Probes(core, datastore) + + return inner() def test__init(self): mock_load_data = Mock() |