def setUp(self): """Call before every test case.""" self.__jail = DummyJail() self.__actions = Actions(self.__jail) self.__tmpfile, self.__tmpfilename = tempfile.mkstemp()
class ExecuteActions(unittest.TestCase): def setUp(self): """Call before every test case.""" self.__jail = DummyJail() self.__actions = Actions(self.__jail) self.__tmpfile, self.__tmpfilename = tempfile.mkstemp() def tearDown(self): os.remove(self.__tmpfilename) def defaultActions(self): self.__actions.addAction('ip') self.__ip = self.__actions.getAction('ip') self.__ip.setActionStart('echo ip start 64 >> "%s"' % self.__tmpfilename) self.__ip.setActionBan('echo ip ban <ip> >> "%s"' % self.__tmpfilename) self.__ip.setActionUnban('echo ip unban <ip> >> "%s"' % self.__tmpfilename) self.__ip.setActionCheck('echo ip check <ip> >> "%s"' % self.__tmpfilename) self.__ip.setActionStop('echo ip stop >> "%s"' % self.__tmpfilename) def testActionsManipulation(self): self.__actions.addAction('test') self.assertTrue(self.__actions.getAction('test')) self.assertTrue(self.__actions.getLastAction()) self.assertRaises(KeyError, self.__actions.getAction, *['nonexistant action']) self.__actions.addAction('test1') self.__actions.delAction('test') self.__actions.delAction('test1') self.assertRaises(KeyError, self.__actions.getAction, *['test']) self.assertRaises(IndexError, self.__actions.getLastAction) self.__actions.setBanTime(127) self.assertEqual(self.__actions.getBanTime(), 127) self.assertRaises(ValueError, self.__actions.removeBannedIP, '127.0.0.1') def testActionsOutput(self): self.defaultActions() self.__actions.start() f = open(self.__tmpfilename) time.sleep(3) self.assertEqual(f.read(), "ip start 64\n") self.__actions.stop() self.__actions.join() self.assertEqual(self.__actions.status(), [("Currently banned", 0), ("Total banned", 0), ("IP list", [])])
class ExecuteActions(unittest.TestCase): def setUp(self): """Call before every test case.""" self.__jail = DummyJail() self.__actions = Actions(self.__jail) self.__tmpfile, self.__tmpfilename = tempfile.mkstemp() def tearDown(self): os.remove(self.__tmpfilename) def defaultActions(self): self.__actions.addAction('ip') self.__ip = self.__actions.getAction('ip') self.__ip.setActionStart('echo ip start 64 >> "%s"' % self.__tmpfilename ) self.__ip.setActionBan('echo ip ban <ip> >> "%s"' % self.__tmpfilename ) self.__ip.setActionUnban('echo ip unban <ip> >> "%s"' % self.__tmpfilename ) self.__ip.setActionCheck('echo ip check <ip> >> "%s"' % self.__tmpfilename ) self.__ip.setActionStop('echo ip stop >> "%s"' % self.__tmpfilename ) def testActionsManipulation(self): self.__actions.addAction('test') self.assertTrue(self.__actions.getAction('test')) self.assertTrue(self.__actions.getLastAction()) self.assertRaises(KeyError,self.__actions.getAction,*['nonexistant action']) self.__actions.addAction('test1') self.__actions.delAction('test') self.__actions.delAction('test1') self.assertRaises(KeyError, self.__actions.getAction, *['test']) self.assertRaises(IndexError,self.__actions.getLastAction) self.__actions.setBanTime(127) self.assertEqual(self.__actions.getBanTime(),127) self.assertRaises(ValueError, self.__actions.removeBannedIP, '127.0.0.1') def testActionsOutput(self): self.defaultActions() self.__actions.start() f = open(self.__tmpfilename) time.sleep(3) self.assertEqual(f.read(),"ip start 64\n") self.__actions.stop() self.__actions.join() self.assertEqual(self.__actions.status(),[("Currently banned", 0 ), ("Total banned", 0 ), ("IP list", [] )])