def testCookieScript(self):
path = os.path.join(modulePath(), 'testservices', 'StoreService.py')
store = ServiceTask(path, self.team1)
store.launch("NewFlag")
time.sleep(0.1)
store.finish()
store.processOutput()
store.quit()
#self.assertEquals(store.status() == ServiceTask.INVALID_FLAG)
self.assertEquals(store.cookie(), "Store Text")
def testBrokenScript(self):
path = os.path.join(modulePath(), 'testservices', 'BrokenService.py')
broken = ServiceTask(path, self.team1)
broken.launch("NewFlag")
time.sleep(0.1)
broken.finish()
broken.processOutput()
broken.quit()
self.assertEquals(broken.prevFlag(), None)
def testErrorScript(self):
path = os.path.join(modulePath(), 'testservices', 'ErrorService.py')
err = ServiceTask(path, self.team1)
err.launch("NewFlag")
time.sleep(0.1)
err.finish()
err.processOutput()
err.quit()
self.assertEquals(err.error(), "An Error")
def testCookieScript(self):
path = os.path.join(modulePath(),'testservices','StoreService.py')
store = ServiceTask(path,self.team1)
store.launch("NewFlag")
time.sleep(0.1)
store.finish()
store.processOutput()
store.quit()
#self.assertEquals(store.status() == ServiceTask.INVALID_FLAG)
self.assertEquals(store.cookie(),"Store Text")
def testErrorScript(self):
path = os.path.join(modulePath(),'testservices','ErrorService.py')
err = ServiceTask(path,self.team1)
err.launch("NewFlag")
time.sleep(0.1)
err.finish()
err.processOutput()
err.quit()
self.assertEquals(err.error(),"An Error")
def testBrokenScript(self):
path = os.path.join(modulePath(),'testservices','BrokenService.py')
broken = ServiceTask(path,self.team1)
broken.launch("NewFlag")
time.sleep(0.1)
broken.finish()
broken.processOutput()
broken.quit()
self.assertEquals(broken.prevFlag(),None)
def testCorrectScript(self):
self.__ensureValidGoodService()
path = os.path.join(modulePath(), 'testservices', 'GoodService.py')
good = ServiceTask(path, self.team1)
good.launch("NewFlag")
time.sleep(0.1)
good.finish()
good.processOutput()
good.quit()
self.assertEquals(good.prevFlag(), "TestFlag")
def testCorrectScript(self):
self.__ensureValidGoodService()
path = os.path.join(modulePath(),'testservices','GoodService.py')
good = ServiceTask(path,self.team1)
good.launch("NewFlag")
time.sleep(0.1)
good.finish()
good.processOutput()
good.quit()
self.assertEquals(good.prevFlag(),"TestFlag")