def testClassCount(self): """The active class list should be maintained correctly.""" self._start(1) self.assertEqual(conntrack.activeclasses(), ['ALL',]) conntrack.up(2, '127.0.0.1', ('ALL', 'foo')) self.assertEqual(self._getclasses(), ['ALL', 'foo']) self._up(3) self.assertEqual(self._getclasses(), ['ALL', 'foo'])
def testManyClasses(self): """We should correctly add to each class for multi-class connections.""" conntrack._clearmaps() conntrack.up(1, '127.0.0.1', ('foo', 'bar')) self.assertEqual(conntrack.classcount('foo'), 1) self.assertEqual(conntrack.classcount('bar'), 1) conntrack.down(1) self.assertEqual(conntrack.classcount('foo'), 0) self.assertEqual(conntrack.classcount('bar'), 0) self.assertEqual(len(conntrack.activeclasses()), 0)
def testNoClassesLeft(self): """There should be no active classes left listed after up/down.""" self._start(1) conntrack.down(1) self.assertEqual(len(conntrack.activeclasses()), 0)
def _getclasses(self): c = conntrack.activeclasses() c.sort() return c