189 lines
6.8 KiB
Python
189 lines
6.8 KiB
Python
import json
|
|
|
|
from ttfrog.db import schema
|
|
|
|
|
|
def test_manage_character(db, bootstrap):
|
|
with db.transaction():
|
|
darkvision = db.AncestryTrait.filter_by(name="Darkvision").one()
|
|
human = db.Ancestry.filter_by(name="human").one()
|
|
|
|
# create a human character (the default)
|
|
char = schema.Character(name="Test Character", ancestry=human)
|
|
db.add_or_update(char)
|
|
assert char.id == 3
|
|
assert char.name == "Test Character"
|
|
assert char.ancestry.name == "human"
|
|
assert char.armor_class == 10
|
|
assert char.hit_points == 10
|
|
assert char.strength == 10
|
|
assert char.dexterity == 10
|
|
assert char.constitution == 10
|
|
assert char.intelligence == 10
|
|
assert char.wisdom == 10
|
|
assert char.charisma == 10
|
|
assert darkvision not in char.traits
|
|
|
|
# switch ancestry to tiefling
|
|
tiefling = db.Ancestry.filter_by(name="tiefling").one()
|
|
char.ancestry = tiefling
|
|
db.add_or_update(char)
|
|
char = db.session.get(schema.Character, char.id)
|
|
assert char.ancestry_id == tiefling.id
|
|
assert char.ancestry.name == "tiefling"
|
|
assert darkvision in char.traits
|
|
|
|
# switch ancestry to dragonborn and assert darkvision persists
|
|
char.ancestry = db.Ancestry.filter_by(name="dragonborn").one()
|
|
db.add_or_update(char)
|
|
assert darkvision in char.traits
|
|
|
|
# switch ancestry to human and assert darkvision is removed
|
|
char.ancestry = human
|
|
db.add_or_update(char)
|
|
assert darkvision not in char.traits
|
|
|
|
fighter = db.CharacterClass.filter_by(name="fighter").one()
|
|
rogue = db.CharacterClass.filter_by(name="rogue").one()
|
|
|
|
# assign a class and level
|
|
char.add_class(fighter, level=1)
|
|
db.add_or_update(char)
|
|
assert char.levels == {"fighter": 1}
|
|
assert char.level == 1
|
|
assert char.class_attributes == {}
|
|
|
|
# 'fighting style' is available, but not at this level
|
|
fighting_style = fighter.attributes_by_level[2]["Fighting Style"]
|
|
assert char.add_class_attribute(fighting_style, fighting_style.options[0]) is False
|
|
db.add_or_update(char)
|
|
assert char.class_attributes == {}
|
|
|
|
# level up
|
|
char.add_class(fighter, level=2)
|
|
db.add_or_update(char)
|
|
assert char.levels == {"fighter": 2}
|
|
assert char.level == 2
|
|
|
|
# Assert the fighting style is added automatically and idempotent...ly?
|
|
assert char.class_attributes[fighting_style.name] == fighting_style.options[0]
|
|
assert char.add_class_attribute(fighting_style, fighting_style.options[0]) is True
|
|
db.add_or_update(char)
|
|
|
|
# classes
|
|
char.add_class(rogue, level=1)
|
|
db.add_or_update(char)
|
|
assert char.level == 3
|
|
assert char.levels == {"fighter": 2, "rogue": 1}
|
|
|
|
# remove a class
|
|
char.remove_class(rogue)
|
|
db.add_or_update(char)
|
|
assert char.levels == {"fighter": 2}
|
|
assert char.level == 2
|
|
|
|
# remove remaining class by setting level to zero
|
|
char.add_class(fighter, level=0)
|
|
db.add_or_update(char)
|
|
assert char.levels == {}
|
|
assert char.class_attributes == {}
|
|
|
|
# ensure we're not persisting any orphan records in the map tables
|
|
dump = json.loads(db.dump())
|
|
assert not [m for m in dump["character_class_attribute_map"] if m["character_id"] == char.id]
|
|
assert not [m for m in dump["class_map"] if m["character_id"] == char.id]
|
|
|
|
|
|
def test_ancestries(db):
|
|
with db.transaction():
|
|
# create the Pygmy Orc ancestry
|
|
porc = schema.Ancestry(
|
|
name="Pygmy Orc",
|
|
size="Small",
|
|
walk_speed=25,
|
|
)
|
|
db.add_or_update(porc)
|
|
assert porc.name == "Pygmy Orc"
|
|
assert porc.creature_type == "humanoid"
|
|
assert porc.size == "Small"
|
|
assert porc.speed == 25
|
|
|
|
# create the Relentless Endurance trait and add it to the Orc
|
|
endurance = schema.AncestryTrait(name="Relentless Endurance")
|
|
db.add_or_update(endurance)
|
|
porc.add_trait(endurance, level=1)
|
|
db.add_or_update(porc)
|
|
assert endurance in porc.traits
|
|
|
|
# add a +1 STR modifier
|
|
str_plus_one = schema.Modifier(
|
|
name="STR+1 (Pygmy Orc)",
|
|
target="strength",
|
|
relative_value=1,
|
|
description="Your Strength score is increased by 1.",
|
|
)
|
|
assert porc.add_modifier(str_plus_one) is True
|
|
assert porc.add_modifier(str_plus_one) is False # test idempotency
|
|
assert str_plus_one in porc.modifiers["strength"]
|
|
|
|
# now create an orc character and assert it gets traits and modifiers
|
|
grognak = schema.Character(name="Grognak the Mighty", ancestry=porc)
|
|
db.add_or_update(grognak)
|
|
assert endurance in grognak.traits
|
|
|
|
# verify the strength bonus is applied
|
|
assert grognak.strength.base == 10
|
|
assert str_plus_one in grognak.modifiers["strength"]
|
|
assert grognak.strength == 11
|
|
|
|
|
|
def test_modifiers(db, bootstrap):
|
|
with db.transaction():
|
|
human = db.Ancestry.filter_by(name="human").one()
|
|
tiefling = db.Ancestry.filter_by(name="tiefling").one()
|
|
|
|
# no modifiers; speed is ancestry speed
|
|
carl = schema.Character(name="Carl", ancestry=tiefling)
|
|
marx = schema.Character(name="Marx", ancestry=human)
|
|
db.add_or_update([carl, marx])
|
|
assert carl.speed == carl.ancestry.speed == 30
|
|
|
|
cold = schema.Modifier(target="speed", relative_value=-10, name="Cold")
|
|
hasted = schema.Modifier(target="speed", multiply_value=2.0, name="Hasted")
|
|
slowed = schema.Modifier(target="speed", multiply_value=0.5, name="Slowed")
|
|
restrained = schema.Modifier(target="speed", absolute_value=0, name="Restrained")
|
|
reduced = schema.Modifier(target="size", new_value="Tiny", name="Reduced")
|
|
|
|
# reduce speed by 10
|
|
assert carl.add_modifier(cold)
|
|
assert carl.speed == 20
|
|
|
|
# make sure modifiers only apply to carl. Carl is having a bad day.
|
|
assert marx.speed == 30
|
|
|
|
# speed is doubled
|
|
assert carl.remove_modifier(cold)
|
|
assert carl.add_modifier(hasted)
|
|
assert carl.speed == 60
|
|
|
|
# speed is halved
|
|
assert carl.remove_modifier(hasted)
|
|
assert carl.add_modifier(slowed)
|
|
assert carl.speed == 15
|
|
|
|
# speed is 0
|
|
assert carl.add_modifier(restrained)
|
|
assert carl.speed == 0
|
|
|
|
# no longer restrained, but still slowed
|
|
assert carl.remove_modifier(restrained)
|
|
assert carl.speed == 15
|
|
|
|
# back to normal
|
|
assert carl.remove_modifier(slowed)
|
|
assert carl.speed == carl.ancestry.speed
|
|
|
|
# modifiers can modify string values too
|
|
assert carl.add_modifier(reduced)
|
|
assert carl.size == "Tiny"
|