Python FailManager示例

示例#1

	def setUp(self):
		"""Call before every test case."""
		self.__items = [['193.168.0.128', 1167605999.0],
					    ['193.168.0.128', 1167605999.0],
					    ['193.168.0.128', 1167605999.0],
					    ['193.168.0.128', 1167605999.0],
					    ['193.168.0.128', 1167605999.0],
					    ['87.142.124.10', 1167605999.0],
					    ['87.142.124.10', 1167605999.0],
					    ['87.142.124.10', 1167605999.0]]
		
		self.__failManager = FailManager()
		for i in self.__items:
			self.__failManager.addFailure(FailTicket(i[0], i[1]))

示例#2

	def setUp(self):
		"""Call before every test case."""
		self.__items = [['193.168.0.128', 1167605999.0],
					    ['193.168.0.128', 1167605999.0],
					    ['193.168.0.128', 1167605999.0],
					    ['193.168.0.128', 1167605999.0],
					    ['193.168.0.128', 1167605999.0],
					    ['87.142.124.10', 1167605999.0],
					    ['87.142.124.10', 1167605999.0],
					    ['87.142.124.10', 1167605999.0]]
		
		self.__failManager = FailManager()
		for i in self.__items:
			self.__failManager.addFailure(FailTicket(i[0], i[1]))

示例#3

文件： failmanagertestcase.py 项目： JohnyByk/fail2ban

    def setUp(self):
        """Call before every test case."""
        self.__items = [
            ["193.168.0.128", 1167605999.0],
            ["193.168.0.128", 1167605999.0],
            ["193.168.0.128", 1167605999.0],
            ["193.168.0.128", 1167605999.0],
            ["193.168.0.128", 1167605999.0],
            ["87.142.124.10", 1167605999.0],
            ["87.142.124.10", 1167605999.0],
            ["87.142.124.10", 1167605999.0],
            ["100.100.10.10", 1000000000.0],
            ["100.100.10.10", 1000000500.0],
            ["100.100.10.10", 1000001000.0],
            ["100.100.10.10", 1000001500.0],
            ["100.100.10.10", 1000002000.0],
        ]

        self.__failManager = FailManager()
        for i in self.__items:
            self.__failManager.addFailure(FailTicket(i[0], i[1]))

示例#4

class AddFailure(unittest.TestCase):
	def setUp(self):
		"""Call before every test case."""
		self.__items = [['193.168.0.128', 1167605999.0],
					    ['193.168.0.128', 1167605999.0],
					    ['193.168.0.128', 1167605999.0],
					    ['193.168.0.128', 1167605999.0],
					    ['193.168.0.128', 1167605999.0],
					    ['87.142.124.10', 1167605999.0],
					    ['87.142.124.10', 1167605999.0],
					    ['87.142.124.10', 1167605999.0],
					    ['100.100.10.10', 1000000000.0],
					    ['100.100.10.10', 1000000500.0],
					    ['100.100.10.10', 1000001000.0],
					    ['100.100.10.10', 1000001500.0],
					    ['100.100.10.10', 1000002000.0]]
		
		self.__failManager = FailManager()
		for i in self.__items:
			self.__failManager.addFailure(FailTicket(i[0], i[1]))
	def tearDown(self):
		"""Call after every test case."""
	
	def testAdd(self):
		self.assertEqual(self.__failManager.size(), 3)
	
	def _testDel(self):
		self.__failManager.delFailure('193.168.0.128')
		self.__failManager.delFailure('111.111.1.111')
		
		self.assertEqual(self.__failManager.size(), 1)
		
	def testCleanupOK(self):
		timestamp = 1167606999.0
		self.__failManager.cleanup(timestamp)
		self.assertEqual(self.__failManager.size(), 0)
		
	def testCleanupNOK(self):
		timestamp = 1167605990.0
		self.__failManager.cleanup(timestamp)
		self.assertEqual(self.__failManager.size(), 2)
	
	def testbanOK(self):
		self.__failManager.setMaxRetry(5)
		
		ticket = self.__failManager.toBan()
		self.assertEqual(ticket.getIP(), "193.168.0.128")
	
	def testbanNOK(self):
		self.__failManager.setMaxRetry(10)
		self.assertRaises(FailManagerEmpty, self.__failManager.toBan)
	def testWindow(self):
		ticket = self.__failManager.toBan()
		self.assertNotEqual(ticket.getIP(), "100.100.10.10")
		ticket = self.__failManager.toBan()
		self.assertNotEqual(ticket.getIP(), "100.100.10.10")
		self.assertRaises(FailManagerEmpty, self.__failManager.toBan)

示例#5

文件： failmanagertestcase.py 项目： Jamyn/fail2ban

class AddFailure(unittest.TestCase):

	def setUp(self):
		"""Call before every test case."""
		self.__items = [[u'193.168.0.128', 1167605999.0],
					    [u'193.168.0.128', 1167605999.0],
					    [u'193.168.0.128', 1167605999.0],
					    [u'193.168.0.128', 1167605999.0],
					    [u'193.168.0.128', 1167605999.0],
					    ['87.142.124.10', 1167605999.0],
					    ['87.142.124.10', 1167605999.0],
					    ['87.142.124.10', 1167605999.0],
					    ['100.100.10.10', 1000000000.0],
					    ['100.100.10.10', 1000000500.0],
					    ['100.100.10.10', 1000001000.0],
					    ['100.100.10.10', 1000001500.0],
					    ['100.100.10.10', 1000002000.0]]
		
		self.__failManager = FailManager()
		for i in self.__items:
			self.__failManager.addFailure(FailTicket(i[0], i[1]))

	def tearDown(self):
		"""Call after every test case."""
	
	def testAdd(self):
		self.assertEqual(self.__failManager.size(), 3)
	
	def _testDel(self):
		self.__failManager.delFailure('193.168.0.128')
		self.__failManager.delFailure('111.111.1.111')
		
		self.assertEqual(self.__failManager.size(), 1)
		
	def testCleanupOK(self):
		timestamp = 1167606999.0
		self.__failManager.cleanup(timestamp)
		self.assertEqual(self.__failManager.size(), 0)
		
	def testCleanupNOK(self):
		timestamp = 1167605990.0
		self.__failManager.cleanup(timestamp)
		self.assertEqual(self.__failManager.size(), 2)
	
	def testbanOK(self):
		self.__failManager.setMaxRetry(5)
		#ticket = FailTicket('193.168.0.128', None)
		ticket = self.__failManager.toBan()
		self.assertEqual(ticket.getIP(), "193.168.0.128")
		self.assertTrue(isinstance(ticket.getIP(), str))

		# finish with rudimentary tests of the ticket
		# verify consistent str
		ticket_str = str(ticket)
		self.assertEqual(
			ticket_str,
			'FailTicket: ip=193.168.0.128 time=1167605999.0 #attempts=5')
		# and some get/set-ers otherwise not tested
		ticket.setTime(1000002000.0)
		self.assertEqual(ticket.getTime(), 1000002000.0)
		# and str() adjusted correspondingly
		self.assertEqual(
			str(ticket),
			'FailTicket: ip=193.168.0.128 time=1000002000.0 #attempts=5')
	
	def testbanNOK(self):
		self.__failManager.setMaxRetry(10)
		self.assertRaises(FailManagerEmpty, self.__failManager.toBan)

	def testWindow(self):
		ticket = self.__failManager.toBan()
		self.assertNotEqual(ticket.getIP(), "100.100.10.10")
		ticket = self.__failManager.toBan()
		self.assertNotEqual(ticket.getIP(), "100.100.10.10")
		self.assertRaises(FailManagerEmpty, self.__failManager.toBan)

示例#6

class AddFailure(unittest.TestCase):
    def setUp(self):
        """Call before every test case."""
        self.__items = [[u'193.168.0.128', 1167605999.0],
                        [u'193.168.0.128', 1167605999.0],
                        [u'193.168.0.128', 1167605999.0],
                        [u'193.168.0.128', 1167605999.0],
                        [u'193.168.0.128', 1167605999.0],
                        ['87.142.124.10', 1167605999.0],
                        ['87.142.124.10', 1167605999.0],
                        ['87.142.124.10', 1167605999.0],
                        ['100.100.10.10', 1000000000.0],
                        ['100.100.10.10', 1000000500.0],
                        ['100.100.10.10', 1000001000.0],
                        ['100.100.10.10', 1000001500.0],
                        ['100.100.10.10', 1000002000.0]]

        self.__failManager = FailManager()
        for i in self.__items:
            self.__failManager.addFailure(FailTicket(i[0], i[1]))

    def tearDown(self):
        """Call after every test case."""

    def testAdd(self):
        self.assertEqual(self.__failManager.size(), 3)

    def _testDel(self):
        self.__failManager.delFailure('193.168.0.128')
        self.__failManager.delFailure('111.111.1.111')

        self.assertEqual(self.__failManager.size(), 1)

    def testCleanupOK(self):
        timestamp = 1167606999.0
        self.__failManager.cleanup(timestamp)
        self.assertEqual(self.__failManager.size(), 0)

    def testCleanupNOK(self):
        timestamp = 1167605990.0
        self.__failManager.cleanup(timestamp)
        self.assertEqual(self.__failManager.size(), 2)

    def testbanOK(self):
        self.__failManager.setMaxRetry(5)
        #ticket = FailTicket('193.168.0.128', None)
        ticket = self.__failManager.toBan()
        self.assertEqual(ticket.getIP(), "193.168.0.128")
        self.assertTrue(isinstance(ticket.getIP(), str))

    def testbanNOK(self):
        self.__failManager.setMaxRetry(10)
        self.assertRaises(FailManagerEmpty, self.__failManager.toBan)

    def testWindow(self):
        ticket = self.__failManager.toBan()
        self.assertNotEqual(ticket.getIP(), "100.100.10.10")
        ticket = self.__failManager.toBan()
        self.assertNotEqual(ticket.getIP(), "100.100.10.10")
        self.assertRaises(FailManagerEmpty, self.__failManager.toBan)

示例#7

class AddFailure(unittest.TestCase):
    def setUp(self):
        """Call before every test case."""
        self.__items = [[u'193.168.0.128', 1167605999.0],
                        [u'193.168.0.128', 1167605999.0],
                        [u'193.168.0.128', 1167605999.0],
                        [u'193.168.0.128', 1167605999.0],
                        [u'193.168.0.128', 1167605999.0],
                        ['87.142.124.10', 1167605999.0],
                        ['87.142.124.10', 1167605999.0],
                        ['87.142.124.10', 1167605999.0],
                        ['100.100.10.10', 1000000000.0],
                        ['100.100.10.10', 1000000500.0],
                        ['100.100.10.10', 1000001000.0],
                        ['100.100.10.10', 1000001500.0],
                        ['100.100.10.10', 1000002000.0]]

        self.__failManager = FailManager()
        for i in self.__items:
            self.__failManager.addFailure(FailTicket(i[0], i[1]))

    def tearDown(self):
        """Call after every test case."""

    def testAdd(self):
        self.assertEqual(self.__failManager.size(), 3)

    def _testDel(self):
        self.__failManager.delFailure('193.168.0.128')
        self.__failManager.delFailure('111.111.1.111')

        self.assertEqual(self.__failManager.size(), 1)

    def testCleanupOK(self):
        timestamp = 1167606999.0
        self.__failManager.cleanup(timestamp)
        self.assertEqual(self.__failManager.size(), 0)

    def testCleanupNOK(self):
        timestamp = 1167605990.0
        self.__failManager.cleanup(timestamp)
        self.assertEqual(self.__failManager.size(), 2)

    def testbanOK(self):
        self.__failManager.setMaxRetry(5)
        #ticket = FailTicket('193.168.0.128', None)
        ticket = self.__failManager.toBan()
        self.assertEqual(ticket.getIP(), "193.168.0.128")
        self.assertTrue(isinstance(ticket.getIP(), str))

        # finish with rudimentary tests of the ticket
        # verify consistent str
        ticket_str = str(ticket)
        self.assertEqual(
            ticket_str,
            'FailTicket: ip=193.168.0.128 time=1167605999.0 #attempts=5')
        # and some get/set-ers otherwise not tested
        ticket.setTime(1000002000.0)
        self.assertEqual(ticket.getTime(), 1000002000.0)
        # and str() adjusted correspondingly
        self.assertEqual(
            str(ticket),
            'FailTicket: ip=193.168.0.128 time=1000002000.0 #attempts=5')

    def testbanNOK(self):
        self.__failManager.setMaxRetry(10)
        self.assertRaises(FailManagerEmpty, self.__failManager.toBan)

    def testWindow(self):
        ticket = self.__failManager.toBan()
        self.assertNotEqual(ticket.getIP(), "100.100.10.10")
        ticket = self.__failManager.toBan()
        self.assertNotEqual(ticket.getIP(), "100.100.10.10")
        self.assertRaises(FailManagerEmpty, self.__failManager.toBan)