rithm-warbler/test_user_model.py at main · mtbocim/rithm-warbler · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
"""User model tests."""

# run these tests like:
#
#    python -m unittest test_user_model.py


import os
from unittest import TestCase

from models import db, User, Message, Follows, connect_db

# BEFORE we import our app, let's set an environmental variable
# to use a different database for tests (we need to do this
# before we import our app, since that will have already
# connected to the database

os.environ['DATABASE_URL'] = "postgresql:///warbler_test"

# Now we can import app

from app import app

# Create our tables (we do this here, so we only create the tables
# once for all tests --- in each test, we'll delete the data
# and create fresh new clean test data

connect_db(app)

db.drop_all()
db.create_all()


class UserModelTestCase(TestCase):
    """Test for User model."""

    def setUp(self):
        """Create test users and add seed data."""
        User.query.delete()

        u1 = User.signup("u1", "u1@email.com", "password", None)
        u2 = User.signup("u2", "u2@email.com", "password", None)

        # db.session.flush()
        db.session.commit()

        self.u1_id = u1.id
        self.u2_id = u2.id

        self.client = app.test_client()

    def tearDown(self):
        """Clean up"""

        db.session.rollback()

    def test_user_model(self):
        """Test create new user"""

        u = User.signup("u3", "u3@email.com", "password", None)
        db.session.add(u)
        db.session.commit()

        self.assertEqual(u.email, "u3@email.com")
        self.assertEqual(len(u.liked_messages), 0)

    def test_user_repr(self):
        """Test model repr"""

        u1 = User.query.get(self.u1_id)
        self.assertEqual(repr(u1), f"<User #{self.u1_id}: u1, u1@email.com>")

    def test_user_login(self):
        """Test user login authenication """

        u1 = User.authenticate("u1", "password")
        self.assertEqual(u1.email, "u1@email.com")

        u_bad = User.authenticate("u1", "wrong")
        self.assertEqual(u_bad, False)

        u_bad = User.authenticate("made_up", "pleaselogin?")
        self.assertEqual(u_bad, False)

    def test_user_is_followed_by(self):
        """Test if user is followed by another"""

        u = User.signup("u3", "u3@email.com", "password", None)
        db.session.add(u)
        db.session.commit()

        self.assertEqual(len(u.followers), 0)

        u1 = User.query.get_or_404(self.u1_id)
        u2 = User.query.get_or_404(self.u2_id)
        u1.following.append(u)
        db.session.commit()

        self.assertEqual(len(u.followers), 1)
        self.assertEqual(u.is_followed_by(u1), True)
        self.assertEqual(u.is_followed_by(u2), False)

    def test_user_is_following(self):
        """Test user is following someone else"""

        u = User.signup("u3", "u3@email.com", "password", None)
        db.session.add(u)
        db.session.commit()

        u1 = User.query.get_or_404(self.u1_id)
        u2 = User.query.get_or_404(self.u2_id)

        self.assertEqual(u.is_following(u1), False)

        u.following.append(u1)
        db.session.commit()

        self.assertEqual(u.is_following(u1), True)
        self.assertEqual(u.is_following(u2), False)