def test_repr():
""" Test __repr__. """
u = User(api_key=3)
assert u.__repr__() == '<User None, api_key: 3>'
db.session.add(u)
db.session.commit()
assert u.__repr__() == '<User %d, api_key: 3>' % u.id
def test_repr() -> None:
channel = Channel(name='channelName', password='******')
assert channel.__repr__() == "Channel(name='channelName')"
user = User(username='******',
password='******',
email='*****@*****.**')
assert user.__repr__() == "User(name='userName')"
with app.app_context():
db.drop_all()
db.create_all()
db.session.add(user)
db.session.add(channel)
channel_id = Channel.query.first().id
user_id = User.query.first().id
channel_allow_list = ChannelAllowList(user_id=user_id,
channel_id=channel_id,
user_role=UserRole.ADMIN.value)
assert channel_allow_list.__repr__() == \
f'ChannelAllowList(channel_id=1, user_id=1, user_role={UserRole.ADMIN.value})'
time = datetime.utcnow()
message = Message(content='content',
author_id=user_id,
target_channel=channel_id,
time=time)
assert message.__repr__(
) == f"Message(author_id=1, target_channel=1, time={time})"
def test_user_repr(self):
user = User(username='******',
password='******',
email='*****@*****.**')
self.assertTrue(user.__repr__() == "User 'janedoe'")
with self.assertRaises(AttributeError):
user.password
def test_repr(self):
user1 = User(first_name='John',
last_name='Doe',
username='******',
email='*****@*****.**',
password='******')
self.assertEqual(user1.__repr__(), '<User [email protected]>')
def test_Users_and_Posts(self):
import datetime
# create user 1
u = User(nickname='john', email='*****@*****.**')
db.session.add(u)
db.session.commit()
# create user 2
u = User(nickname='susan', email='*****@*****.**')
db.session.add(u)
db.session.commit()
users = User.query.all()
# Access everything
self.assertEqual(users.__repr__(), "[<User 'john'>, <User 'susan'>]")
# -------------- CHECK JOHN ---------------------------
# Access john's id
self.assertEqual(users[0].id, 1, "john's id should be 1")
# Access john's nickname
self.assertEqual(users[0].nickname, 'john', 'john should be john')
# Access john's email
self.assertEqual(users[0].email, '*****@*****.**',
'john should be john')
# -------------- CHECK SUSAN ---------------------------
# Access susan's id
self.assertEqual(users[1].id, 2, "susan's id should be 1")
# Access susan's nickname
self.assertEqual(users[1].nickname, 'susan', 'susan should be susan')
# Access susan's email
self.assertEqual(users[1].email, '*****@*****.**',
'susan should be susan')
# Get users with id
u = User.query.get(1)
self.assertEqual(u.__repr__(), "<User 'john'>")
susan = User.query.get(2)
self.assertEqual(susan.__repr__(), "<User 'susan'>")
# Get all posts from Susan
post = Post(body='my first post!',
timestamp=datetime.datetime.utcnow(),
author=susan)
db.session.add(post)
db.session.commit()
posts = susan.posts.all()
self.assertEqual(posts.__repr__(), "[<Post 'my first post!'>]")
# obtain author of each post
for post in posts:
self.assertEqual(post.id, 1)
self.assertEqual(post.author.nickname, 'susan')
self.assertEqual(post.body, "my first post!")
pass
# get all users in reverse alphabetical order
nicknames_desc = User.query.order_by('nickname desc').all()
expected = "[<User 'susan'>, <User 'john'>]"
self.assertEqual(nicknames_desc.__repr__(), expected)
print('I am testing User_and_Posts')
def test_email_username_id_repr_tablename(self):
u1 = User(email='*****@*****.**',username='******',password='******')
db.session.add(u1)
db.session.commit()
self.assertEqual(u1.email,'*****@*****.**')
self.assertEqual(u1.username,'test')
self.assertEqual(u1.id,1)
self.assertEqual(u1.__repr__(),'<User test>')
self.assertEqual(u1.__tablename__, 'users')
def test_Users_and_Posts(self):
import datetime
# create user 1
u = User(nickname='john', email='*****@*****.**')
db.session.add(u)
db.session.commit()
# create user 2
u = User(nickname='susan', email='*****@*****.**')
db.session.add(u)
db.session.commit()
users = User.query.all()
# Access everything
self.assertEqual(users.__repr__(), "[<User 'john'>, <User 'susan'>]")
# -------------- CHECK JOHN ---------------------------
# Access john's id
self.assertEqual(users[0].id, 1, "john's id should be 1")
# Access john's nickname
self.assertEqual(users[0].nickname, 'john', 'john should be john')
# Access john's email
self.assertEqual(users[0].email, '*****@*****.**', 'john should be john')
# -------------- CHECK SUSAN ---------------------------
# Access susan's id
self.assertEqual(users[1].id, 2, "susan's id should be 1")
# Access susan's nickname
self.assertEqual(users[1].nickname, 'susan', 'susan should be susan')
# Access susan's email
self.assertEqual(users[1].email, '*****@*****.**', 'susan should be susan')
# Get users with id
u = User.query.get(1)
self.assertEqual(u.__repr__(), "<User 'john'>")
susan = User.query.get(2)
self.assertEqual(susan.__repr__(), "<User 'susan'>")
# Get all posts from Susan
post = Post(body='my first post!', timestamp=datetime.datetime.utcnow(), author=susan)
db.session.add(post)
db.session.commit()
posts = susan.posts.all()
self.assertEqual(posts.__repr__(), "[<Post 'my first post!'>]")
# obtain author of each post
for post in posts:
self.assertEqual(post.id, 1)
self.assertEqual(post.author.nickname, 'susan')
self.assertEqual(post.body, "my first post!")
pass
# get all users in reverse alphabetical order
nicknames_desc = User.query.order_by('nickname desc').all()
expected = "[<User 'susan'>, <User 'john'>]"
self.assertEqual(nicknames_desc.__repr__(), expected)
print('I am testing User_and_Posts')
def test_user(self):
new_user = User(username="******", two_factor="1231231234")
assert new_user.username == "new_user"
assert new_user.two_factor == "1231231234"
assert new_user.password is None
new_user.set_password("password")
assert new_user.password != "password"
assert new_user.check_password("password") is True
assert new_user.__repr__() == "<User new_user>"
def sign_up():
data = MultiDict(mapping=request.json)
inputs = SignUpForm(data, csrf_enabled=False)
if not inputs.validate():
return bad_request_error(inputs.errors)
else:
data = request.get_json()
firstname = data['firstname']
lastname = data['lastname']
email = data['email']
password = data['password']
user = User(firstname, lastname, email, password)
db.session.add(user)
db.session.commit()
response = jsonify(user.__repr__())
response.status_code = 201
return response
def signup():
data = MultiDict(mapping = request.json)
inputs = SignUpForm(data , csrf_enabled=False)
if not inputs.validate():
return bad_request_error(inputs.errors)
else:
data = request.get_json()
firstname = data['firstname']
lastname = data['lastname']
email = data['email']
password = data['password']
user = User(firstname , lastname , email , password)
db.session.add(user)
db.session.commit()
response = jsonify(user.__repr__())
response.status_code = 201
return response
def test_relationships(self):
food1 = FoodItem(name='food1',
img_url="",
nf_calories=Decimal(1),
nf_total_fat=Decimal(2),
nf_cholesterol=Decimal(3),
nf_saturated_fat=Decimal(4),
nf_sodium=Decimal(5),
nf_total_carbohydrate=Decimal(6),
nf_dietary_fiber=Decimal(7),
nf_sugars=Decimal(8),
nf_protein=Decimal(9))
food2 = FoodItem(name='food2',
img_url="",
nf_calories=Decimal(11),
nf_total_fat=Decimal(12),
nf_cholesterol=Decimal(13),
nf_saturated_fat=Decimal(14),
nf_sodium=Decimal(15),
nf_total_carbohydrate=Decimal(16),
nf_dietary_fiber=Decimal(17),
nf_sugars=Decimal(18),
nf_protein=Decimal(19))
cart1 = Cart()
cart2 = Cart()
user1 = User(email="*****@*****.**",username="******",password="******")
user2 = User(email="*****@*****.**",username="******",password="******")
food1.cart = cart1
cart1.user = user1
food2.cart = cart2
cart2.user = user2
# check relationships
self.assertTrue(food1.cart == cart1)
self.assertTrue(cart1.user == user1)
self.assertTrue(food2.cart == cart2)
self.assertTrue(cart2.user == user2)
# test relationships are showed in __repr__
self.assertTrue(food1.__repr__() == 'food1 in Cart of <User one>')
self.assertTrue(cart1.__repr__() == 'Cart of <User one>')
self.assertTrue(user1.__repr__() == '<User one>')
def test_user_creation(self):
u = User(username='******')
u.set_password('cat')
self.assertTrue(u.__repr__()) == 'User testuser'
def test_repr_email(app):
obj = User(email_dict)
assert obj.__repr__() == '<User: {}>'.format(email_dict['email'])
def test_repr_email(app):
obj = User(email_dict)
assert obj.__repr__() == '<User: {}>'.format(email_dict['email'])
