/rethinkdb-mock

Mock library for Python rethinkdb==2.4.9

Primary LanguagePythonMIT LicenseMIT

build rethinkdb PyPI version

RethinkDB Mock

Formerly known as MockThink by Scivey, this is a continuation of that work released under a new name.

Rethinkdb-mock is an in-process Python clone of RethinkDB's API. For testing.

MockThink provides a stub connection object which can be passed to normal ReQL queries. Instead of being serialized and sent to the server, the ReQL AST is run through an interpreter in the same process. "Tables" and "databases" are based on data given to the MockThink constructor.

Avoiding network calls (for tests themselves as well as setup/teardown) makes testing queries with MockThink orders of magnitude faster.

Install

pip install rethinkdb-mock
# or
pipenv install --dev rethinkdb-mock

Usage

Basic

from pprint import pprint
from rethinkdb import RethinkDB
from rethinkdb_mock import MockThink

r = RethinkDB()
db = MockThink({
    'dbs': {
        'tara': {
            'tables': {
                'people': [
                    {'id': 'john-id', 'name': 'John'},
                    {'id': 'sam-id', 'name': 'Sam'}
                ]
            }
        }
    }
})

def test_mytest():
    with db.connect() as conn:
        result = r.db('tara').table('people').map(
            lambda doc: doc.merge({'also_name': doc['name']})
        ).run(conn)
        pprint(list(result))

        # [
        #    {'also_name': 'John', 'id': 'john-id', 'name': 'John'},
        #    {'also_name': 'Sam', 'id': 'sam-id', 'name': 'Sam'}
        # ]

        r.db('tara').table('people').update(
            {'likes_fonz': True}
        ).run(conn)

        result = r.db('tara').table('people').run(conn)
        pprint(list(result))

        # [
        #    {'id': 'john-id', 'likes_fonz': True, 'name': 'John'},
        #    {'id': 'sam-id', 'likes_fonz': True, 'name': 'Sam'}
        # ]

    # data is reset at exit of context manager above

    with db.connect() as conn:
        result = r.db('tara').table('people').run(conn)
        pprint(list(result))
        # [
        #    {'id': 'john-id', 'name': 'John'},
        #    {'id': 'sam-id', 'name': 'Sam'}
        # ]

Set the default database for the connection

Like r.connect(db='database')

from pprint import pprint
from rethinkdb import RethinkDB
from rethinkdb_mock import MockThink

r = RethinkDB()

def test_mytest_with_default_db():
    db = MockThink(
        {
            "dbs": {
                "tara": {
                    "tables": {
                        "people": [
                            {
                                "id": "john-id",
                                "first_name": "John",
                                "last_name": "Generic",
                            },
                            {
                                "id": "sam-id", "first_name":
                                "Sam", "last_name": "Dull"
                            },
                            {
                                "id": "adam-id",
                                "first_name": "Adam",
                                "last_name": "Average",
                            },
                        ]
                    }
                }
            },
            "default": "tara",
        }
    )

    with db.connect() as conn:
        r.table("people").index_create(
            "full_name", lambda doc: doc["last_name"] + doc["first_name"]
        ).run(conn)

        r.table("people").index_wait().run(conn)

        result = (
            r.table("people")
            .get_all("GenericJohn", "AverageAdam", index="full_name")
            .run(conn)
        )
        pprint(list(result))
        # {'id': 'john-id', 'first_name': 'John', 'last_name': 'Generic'},
        # {'id': 'adam-id', 'first_name': 'Adam', 'last_name': 'Average'}

Full support for secondary indexes

from pprint import pprint
from rethinkdb import RethinkDB
from rethinkdb_mock import MockThink

r = RethinkDB()

db = MockThink({
    'dbs': {
        'tara': {
            'tables': {
                'people': [
                    {'id': 'john-id', 'first_name': 'John', 'last_name': 'Generic'},
                    {'id': 'sam-id', 'first_name': 'Sam', 'last_name': 'Dull'},
                    {'id': 'adam-id', 'first_name': 'Adam', 'last_name': 'Average'}
                ]
            }
        }
    }
})

with db.connect() as conn:

    r.db('tara').table('people').index_create(
        'full_name',
        lambda doc: doc['last_name'] + doc['first_name']
    ).run(conn)

    r.db('tara').table('people').index_wait().run(conn)

    result = r.db('tara').table('people').get_all(
        'GenericJohn', 'AverageAdam', index='full_name'
    ).run(conn)
    pprint(list(result))
    # {'id': 'john-id', 'first_name': 'John', 'last_name': 'Generic'},
    # {'id': 'adam-id', 'first_name': 'Adam', 'last_name': 'Average'}

Testing

The most confusing test failures are those caused by errors in test frameworks and harnesses themselves. This means they need to be tested very thoroughly.

The main testing is a suite of functional tests which are targeted at the individual query level, e.g. testing all the ways in which r.merge might be called.

These are all complete ReQL queries, but avoid complexity beyond the target query to make failures easier to diagnose.

The integration tests cover more complicated queries, e.g. eq_join->map->eq_join->map.

Both the functional and integration tests have two modes of execution, rethinkdb_mock and rethink. The second mode runs the same tests against a running RethinkDB instance, and is much slower due to the network calls. rethinkdb_mock mode is for testing MockThink's behavior against our expectations; rethink mode is for testing our expectations against reality.

Run tests

# Install dependencies
pip install pipenv
pipenv sync --dev

# Run the unit tests
pipenv run test

Run tests against a live server

The docker folder contains a dockerfile that grabs rethinkdb and runs the python tests

Simply running the below command outputs the results

docker-compose up --build

Code formatting

The tox.ini file contains the configuration for code formatting

# Install dependencies
pip install pipenv
pipenv sync --dev

# Format imports
pipenv run isort

# Check styling with Flake8
pipenv run lint

License

The MIT License (MIT)