def test_get_all_records(self): dbdriver = DBdriver(config.database) dbdriver.connect() document = dbdriver.get_all_records() if len(document) == 2: object_id = dbdriver.cHandle.insert_one({"status": "successful"}).inserted_id document = dbdriver.get_all_records() self.assertIsNotNone(document) dbdriver.cHandle.remove({"_id": object_id}) self.assertGreater(len(document), 2) dbdriver.disconnect()