PonyORM in Pytest
Plugin for use Pony ORM in Pytest.
This Pytest plugin was generated with Cookiecutter along with @hackebrot's Cookiecutter-pytest-plugin template.
- Access Database instance (db) via ponydb fixture
- Mark tests to auto allow use of pony.orm.db_session
- Database is by default cleared after each test
- Database reset can be cancelled for test/class/module
Tested from Pony ORM >= '0.7.3' and python : 3.5, 3.6
You can install "pytest-ponyorm" via pip from PyPI:
$ pip install pytest-ponyorm or $ pipenv install -d pytest-ponyorm
Warning
By default, database is cleared after each test. You must never use it in a production environnement. This plugin doesn't change the database config.It's you'job to create the testing environnement and change database params.
First, configure PONY_DB in pytest.ini. PonyORM main Database instance module location must be specified in pytest.ini to make it work : for example if db is in /path/models/main.py, you must configure like this :
[pytest]
PONY_DB=path.models.main
Then just apply the pony marker :
# models.py
db = Database()
# test.py
@pytest.mark.pony
def my_test(ponydb):
new_mod = ponydb.Mymodel(name="me",...)
You can mark a class or function with @pytest.mark.pony or the whole module with pytestmark = pytest.mark.pony
The marker pony takes one argument : reset_db, default is True. In this case the marked test doesn't reset the database at ending.
# test.py
pytestmark = pytest.mark.pony # marks all tests of the module
def test1:
pass
def test2:
pass
@pytest.mark.pony(reset_db=False)
def test 3:
pass
# test3 will use database in the state that test2 left it.
Test database tables are dropped/recreate before the test SESSION.
Initialy this plugin did drop/recreate table for each test. Due tu perfomance reason, this has changed. Now tables are cleared but not dropped.
Sql sequences like "auto PrimaryKey" will also be reseted but actually, it's only supported for PostreSQL and Sqlite. For other databases, Primarikey might not start at 1 for each test. This has to be considered when writing tests.
Fixtures should not use db_session decorator or context manager. Each test is automaticaly run inside a db_session and each operation inside fixtures will be commited at test start.
Contributions are very welcome. Tests can be run with tox, please ensure the coverage at least stays the same before you submit a pull request.
Distributed under the terms of the GNU GPL v3.0 license, "pytest-ponyorm" is free and open source software
If you encounter any problems, please file an issue along with a detailed description.
- 0.3.1 :
- sql sequence reset for postgre and sqlite
- testing py35/36 and pg/sqlite
- 0.3.0 :
- made faster with just deleting database entries after each test
- no drop/recreate
- db_session splitted into 2 hooks, not anymore in a fixture
- 0.2.9 :
- add fixture autocommit before run test
- 0.2.0 :
- add marker
- 0.1.5 :
- add db_session for each test
- 0.1.0 :
- auto clear database