summaryrefslogtreecommitdiffstats
path: root/tests
diff options
context:
space:
mode:
Diffstat (limited to 'tests')
-rw-r--r--tests/test_discovery.py4
1 files changed, 1 insertions, 3 deletions
diff --git a/tests/test_discovery.py b/tests/test_discovery.py
index 68d3cbd..c3a1329 100644
--- a/tests/test_discovery.py
+++ b/tests/test_discovery.py
@@ -16,8 +16,6 @@
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA
#
-from pynag import Model
-
from plugins import discovery, server_utils
from plugins.config_generator import GLUSTER_AUTO_CONFIG
from plugins.config_generator import HOST_SERVICES
@@ -143,7 +141,7 @@ class TestDiscovery(TestCaseBase):
'10.70.43.0!10.70.43.57'})
def fillServiceModel(self, values):
- serviceModel = Model.Service()
+ serviceModel = {}
for key, value in values.iteritems():
serviceModel[key] = value
return serviceModel