74 lines
2.4 KiB
Python
74 lines
2.4 KiB
Python
import json
|
|
from pathlib import Path
|
|
from unittest.mock import MagicMock
|
|
|
|
import pytest
|
|
|
|
from ttfrog.db import schema
|
|
from ttfrog.db.manager import db as _db
|
|
|
|
FIXTURE_PATH = Path(__file__).parent / "fixtures"
|
|
|
|
|
|
def load_fixture(db, fixture_name):
|
|
with db.transaction():
|
|
data = json.loads((FIXTURE_PATH / f"{fixture_name}.json").read_text())
|
|
for schema_name in data:
|
|
for record in data[schema_name]:
|
|
print(f"Loading {schema_name} {record = }")
|
|
obj = getattr(schema, schema_name)(**record)
|
|
db.session.add(obj)
|
|
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def db(monkeypatch):
|
|
monkeypatch.setattr("ttfrog.db.manager.database", MagicMock(return_value=""))
|
|
monkeypatch.setenv("DATABASE_URL", "sqlite:///:memory:")
|
|
monkeypatch.setenv("DEBUG", "1")
|
|
_db.init()
|
|
yield _db
|
|
_db.metadata.drop_all(bind=_db.engine)
|
|
|
|
|
|
@pytest.fixture
|
|
def bootstrap(db):
|
|
with db.transaction():
|
|
# ancestries
|
|
human = schema.Ancestry("human")
|
|
|
|
tiefling = schema.Ancestry("tiefling")
|
|
tiefling.add_modifier(schema.Modifier("Ability Score Increase", target="intelligence", relative_value=1))
|
|
tiefling.add_modifier(schema.Modifier("Ability Score Increase", target="charisma", relative_value=2))
|
|
|
|
# ancestry traits
|
|
darkvision = schema.AncestryTrait("Darkvision")
|
|
darkvision.add_modifier(schema.Modifier("Darkvision", target="vision_in_darkness", absolute_value=120))
|
|
tiefling.add_trait(darkvision)
|
|
|
|
dragonborn = schema.Ancestry("dragonborn")
|
|
dragonborn.add_trait(darkvision)
|
|
|
|
db.add_or_update([human, dragonborn, tiefling])
|
|
|
|
# classes
|
|
fighting_style = schema.ClassAttribute("Fighting Style")
|
|
fighting_style.add_option(name="Archery")
|
|
fighting_style.add_option(name="Defense")
|
|
db.add_or_update(fighting_style)
|
|
|
|
fighter = schema.CharacterClass("fighter", hit_dice="1d10", hit_dice_stat="CON")
|
|
fighter.add_attribute(fighting_style, level=2)
|
|
|
|
rogue = schema.CharacterClass("rogue", hit_dice="1d8", hit_dice_stat="DEX")
|
|
db.add_or_update([rogue, fighter])
|
|
|
|
# characters
|
|
foo = schema.Character("Foo", ancestry=tiefling, _intelligence=14)
|
|
foo.add_class(fighter, level=2)
|
|
foo.add_class(rogue, level=3)
|
|
|
|
bar = schema.Character("Bar", ancestry=human)
|
|
|
|
# persist all the records we've created
|
|
db.add_or_update([foo, bar])
|