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)