def testCorrectScript(self):
self.__ensureValidGoodService()
path = os.path.join(modulePath(), 'testservices', 'GoodService.py')
good = ServiceTask(path, 1, "NewFlag", "0.0.0.0", None)
good.start()
self.assert_(good.status() == ServiceTask.OK)
self.assert_(good.getPrevFlag() == "TestFlag")
def testSimpleInsert(self):
test = ServiceTaskRunner()
test.start()
script = modulePath() + "/testservices/HangService.py"
task = ServiceTask(script, 2, "Flag", "10.0.0.1", None)
test.insert(task, -1, -1)
test.terminateOnEmpty()
self.assert_(task.status() == ServiceTask.KILLED)
def build(self,teamId,serviceId,round): team = self.conf.getTeamInfoById(teamId) service = self.servicebot_conf.getServiceInfoById(serviceId) flag = Flag(teamId,serviceId,round,time.time()) task = ServiceTask( service.script, service.timeout, self.flag_manager.toTxt(flag), team.host, self.storage[teamId][serviceId], self.service_loggers[service.name]) return task
def testErrorInsert(self):
test = ServiceTaskRunner()
test.start()
script = modulePath() + "/testservices/ErrorService.py"
tasks = []
for i in range(0, 10):
task = ServiceTask(script, 1, "Flag", "10.0.0.1", None)
tasks.append(task)
test.insert(task, -1, -1)
test.terminateOnEmpty()
for task in tasks:
self.assert_(task.status() == ServiceTask.ERROR)
def testPastTimeout(self):
path = os.path.join(modulePath(), 'testservices', 'GoodService.py')
good = ServiceTask(path, 1, "NewFlag", "0.0.0.0", None)
good.start()
time.sleep(2)
self.assert_(good.status() == ServiceTask.OK)
def testStoreScript(self):
path = os.path.join(modulePath(), 'testservices', 'StoreService.py')
store = ServiceTask(path, 0.5, "NewFlag", "0.0.0.0", None)
store.start()
self.assert_(store.status() == ServiceTask.INVALID_FLAG)
self.assert_(store.store() == "Store Text")
def testHangScript(self):
path = os.path.join(modulePath(), 'testservices', 'HangService.py')
hang = ServiceTask(path, 1, "NewFlag", "0.0.0.0", None)
hang.start()
self.assert_(hang.status() == ServiceTask.KILLED)
def testErrorScript(self):
path = os.path.join(modulePath(), 'testservices', 'ErrorService.py')
err = ServiceTask(path, 1, "NewFlag", "0.0.0.0", None)
err.start()
self.assert_(err.status() == ServiceTask.ERROR)
self.assert_(err.error() == "An Error")
def testBrokenScript(self):
path = os.path.join(modulePath(), 'testservices', 'BrokenService.py')
err = ServiceTask(path, 2, "NewFlag", "0.0.0.0", None)
err.start()
self.assert_(err.status() == ServiceTask.INVALID_FLAG)