def testGetPendingWorkItems(self):
workerDataDB = initializeWorkerDataDB(self.testName)
workerData = WorkerData(None, {
"request_id": "foo1",
"worker_id": "abcd",
"fib_id": 3
})
workerDataDB.addWorkerData(workerData)
workerData = WorkerData(None, {
"request_id": "foo2",
"worker_id": "efgh",
"fib_id": 5
})
workerDataDB.addWorkerData(workerData)
timestamp = nowInSeconds()
workerDatas = workerDataDB.getWorkItems(isPending=True)
for workerData in workerDatas:
self.assertTrue(workerData.fibValue == -1)
self.assertTrue(workerData.finishedDate == None)
workerDatas = workerDataDB.getWorkItems(isPending=False)
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 test1AddWorkerDataAtStart(self):
workerDataDB = initializeWorkerDataDB(self.testName)
workerData = WorkerData(None, {"request_id":"foo1","worker_id":"abcd","fib_id":3})
newWorkerData = workerDataDB.addWorkerData(workerData)
assert newWorkerData.workerId > 0
def testUpdateRetryCount(self):
workerDataDB = initializeWorkerDataDB(self.testName)
workerData = WorkerData(None, {"request_id":"foo1","worker_id":"abcd","fib_id":3})
workerDataDB.addWorkerData(workerData)
newRetryCount = workerData.retryCount = workerData.retryCount+1
workerDataDB.updateWorkerData(workerData)
updatedWorkerData = workerDataDB.getWorkerData(workerId = "abcd")
self.assertTrue(updatedWorkerData.retryCount == newRetryCount)
def test1AddWorkerDataAtStart(self):
workerDataDB = initializeWorkerDataDB(self.testName)
workerData = WorkerData(None, {
"request_id": "foo1",
"worker_id": "abcd",
"fib_id": 3
})
newWorkerData = workerDataDB.addWorkerData(workerData)
assert newWorkerData.workerId > 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 testUpdateRetryCount(self):
workerDataDB = initializeWorkerDataDB(self.testName)
workerData = WorkerData(None, {
"request_id": "foo1",
"worker_id": "abcd",
"fib_id": 3
})
workerDataDB.addWorkerData(workerData)
newRetryCount = workerData.retryCount = workerData.retryCount + 1
workerDataDB.updateWorkerData(workerData)
updatedWorkerData = workerDataDB.getWorkerData(workerId="abcd")
self.assertTrue(updatedWorkerData.retryCount == newRetryCount)
def test2GetWorkerData(self):
workerDataDB = initializeWorkerDataDB(self.testName)
workerData = WorkerData(None, {"request_id":"foo1","worker_id":"abcd","fib_id":3})
newWorkerData = workerDataDB.addWorkerData(workerData)
checkWorkerData = workerDataDB.getWorkerData(workerId = newWorkerData.workerId)
self.assertTrue(checkWorkerData.requestId == workerData.requestId)
self.assertTrue(checkWorkerData.workerId == workerData.workerId)
self.assertTrue(checkWorkerData.fibId == workerData.fibId)
checkWorkerData = workerDataDB.getWorkerData(requestId = newWorkerData.requestId)
self.assertTrue(checkWorkerData.requestId == workerData.requestId)
self.assertTrue(checkWorkerData.workerId == workerData.workerId)
self.assertTrue(checkWorkerData.fibId == workerData.fibId)
def testGetPendingWorkItems(self):
workerDataDB = initializeWorkerDataDB(self.testName)
workerData = WorkerData(None, {"request_id":"foo1","worker_id":"abcd","fib_id":3})
workerDataDB.addWorkerData(workerData)
workerData = WorkerData(None, {"request_id":"foo2","worker_id":"efgh","fib_id":5})
workerDataDB.addWorkerData(workerData)
timestamp = nowInSeconds()
workerDatas = workerDataDB.getWorkItems(isPending = True)
for workerData in workerDatas:
self.assertTrue(workerData.fibValue == -1)
self.assertTrue(workerData.finishedDate == None)
workerDatas = workerDataDB.getWorkItems(isPending = False)
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 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 test2GetWorkerData(self):
workerDataDB = initializeWorkerDataDB(self.testName)
workerData = WorkerData(None, {
"request_id": "foo1",
"worker_id": "abcd",
"fib_id": 3
})
newWorkerData = workerDataDB.addWorkerData(workerData)
checkWorkerData = workerDataDB.getWorkerData(
workerId=newWorkerData.workerId)
self.assertTrue(checkWorkerData.requestId == workerData.requestId)
self.assertTrue(checkWorkerData.workerId == workerData.workerId)
self.assertTrue(checkWorkerData.fibId == workerData.fibId)
checkWorkerData = workerDataDB.getWorkerData(
requestId=newWorkerData.requestId)
self.assertTrue(checkWorkerData.requestId == workerData.requestId)
self.assertTrue(checkWorkerData.workerId == workerData.workerId)
self.assertTrue(checkWorkerData.fibId == workerData.fibId)
