def testMultiIPConnects(self): """We should correctly count how many times the same IP is connected, even with different classes.""" self._start(1) conntrack.up(2, '127.0.0.1', ('foo',)) self.assertEqual(conntrack.ipcount('127.0.0.1'), 2) conntrack.up(3, '127.0.0.1', ('bar',)) self.assertEqual(conntrack.ipcount('127.0.0.1'), 3) conntrack.down(1) self.assertEqual(conntrack.ipcount('127.0.0.1'), 2) conntrack.down(3) self.assertEqual(conntrack.ipcount('127.0.0.1'), 1)
def doesfail(self, hi, what, cls = None): if what not in self.dirs: return False if not cls: cls = self val = self.dirs[what] if what == "reject": return True elif what == "ipmax": return conntrack.ipcount(hi.getip()) >= val else: return conntrack.classcount(cls.name) >= val
def testInsertRemove(self): """If we insert then remove, the result should be null.""" self._start(1) conntrack.down(1) self.assertEqual(conntrack.ipcount('127.0.0.1'), 0) self.assertEqual(conntrack.classcount('ALL'), 0)
def testInsert(self): """If we insert, we should get expected results.""" self._start(1) self.assertEqual(conntrack.ipcount('127.0.0.1'), 1) self.assertEqual(conntrack.classcount('ALL'), 1)
def testEmptyCount(self): "Test that we can get a zero count for values not in the system" conntrack._clearmaps() self.assertEqual(conntrack.ipcount("127.100.100.0"), 0) self.assertEqual(conntrack.classcount("FOOBAR"), 0)