/
test_message_model.py
142 lines (117 loc) · 3.79 KB
/
test_message_model.py
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
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
"""Message View tests."""
# run these tests like:
#
# FLASK_ENV=production python -m unittest test_message_views.py
import os
from unittest import TestCase
from sqlalchemy import exc
from models import db, User, Message, Follows, Likes
# 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, CURR_USER_KEY
# 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
db.create_all()
# Don't have WTForms use CSRF at all, since it's a pain to test
app.config['WTF_CSRF_ENABLED'] = False
class MessageModelTestCase(TestCase):
"""Test views for messages."""
def setUp(self):
"""Create test client, add sample data."""
User.query.delete()
Message.query.delete()
Follows.query.delete()
self.client = app.test_client()
def test_message_model(self):
"""Does basic model work?"""
u = User(
email="test@test.com",
username="testuser",
password="HASHED_PASSWORD"
)
db.session.add(u)
db.session.commit()
m = Message(
text="warblewarble",
user_id=u.id
)
db.session.add(m)
db.session.commit()
self.assertEqual(len(u.messages), 1)
self.assertEqual(m.text,"warblewarble")
# def test_empty_message_model(self):
# u2 = User(
# email="test2@test.com",
# username="test2user",
# password="HASHED_PASSWORD"
# )
# db.session.add(u2)
# db.session.commit()
# m2 = Message(
# text=None,
# user_id=u2.id
# )
# with self.assertRaises(exc.IntegrityError):
# db.session.add(m2)
# db.session.commit()
# self.assertEqual(len(u2.messages), 0)
def test_message_like_model(self):
"""Does basic model work?"""
u = User(
email="test@test.com",
username="testuser",
password="HASHED_PASSWORD"
)
db.session.add(u)
db.session.commit()
m = Message(
text="warblewarble",
user_id=u.id
)
db.session.add(m)
db.session.commit()
l = Likes(user_that_liked=u.id, message_liked=m.id)
db.session.add(l)
db.session.commit()
self.assertEqual(len(u.likes), 1)
def test_delete_messages_likes(self):
"""test if message got deleted by the user and it deletes from likes table"""
u = User(
email="test@test.com",
username="testuser",
password="HASHED_PASSWORD"
)
db.session.add(u)
db.session.commit()
u2 = User(
email="test@test2.com",
username="testuser2",
password="HASHED_PASSWORD"
)
db.session.add(u2)
db.session.commit()
follow = Follows(
user_being_followed_id=u.id,
user_following_id=u2.id
)
db.session.add(follow)
db.session.commit()
m = Message(
text="warblewarble",
user_id=u.id
)
db.session.add(m)
db.session.commit()
l = Likes(user_that_liked=u2.id, message_liked=m.id)
db.session.add(l)
db.session.commit()
self.assertEqual(len(u2.likes), 1)
db.session.delete(m)
db.session.commit()
self.assertEqual(len(u2.likes), 0)
self.assertEqual(len(u.messages), 0)