71 lines
2.5 KiB
Python
71 lines
2.5 KiB
Python
from ttfrog.db import schema
|
|
|
|
|
|
def test_create_character(db, classes_factory, ancestries_factory):
|
|
with db.transaction():
|
|
# load the fixtures so they are bound to the current session
|
|
classes = classes_factory()
|
|
ancestries = ancestries_factory()
|
|
darkvision = db.session.query(schema.AncestryTrait).filter_by(name="Darkvision")[0]
|
|
|
|
# create a human character (the default)
|
|
char = schema.Character(name="Test Character")
|
|
db.add(char)
|
|
assert char.id == 1
|
|
assert char.armor_class == 10
|
|
assert char.name == "Test Character"
|
|
assert char.ancestry.name == "human"
|
|
assert darkvision not in char.traits
|
|
|
|
# switch ancestry to tiefling
|
|
char.ancestry = ancestries["tiefling"]
|
|
db.add(char)
|
|
char = db.session.get(schema.Character, 1)
|
|
assert char.ancestry.name == "tiefling"
|
|
assert darkvision in char.traits
|
|
|
|
# assign a class and level
|
|
char.add_class(classes["fighter"], level=1)
|
|
db.add(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 = char.classes["fighter"].attributes_by_level[2]["Fighting Style"]
|
|
assert char.add_class_attribute(fighting_style, fighting_style.options[0]) is False
|
|
db.add(char)
|
|
assert char.class_attributes == {}
|
|
|
|
# level up
|
|
char.add_class(classes["fighter"], level=2)
|
|
db.add(char)
|
|
assert char.levels == {"fighter": 2}
|
|
assert char.level == 2
|
|
|
|
# Assign the fighting style
|
|
assert char.add_class_attribute(fighting_style, fighting_style.options[0])
|
|
db.add(char)
|
|
assert char.class_attributes[fighting_style.name] == fighting_style.options[0]
|
|
|
|
# classes
|
|
char.add_class(classes["rogue"], level=1)
|
|
db.add(char)
|
|
assert char.level == 3
|
|
assert char.levels == {"fighter": 2, "rogue": 1}
|
|
|
|
# remove a class
|
|
char.remove_class(classes["rogue"])
|
|
db.add(char)
|
|
assert char.levels == {"fighter": 2}
|
|
assert char.level == 2
|
|
|
|
# remove all remaining classes
|
|
char.remove_class(classes["fighter"])
|
|
db.add(char)
|
|
|
|
# ensure we're not persisting any orphan records in the map tables
|
|
dump = db.dump()
|
|
assert dump["class_map"] == []
|
|
assert dump["character_class_attribute_map"] == []
|