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)