def testBasicAttack(self): path = os.path.join(modulePath(),'testexploits','Succesful.py') attack = AttackTask(path,"127.0.0.1",2,None) attack.launch() time.sleep(0.1) attack.processOutput() flags = attack.collectFlags() self.assertEquals(flags,["foobarbaz"]) attack.quit()
def testRestartAttack(self): path = os.path.join(modulePath(),'testexploits','Succesful.py') attack = AttackTask(path,"127.0.0.1",2,None) for i in range(5): attack.launch() time.sleep(0.1) self.assertFalse(attack.isAlive()) attack.processOutput() flags = attack.collectFlags() self.assertEquals(flags,["foobarbaz"]*5) attack.quit()
def testCookieAttack(self): path = os.path.join(modulePath(),'testexploits','Cookie.py') attack = AttackTask(path,"127.0.0.1",2,None) for i in range(5): attack.launch() time.sleep(0.1) self.assertFalse(attack.isAlive()) attack.processOutput() flags = attack.collectFlags() for i in range(len(flags)): self.assertEquals(flags[i],"COOKIEFLAG%d"%i) attack.quit()