def main(): db = Database(mode='PROD', echo=True) if not db.has_tables(): db.make_tables() with open('.smurfs.txt') as f: for line in f: fetch_and_store_summoner(line.strip(), db) session = db.SessionMaker() print(session.query(GameStats, GameStats.champion).all())
class TestObjectInitialization(unittest.TestCase): @classmethod def setUpClass(self): self.db = Database() self.db.make_tables() @classmethod def tearDownClass(self): self.db.destroy_tables() def tearDown(self): self.db.truncate_tables() def test_when_game_is_made(self): session = self.db.SessionMaker() def query_for_game(session): return session.query(Game).first() expected = add_a_game(session) observed = query_for_game(session) session.close() assert expected is observed, 'Observed:\t{0}\nExpected:\t{1}'.format( observed, expected) def test_when_game_stats_is_made(self): session = self.db.SessionMaker() def query_for_game_stats(): return session.query(GameStats).filter_by( summoner_name='Bjerson').first() expected = add_game_stats(session) observed = query_for_game_stats() session.close() assert expected is observed, 'Observed:\t{0}\nExpected:\t{1}'.format( observed, expected) def test_when_summoner_is_made(self): session = self.db.SessionMaker() def query_for_summoner(): return session.query(Summoner).first() expected = add_a_summoner(session) observed = query_for_summoner() session.close() assert expected is observed, 'Observed:\t{0}\nExpected:\t{1}'.format( observed, expected) def test_when_summoner_name_is_made(self): session = self.db.SessionMaker() def query_for_summoner_name(): return session.query(SummonerName).first() expected = add_a_summoner_name(session) observed = query_for_summoner_name() session.close() assert expected is observed, 'Observed:\t{0}\nExpected:\t{1}'.format( observed, expected) def test_when_they_work_together(self): pass