1 files changed, 24 insertions, 0 deletions
diff --git a/mongita_test.py b/mongita_test.py
new file mode 100644
index 0000000..18a4864
--- /dev/null
+++ b/mongita_test.py
@@ -0,0 +1,24 @@
+from mongita import MongitaClientDisk
+from pymongo import MongoClient
+
+
+def test_ids(client):
+    collection = client['test_db']['test_collection']
+    collection.replace_one(
+        {'_id': 'id_from_filter'},
+        replacement={'key': 'value'},
+        upsert=True,
+    )
+    doc = collection.find_one({'_id': 'id_from_filter'})
+    print(f'Fetched document by ID: {doc}')
+
+    print('All IDs:')
+    for d in collection.find({}):
+        print(d['_id'])
+
+
+print('pymongo\n----------')
+test_ids(MongoClient())
+
+print('\nmongita\n----------')
+test_ids(MongitaClientDisk())