diff options
Diffstat (limited to 'testsuite/Testsrc/Testlib')
| -rw-r--r-- | testsuite/Testsrc/Testlib/TestServer/TestPlugins/TestProbes.py | 12 |
1 files changed, 8 insertions, 4 deletions
diff --git a/testsuite/Testsrc/Testlib/TestServer/TestPlugins/TestProbes.py b/testsuite/Testsrc/Testlib/TestServer/TestPlugins/TestProbes.py index 38d3c08e6..ac6f6b8e7 100644 --- a/testsuite/Testsrc/Testlib/TestServer/TestPlugins/TestProbes.py +++ b/testsuite/Testsrc/Testlib/TestServer/TestPlugins/TestProbes.py @@ -502,6 +502,8 @@ text for cname, cdata in self.get_test_probedata().items(): client = Mock() client.hostname = cname + cgroups = [] + cprobedata = ClientProbeDataSet() for pname, pdata in cdata.items(): dataitem = lxml.etree.Element("Probe", name=pname) if pname == "text": @@ -514,11 +516,13 @@ text else: dataitem.text = str(pdata) - probes.ReceiveDataItem(client, dataitem) + probes.ReceiveDataItem(client, dataitem, cgroups, cprobedata) - self.assertIn(client.hostname, probes.probedata) - self.assertIn(pname, probes.probedata[cname]) - self.assertEqual(pdata, probes.probedata[cname][pname]) + probes.cgroups[client.hostname] = cgroups + probes.probedata[client.hostname] = cprobedata + self.assertIn(client.hostname, probes.probedata) + self.assertIn(pname, probes.probedata[cname]) + self.assertEqual(pdata, probes.probedata[cname][pname]) self.assertIn(client.hostname, probes.cgroups) self.assertEqual(probes.cgroups[cname], self.get_test_cgroups()[cname]) |