def test_get_by_id(self):
"""Get user by ID."""
user = User('foo', '*****@*****.**')
user.save()
retrieved = User.get_by_id(user.id)
assert retrieved == user
def test_favorite_an_article(self):
u1 = User('foo', '*****@*****.**')
u1.save()
p1 = UserProfile(u1)
p1.save()
article = Article(p1, 'title', 'some body', description='some')
article.save()
assert article.favourite(u1.profile)
assert article.is_favourite(u1.profile)
def test_check_password(self):
"""Check password."""
user = User.create(username='******',
email='*****@*****.**',
password='******')
assert user.check_password('foobarbaz123')
assert not user.check_password('barfoobaz')
def test_follow_user(self):
u1 = User('foo', '*****@*****.**')
u1.save()
u2 = User('foo1', '*****@*****.**')
u2.save()
p1 = UserProfile(u1)
p2 = UserProfile(u2)
p1.save()
p2.save()
p1.follow(p2)
assert p1.is_following(p2)
def test_unfavorite_an_article(self):
u1 = User('foo', '*****@*****.**')
u1.save()
p1 = UserProfile(u1)
p1.save()
u2 = User('foo1', '*****@*****.**')
u2.save()
p2 = UserProfile(u2)
p2.save()
article = Article(p1, 'title', 'some body', description='some')
article.save()
assert article.favourite(p1)
assert article.unfavourite(p1)
assert not article.is_favourite(p1)
def test_unfollow_self(self):
u1 = User('foo', '*****@*****.**')
u1.save()
p1 = UserProfile(u1)
assert not p1.unfollow(p1)
def test_password_is_nullable(self):
"""Test null password."""
user = User(username='******', email='*****@*****.**')
user.save()
assert user.password is None
def test_created_at_defaults_to_datetime(self):
"""Test creation date."""
user = User(username='******', email='*****@*****.**')
user.save()
assert bool(user.created_at)
assert isinstance(user.created_at, dt.datetime)
def jwt_identity(payload):
return User.get_by_id(payload)