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),
)
