def setUp(self):
# TODO Lots of redundancy here with main app and other tests. Can we
# abstract?
db.init(':memory:')
classes = PMDataObject.__subclasses__()
db.create_tables(classes)
for class_ in classes:
if class_.__name__ not in PMDataObject.deferred_relations:
PMDataObject.deferred_relations[class_.__name__] = class_
league = League.create(name="Band Battle")
season = Season.create(league=league,
start_year=2016,
end_year=2017,
is_current=True)
self.team1 = Team.create(name="Sex Bob-omb",
city="Toronto",
skill=90,
abbreviation="SBO",
league=league)
TeamStats.create(team=self.team1, season=season)
self.team2 = Team.create(name="The Clash at Demonhead",
city="New York",
skill=90,
abbreviation="TCD",
league=league)
TeamStats.create(team=self.team2, season=season)
def new_season(league):
"""Start a new season"""
old_season = league.current_season
if old_season is None:
new_season = Season.create(league=league,
start_year=2016,
end_year=2017,
is_current=True)
else:
old_season.is_current = False
old_season.save()
new_season = Season.create(league=league,
start_year=old_season.start_year + 1,
end_year=old_season.end_year + 1,
is_current=True)
for team in league.teams:
TeamStats.create(team=team, season=new_season)
def setUp(self):
db.init(':memory:')
classes = [Team, League, Season, TeamStats]
db.create_tables(classes)
for class_ in classes:
if class_.__name__ not in PMDataObject.deferred_relations:
PMDataObject.deferred_relations[class_.__name__] = class_
self.league = League.create(name="Band Battle")
self.season = Season.create(league=self.league,
start_year=2016,
end_year=2017,
is_current=True)
self.team = Team.create(name="Sex Bob-omb",
city="Toronto",
skill=90,
abbreviation="TOR",
league=self.league)
self.team_stats = TeamStats.create(team=self.team,
season=self.season)
