def testHostSampleReportsNewInterface(self): hs_before = sampling.HostSample(os.getpid()) interfaces_before = set(hs_before.interfaces.iterkeys()) with dummy_if() as dummy_name: hs_after = sampling.HostSample(os.getpid()) interfaces_after = set(hs_after.interfaces.iterkeys()) interfaces_diff = interfaces_after - interfaces_before self.assertEqual(interfaces_diff, set([dummy_name]))
def WrapHostSample(pid): self._sampleCount += 1 if self._sampleCount == self.FAILED_SAMPLE: raise ValueError if self._sampleCount == self.STOP_SAMPLE: self._hs.stop() self._samplingDone.set() return sampling.HostSample(1)
def testHostSampleHandlesDisappearingVlanInterfaces(self): original_getLinks = ipwrapper.getLinks def faultyGetLinks(): all_links = list(original_getLinks()) ipwrapper.linkDel(self.NEW_VLAN) return iter(all_links) with MonkeyPatchScope([(ipwrapper, 'getLinks', faultyGetLinks)]): with dummy_if() as dummy_name: with vlan(self.NEW_VLAN, dummy_name, 999): hs = sampling.HostSample(os.getpid()) self.assertNotIn(self.NEW_VLAN, hs.interfaces)