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 testHangScript(self):
path = os.path.join(modulePath(), 'testservices', 'HangService.py')
hang = ServiceTask(path, self.team1)
hang.launch("NewFlag")
time.sleep(0.1)
self.assertEquals(hang.isAlive(), True)
hang.finish()
self.assertEquals(hang.isAlive(), False)
hang.quit()
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 testHangScript(self):
path = os.path.join(modulePath(),'testservices','HangService.py')
hang = ServiceTask(path,self.team1)
hang.launch("NewFlag")
time.sleep(0.1)
self.assertEquals(hang.isAlive(),True)
hang.finish()
self.assertEquals(hang.isAlive(),False)
hang.quit()
def __init__(self, conf, init=False):
self.conf = conf
self.servicebot_conf = self.conf.getSection("SERVICE_BOT")
self.flag_manager = self.conf.buildFlagManager()
self.tasks = []
for service in self.servicebot_conf.services:
logger = conf.buildLogger("Service_%s" % service.name)
for team in self.conf.teams:
self.tasks.append(
(team, service, ServiceTask(service.script, team, logger)))
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")
