Fix: test after hg fix

shinken-solutions · Jan 8, 2015 · db2547e · db2547e
1 parent b366eee
commit db2547e
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/test/test_inheritance_and_plus.py b/test/test_inheritance_and_plus.py
@@ -49,15 +49,15 @@ def test_inheritance_and_plus(self):
         # HOST 1 is lin-servers,dmz, so should be in linux AND DMZ group
         for hg in host1.hostgroups:
             print hg.get_name()
-        self.assertIn(linux, host1.hostgroups)
-        self.assertIn(dmz, host1.hostgroups)
+        self.assertIn(linux.get_name(), [hg.get_name() for hg in host1.hostgroups])
+        self.assertIn(dmz.get_name(), [hg.get_name() for hg in host1.hostgroups])
 
         # HOST2 is in lin-servers,dmz and +mysql, so all three of them
         for hg in host2.hostgroups:
             print hg.get_name()
-        self.assertIn(linux, host2.hostgroups)
-        self.assertIn(dmz, host2.hostgroups)
-        self.assertIn(mysql, host2.hostgroups)
+        self.assertIn(linux.get_name(), [hg.get_name() for hg in host2.hostgroups])
+        self.assertIn(dmz.get_name(), [hg.get_name() for hg in host2.hostgroups])
+        self.assertIn(mysql.get_name(), [hg.get_name() for hg in host2.hostgroups])