def test_get_additional_groups(self): TestConnector.test_get_additional_groups(self) probes = self.get_probes_object() test_cgroups = self.get_test_cgroups() probes.cgroups = self.get_test_cgroups() for cname in test_cgroups.keys(): metadata = Mock() metadata.hostname = cname self.assertEqual(test_cgroups[cname], probes.get_additional_groups(metadata)) # test a non-existent client metadata = Mock() metadata.hostname = "nonexistent" self.assertEqual(probes.get_additional_groups(metadata), list())