summaryrefslogtreecommitdiff
path: root/mongita_test.py
blob: 18a486453cb7514b66592b7e5672af60d4132b2f (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
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())