def test_get_one_record(self):
dbdriver = DBdriver(config.database)
dbdriver.connect()
document = dbdriver.get_one_record()
if document is None:
object_id = dbdriver.cHandle.insert_one({"status": "submitted"}).inserted_id
document = dbdriver.get_one_record()
self.failUnlessEqual(str(object_id), str(document["_id"]))
dbdriver.cHandle.remove({"_id": object_id})
else:
self.assertEqual(document['status'], "pending")
dbdriver.cHandle.remove({"_id": document["_id"]})
dbdriver.disconnect()
