Пример #1
0
    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')
Пример #2
0
    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
Пример #3
0
    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