コード例 #1
0
    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()