def test_count_messages(self):
u1_name = 'john'
u1_email = '*****@*****.**'
u2_name = 'alex'
u2_email = '*****@*****.**'
u1 = User(username=u1_name, email=u1_email, password='******')
u2 = User(username=u2_name, email=u2_email, password='******')
db.session.add(u1)
db.session.add(u2)
db.session.commit()
status = 'Accepted'
time = datetime.datetime.utcnow()
m1 = Message(first_username=u1_name,second_username=u2_name,chat='Hello',timestamp=time,chat_by=u1_name)
m2 = Message(first_username=u1_name,second_username=u2_name,chat='Hello',timestamp=time,chat_by=u1_name)
m3 = Message(first_username=u2_name,second_username=u1_name,chat='Hello',timestamp=time,chat_by=u2_name)
db.session.add(m1)
db.session.add(m2)
db.session.add(m3)
db.session.commit()
self.assertEqual(Message.count_messages(u1_name,u2_name), 3)
def test_count_messages(self):
u1_name = 'john'
u1_email = '*****@*****.**'
u2_name = 'alex'
u2_email = '*****@*****.**'
u1 = User(username=u1_name, email=u1_email, password='******')
u2 = User(username=u2_name, email=u2_email, password='******')
db.session.add(u1)
db.session.add(u2)
db.session.commit()
status = 'Accepted'
time = datetime.datetime.utcnow()
message_id = random.randint(1000000, 9999999)
m1 = Message(id=random.randint(1000000, 9999999),
first_username=u1_name,
second_username=u2_name,
chat='Hello',
timestamp=time,
chat_by=u1_name)
m2 = Message(id=random.randint(1000000, 9999999),
first_username=u1_name,
second_username=u2_name,
chat='Hello',
timestamp=time,
chat_by=u1_name)
m3 = Message(id=random.randint(1000000, 9999999),
first_username=u2_name,
second_username=u1_name,
chat='Hello',
timestamp=time,
chat_by=u2_name)
db.session.add(m1)
db.session.add(m2)
db.session.add(m3)
db.session.commit()
self.assertEqual(Message.count_messages(u1_name, u2_name), 3)
