def test_unmatch(self): u1 = User(username="******") u2 = User(username="******") db.session.add(u1) db.session.add(u2) db.session.commit() m = u1.send_match_request(u2, 'date') db.session.add(m) db.session.commit() am = u2.accept_match(u1, 'date') db.session.add(am) db.session.commit() assert u1.is_matched(u2) assert u2.is_matched(u1) assert u1.is_matched(u2, 'date') assert u2.is_matched(u1, 'date') assert not u1.is_matched(u2, 'friend') assert not u2.is_matched(u1, 'friend') u1.unmatch(u2, 'date') assert not u1.is_matched(u2, 'date') assert not u2.is_matched(u1, 'date') m = u1.send_match_request(u2, 'friend') db.session.add(m) db.session.commit() u2.unmatch(u1, 'friend')
def test_pending_matches(self): u1 = User(username="******") u2 = User(username="******") u3 = User(username="******") u4 = User(username="******") db.session.add(u1) db.session.add(u2) db.session.add(u3) db.session.add(u4) db.session.commit() assert u1.get_pending_matches() is None assert u1.get_pending_matches('date') is None assert u1.get_pending_matches('friend') is None m = u1.send_match_request(u2, 'date') db.session.add(m) db.session.commit() m = u1.send_match_request(u3, 'date') db.session.add(m) db.session.commit() m = u1.send_match_request(u4, 'friend') db.session.add(m) db.session.commit() assert len(u1.get_pending_matches()) == 3 assert len(u1.get_pending_matches('date')) == 2 assert len(u1.get_pending_matches('friend')) == 1 m = u2.accept_match(u1, 'date') assert len(u1.get_pending_matches()) == 2 assert len(u1.get_pending_matches('date')) == 1 assert len(u1.get_pending_matches('friend')) == 1 u1.unmatch(u4, 'friend') assert len(u1.get_pending_matches()) == 1 assert len(u1.get_pending_matches('date')) == 1 assert u1.get_pending_matches('friend') is None assert u4.get_pending_matches() is None
def test_get_notifications(self): u1 = User(username="******") u2 = User(username="******") u3 = User(username="******") u4 = User(username="******") db.session.add(u1) db.session.add(u2) db.session.add(u3) db.session.add(u4) db.session.commit() assert u1.get_notifications() is None assert u2.get_notifications() is None m = u1.send_match_request(u2, 'friend') db.session.add(m) db.session.commit() assert u1.get_notifications() is None assert len(u2.get_notifications()) == 1 m = Message(to=u2, to_id = u2.id, from_id = u1.id, author=u1,content="This is the second message!", read=False) db.session.add(m) db.session.commit() assert u1.get_notifications() is None assert len(u2.get_notifications()) == 2 m.read = True db.session.commit() assert u1.get_notifications() is None assert len(u2.get_notifications()) == 1 m = u2.accept_match(u1, 'friend') assert u1.get_notifications() is None assert u2.get_notifications() is None