grung-db/test/test_db.py

from pprint import pprint as print

import pytest
from tinydb import Query
from tinydb.storages import MemoryStorage

from grung import examples
from grung.db import GrungDB
from grung.exceptions import PointerReferenceError, UniqueConstraintError


@pytest.fixture
def db():
    _db = GrungDB.with_schema(examples, storage=MemoryStorage)
    yield _db
    print(_db)


def test_crud(db):
    user = examples.User(name="john", number=23, email="john@foo")
    assert user.uid
    assert user._metadata.fields["uid"].unique

    # insert
    john_something = db.save(user)
    last_insert_id = john_something.doc_id

    # read back
    assert db.User.get(doc_id=last_insert_id) == john_something
    assert john_something.name == user.name
    assert john_something.number == 23
    assert john_something.email == user.email
    assert john_something.uid == user.uid

    # update
    john_something.name = "james?"
    before_update = db.User.get(doc_id=john_something.doc_id)
    after_update = db.save(john_something)
    assert after_update == john_something
    assert before_update != after_update

    # delete
    db.delete(john_something)
    assert len(db.User) == 0


def test_pointers(db):
    user = examples.User(name="john", email="john@foo")
    players = examples.Group(name="players", members=[user])

    with pytest.raises(PointerReferenceError):
        players = db.save(players)

    user = db.save(user)
    players = db.save(examples.Group(name="players", members=[user]))

    user = db.table("User").get(doc_id=user.doc_id)
    assert user.groups.uid == players.uid

    assert players.members[0].groups.uid == players.uid


def test_subgroups(db):
    kirk = db.save(examples.User(name="James T. Kirk", email="riskybiznez@starfleet"))
    pike = db.save(examples.User(name="Christopher Pike", email="hitit@starfleet"))

    tos = db.save(examples.Group(name="The Original Series", members=[kirk]))
    snw = db.save(examples.Group(name="Strange New Worlds", members=[pike]))

    starfleet = db.save(examples.Group(name="Starfleet", groups=[tos, snw]))
    tos = db.table("Group").get(doc_id=tos.doc_id)
    snw = db.table("Group").get(doc_id=snw.doc_id)
    assert tos in starfleet.groups
    assert snw in starfleet.groups

    assert tos.parent.uid == starfleet.uid

    unique_users = set([user for group in starfleet.groups for user in group.members])

    kirk = db.table("User").get(doc_id=kirk.doc_id)
    assert kirk in unique_users


def test_unique(db):
    user1 = examples.User(name="john", email="john@foo")
    user2 = examples.User(name="john", email="john@foo")

    user1 = db.save(user1)
    with pytest.raises(UniqueConstraintError):
        user2 = db.save(user2)
    db.save(user1)


def test_search(db):
    # create crew members
    kirk = db.save(examples.User(name="Captain James T. Kirk", email="riskybiznez@starfleet"))
    bones = db.save(examples.User(name="Doctor McCoy", email="dammitjim@starfleet"))
    ricky = db.save(examples.User(name="Ensign Ricky Redshirt", email="invincible@starfleet"))

    # create the crew record
    crew = db.save(examples.Group(name="Crew", members=[kirk, bones, ricky]))

    User = Query()
    captains = db.User.search(User.name.matches("Captain"))
    assert len(captains) == 1

    # update the crew members so they have the backreference to crew
    kirk = db.table("User").get(doc_id=kirk.doc_id)
    bones = db.table("User").get(doc_id=bones.doc_id)
    ricky = db.table("User").get(doc_id=ricky.doc_id)

    assert kirk in crew.members
    assert bones in crew.members
    assert ricky in crew.members

    Group = Query()
    crew = db.Group.search(Group.name == "Crew", recurse=False)
    assert f"User::{kirk.uid}" in crew[0].members