def test_find_document_by_id(mocker):
db_name = "test_db"
coll_name = "test_coll"
query = {'query', 'query_value'}
mongo_client = mongomock.MongoClient()
conn = MongoConnection(db_name, coll_name)
conn.collection = mongo_client.collection
mocker.patch.object(conn.collection, 'find_one', return_value=False)
conn.find_document_by_id(query)
conn.collection.find_one.assert_called_once_with(query)
def test_create_document(mocker):
db_name = "test_db"
coll_name = "test_coll"
document = {'document', 'document_value'}
mongo_client = mongomock.MongoClient()
conn = MongoConnection(db_name, coll_name)
conn.collection = mongo_client.collection
mocker.patch.object(conn.collection, 'insert_one', return_value=False)
conn.create_document(document)
conn.collection.insert_one.assert_called_once_with(document)
def test_update_document(mocker):
db_name = "test_db"
coll_name = "test_coll"
query = {'query': 'query_value'}
document = {'document', 'document_value'}
mongo_client = mongomock.MongoClient()
conn = MongoConnection(db_name, coll_name)
conn.collection = mongo_client.collection
mocker.patch.object(conn.collection, 'update', return_value=False)
conn.update_document(query, document)
conn.collection.update.assert_called_once_with(query, document)