119 lines
3.7 KiB
Python
119 lines
3.7 KiB
Python
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
|