def testGetCompletedWorkItems(self):
workerDataDB = initializeWorkerDataDB(self.testName)
startDate = nowInSeconds()- 5;
workerData = WorkerData(None, {"request_id":"foo1","worker_id":"abcd","fib_id":3,"startedDate":startDate})
workerDataDB.addWorkerData(workerData)
workerData.fibValue = 3
workerDataDB.updateWorkerData(workerData)
workerData = WorkerData(None, {"request_id":"foo2","worker_id":"efgh","fib_id":5,"startedDate":startDate})
workerDataDB.addWorkerData(workerData)
workerData.fibValue = 8
workerDataDB.updateWorkerData(workerData)
workerDatas = workerDataDB.getWorkItems(isPending = True)
self.assertTrue(len(workerDatas) == 0)
workerDatas = workerDataDB.getWorkItems(isPending = False)
for workerData in workerDatas:
self.assertTrue(workerData.fibValue != -1)
self.assertTrue(workerData.finishedDate != None)
workerDatas = workerDataDB.getWorkItems(isPending = True)
self.assertTrue(len(workerDatas) == 0)
def testGetCompletedWorkItems(self):
workerDataDB = initializeWorkerDataDB(self.testName)
startDate = nowInSeconds() - 5
workerData = WorkerData(
None, {
"request_id": "foo1",
"worker_id": "abcd",
"fib_id": 3,
"startedDate": startDate
})
workerDataDB.addWorkerData(workerData)
workerData.fibValue = 3
workerDataDB.updateWorkerData(workerData)
workerData = WorkerData(
None, {
"request_id": "foo2",
"worker_id": "efgh",
"fib_id": 5,
"startedDate": startDate
})
workerDataDB.addWorkerData(workerData)
workerData.fibValue = 8
workerDataDB.updateWorkerData(workerData)
workerDatas = workerDataDB.getWorkItems(isPending=True)
self.assertTrue(len(workerDatas) == 0)
workerDatas = workerDataDB.getWorkItems(isPending=False)
for workerData in workerDatas:
self.assertTrue(workerData.fibValue != -1)
self.assertTrue(workerData.finishedDate != None)
workerDatas = workerDataDB.getWorkItems(isPending=True)
self.assertTrue(len(workerDatas) == 0)
def testUpdateWorkerData(self):
workerDataDB = initializeWorkerDataDB(self.testName)
lastCheckinDate = nowInSeconds()
lastCheckinDate = lastCheckinDate-5
workerData = WorkerData(None, {"request_id":"foo1","worker_id":"abcd","fib_id":3,"started_date":nowInSeconds()-5})
newWorkerData = workerDataDB.addWorkerData(workerData)
workerData.fibValue=3
workerDataDB.updateWorkerData(workerData)
checkWorkerData = workerDataDB.getWorkerData(workerData.workerId)
self.assertTrue(checkWorkerData.finishedDate != None)
def testUpdateWorkerData(self):
workerDataDB = initializeWorkerDataDB(self.testName)
lastCheckinDate = nowInSeconds()
lastCheckinDate = lastCheckinDate - 5
workerData = WorkerData(
None, {
"request_id": "foo1",
"worker_id": "abcd",
"fib_id": 3,
"started_date": nowInSeconds() - 5
})
newWorkerData = workerDataDB.addWorkerData(workerData)
workerData.fibValue = 3
workerDataDB.updateWorkerData(workerData)
checkWorkerData = workerDataDB.getWorkerData(workerData.workerId)
self.assertTrue(checkWorkerData.finishedDate != None)
