def testDbworkerClosesConnection(self):
dbqueue = mock.Mock()
dbqueue.get.side_effect = self.testdata
Uforia.dbworker(dbqueue, self.db)
self.assertTrue(self.db.connection.close.called, "database connection was not closed")
def testDbworkerStopsWithNoTasks(self):
dbqueue = mock.Mock()
dbqueue.get.return_value = self.sentinel
Uforia.dbworker(dbqueue, self.db)
self.assertTrue(dbqueue.task_done.called, "dbqueue task must be set as done")
def testDbWorkerDoesNotStoreWithNoTasks(self):
dbqueue = mock.Mock()
dbqueue.get.return_value = self.sentinel
Uforia.dbworker(dbqueue, self.db)
self.assertFalse(self.db.store.called, "db.store was called while sentinel value was sent")
def testDbworkerClosesConnection(self):
dbqueue = mock.Mock()
dbqueue.get.side_effect = self.testdata
Uforia.dbworker(dbqueue, self.db)
self.assertTrue(self.db.connection.close.called, "database connection was not closed")
def testDbWorkerDoesNotStoreWithNoTasks(self):
dbqueue = mock.Mock()
dbqueue.get.return_value = self.sentinel
Uforia.dbworker(dbqueue, self.db)
self.assertFalse(self.db.store.called, "db.store was called while sentinel value was sent")
def testDbworkerStopsWithNoTasks(self):
dbqueue = mock.Mock()
dbqueue.get.return_value = self.sentinel
Uforia.dbworker(dbqueue, self.db)
self.assertTrue(dbqueue.task_done.called, "dbqueue task must be set as done")
def testDbworkerStoresDataCorrectly(self):
dbqueue = mock.Mock()
# side_effect allows us to return a different value each time
# the method is called
dbqueue.get.side_effect = self.testdata
Uforia.dbworker(dbqueue, self.db)
expected = [mock.call(*self.testdata[0]), mock.call(*self.testdata[1])]
self.assertTrue(self.db.store.call_args_list == expected,
"data was not written correctly or in the same order")
def testDbworkerStoresDataCorrectly(self):
dbqueue = mock.Mock()
# side_effect allows us to return a different value each time
# the method is called
dbqueue.get.side_effect = self.testdata
Uforia.dbworker(dbqueue, self.db)
expected = [
mock.call(*self.testdata[0]),
mock.call(*self.testdata[1])]
self.assertTrue(self.db.store.call_args_list == expected, "data was not written correctly or in the same order")