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
