This is a package based on the mongomock package to test any code using MongoDB via the PyMongo API.
With mongomock the documents are stored in memory and not available after the process performing the tests exits. In some test cases it is necessary to have a non-volatile (persistent) storage of the database. For example, tests of command line interfaces (CLI) add things into the database and assume they are then available across several processes.
This package adds an option to store the database in a file. This is accomplished by small extensions of the classes in store.py. These extensions are not intended to keep the database synchronized instantly on disk. Rather the database is eventually dumped to a file only just before the ServerStore object is destroyed. After that the file can be used to create other ServerStore objects in the final state of the original ServerStore object.
-
In test cases where a database must be used by several consecutive processes and the database state has to be preserved in the meantimes.
-
In tutorials to learn the basics of using PyMongo-based software.
-
In further single-client applications of MongoDB.
In all use cases one has to make sure that all PyMongo features used are covered by mongomock. See this for further details.
The package can be easily installed using pip:
pip install mongomock-persistence
To use the package the MongoClient class from this package must be imported instead of that from mongomock. Then the persistence can be activated in two ways:
- Set the environment variable
MONGOMOCK_SERVERSTORE_FILEto the name of a non-empty JSON file (initialized with '{}') and callMongoClientclass as usual:
export MONGOMOCK_SERVERSTORE_FILE=/full/path/to/mongomock_file.json
from mongomock_persistence import MongoClient
mongo_client = MongoClient()- Create a custom
ServerStoreobject explicitly by using thefilenamekeyword argument and then pass it toMongoClientcall:
from mongomock_persistence import MongoClient
mongo_store = ServerStore(filename='/full/path/to/mongomock_file.json')
mongo_client = MongoClient(_store=mongo_store)Install the test extra and then run pytest, i.e.
pip install mongomock-persistence[test]
pytest <root folder of repository>