def setUp(self): database.create_tables([Stranger, Talk]) self.stranger_0 = Stranger.create( invitation='foo', telegram_id=31416, ) self.stranger_1 = Stranger.create( invitation='bar', telegram_id=27183, ) self.stranger_2 = Stranger.create( invitation='baz', telegram_id=23571, ) self.stranger_3 = Stranger.create( invitation='zig', telegram_id=11317, ) self.stranger_4 = Stranger.create( invitation='zap', telegram_id=19232, ) self.talk_0 = Talk.create( partner1=self.stranger_0, partner1_sent=1000, partner2=self.stranger_1, partner2_sent=2000, searched_since=datetime.datetime(1980, 1, 1), begin=datetime.datetime(2000, 1, 1), ) self.talk_1 = Talk.create( partner1=self.stranger_2, partner2=self.stranger_3, searched_since=datetime.datetime(1990, 1, 1), begin=datetime.datetime(2000, 1, 2) ) self.talk_2 = Talk.create( partner1=self.stranger_2, partner2=self.stranger_4, searched_since=datetime.datetime(2000, 1, 1), begin=datetime.datetime(2000, 1, 3), end=datetime.datetime(2010, 1, 1), ) self.talk_3 = Talk.create( partner1=self.stranger_0, partner2=self.stranger_2, searched_since=datetime.datetime(2000, 1, 1), begin=datetime.datetime(2000, 1, 4), end=datetime.datetime(2010, 1, 2), ) self.talk_4 = Talk.create( partner1=self.stranger_0, partner2=self.stranger_3, searched_since=datetime.datetime(2000, 1, 1), begin=datetime.datetime(2000, 1, 5), end=datetime.datetime(2010, 1, 3), )
def setup_db(db_dict): for model_name, model_dicts in db_dict.items(): if model_name == 'strangers': for stranger_dict in model_dicts: Stranger.create(**stranger_dict) elif model_name == 'talks': for talk_dict in model_dicts: Talk.create(**talk_dict) else: raise AssertionError(f'Unknown model name: `{model_name}`')
def setUp(self): DATABASE.create_tables([Stranger, Talk]) self.stranger_0 = Stranger.create( invitation='foo', telegram_id=31416, ) self.stranger_1 = Stranger.create( invitation='bar', telegram_id=27183, ) self.stranger_2 = Stranger.create( invitation='baz', telegram_id=23571, ) self.stranger_3 = Stranger.create( invitation='zig', telegram_id=11317, ) self.stranger_4 = Stranger.create( invitation='zap', telegram_id=19232, ) self.talk_0 = Talk.create( partner1=self.stranger_0, partner1_sent=1000, partner2=self.stranger_1, partner2_sent=2000, searched_since=datetime.datetime(1980, 1, 1), begin=datetime.datetime(2000, 1, 1), ) self.talk_1 = Talk.create(partner1=self.stranger_2, partner2=self.stranger_3, searched_since=datetime.datetime(1990, 1, 1), begin=datetime.datetime(2000, 1, 2)) self.talk_2 = Talk.create( partner1=self.stranger_2, partner2=self.stranger_4, searched_since=datetime.datetime(2000, 1, 1), begin=datetime.datetime(2000, 1, 3), end=datetime.datetime(2010, 1, 1), ) self.talk_3 = Talk.create( partner1=self.stranger_0, partner2=self.stranger_2, searched_since=datetime.datetime(2000, 1, 1), begin=datetime.datetime(2000, 1, 4), end=datetime.datetime(2010, 1, 2), ) self.talk_4 = Talk.create( partner1=self.stranger_0, partner2=self.stranger_3, searched_since=datetime.datetime(2000, 1, 1), begin=datetime.datetime(2000, 1, 5), end=datetime.datetime(2010, 1, 3), )