def test_digest_task(client, database, smtp_server):
# Here because it requires the app context to be set
from monolith.task import send_digest
no_stories = 'Start following your favourite users to get ' \
'periodic digests.'
send_digest.apply()
for msg in smtp_server.messages:
assert msg.get_payload() == no_stories
now = dt.datetime.now()
s1 = Story(author_id=2,
text='Story 1',
likes=42,
dislikes=69,
is_draft=False,
deleted=False)
s1.dice_set = ['dice', 'set']
s1.date = now - dt.timedelta(days=1)
database.session.add(s1)
s2 = Story(author_id=2,
text='Story 2',
likes=42,
dislikes=69,
is_draft=False,
deleted=False)
s2.dice_set = ['dice', 'set']
s2.date = now - dt.timedelta(days=60)
database.session.add(s2)
s3 = Story(author_id=3,
text='Story 3',
likes=42,
dislikes=69,
is_draft=False,
deleted=False)
s3.dice_set = ['dice', 'set']
s3.date = now - dt.timedelta(days=4)
database.session.add(s3)
u1 = User.query.get(1)
u2 = User.query.get(2)
u3 = User.query.get(3)
u4 = User.query.get(4)
u1.follows.append(u2)
u2.follows.append(u3)
u3.follows.append(u4)
database.session.commit()
smtp_server.reset()
send_digest.apply()
for msg in smtp_server.messages:
u = User.query.filter_by(email=msg['To']).one()
digest = digests.build_user_digest(u, now - dt.timedelta(weeks=4), now)
payload = msg.get_payload().replace('\r\n', '\n').strip()
digest = digest.get_payload().replace('\r\n', '\n').strip()
assert payload == digest
def test_get_random_story(client, database, templates, story_actions):
example = Story()
example.text = 'very not recent story (months/years ago)'
example.likes = 0
example.author_id = 1
example.date = dt.datetime(2019, 9, 5)
example.is_draft = False
example.deleted = False
example.dice_set = ['a', 'b', 'c']
database.session.add(example)
example = Story()
example.text = 'not recent story (yesterday)'
example.date = dt.datetime.now() - dt.timedelta(days=1)
example.likes = 0
example.author_id = 2
example.is_draft = False
example.deleted = False
example.dice_set = ['a', 'b', 'c']
database.session.add(example)
example = Story()
example.text = 'drafted story'
example.date = dt.datetime.now() - dt.timedelta(days=1)
example.likes = 0
example.author_id = 1
example.is_draft = True
example.deleted = False
example.dice_set = ['a', 'b', 'c']
database.session.add(example)
example = Story()
example.text = 'deleted story'
example.date = dt.datetime.now() - dt.timedelta(days=1)
example.likes = 0
example.author_id = 1
example.is_draft = False
example.deleted = True
example.dice_set = ['a', 'b', 'c']
database.session.add(example)
database.session.commit()
# story found
reply = story_actions.get_random_recent_story()
assert reply.status_code == 200
template_context = templates[-1]
id = template_context['story'].id
message = template_context['message']
assert id == 1 or id == 2
assert message == 'no stories today. Here is a random one:'
def test_random_recent_story(self):
# Random recent story as anonymous user
self.client.get('/stories/random', follow_redirects=True)
self.assert_template_used('story.html')
self.assertEqual(self.get_context_variable('story').text, 'Just another story')
# Login as Admin
payload = {'email': '*****@*****.**', 'password': '******'}
form = LoginForm(data=payload)
self.client.post('/users/login', data=form.data, follow_redirects=True)
# No recent stories
self.client.get('/stories/random', follow_redirects=True)
self.assert_template_used('stories.html')
self.assert_message_flashed('Oops, there are no recent stories by other users!')
# Create a new recent story by Admin2
example = Story()
example.text = 'This is a valid recent story'
example.date = datetime.datetime.now()
example.author_id = 2
example.figures = 'story#recent'
example.is_draft = False
db.session.add(example)
db.session.commit()
# Get the only recent story not written by Admin
response = self.client.get('/stories/random', follow_redirects=True)
self.assert_template_used('story.html')
self.assertEqual(self.get_context_variable('story').text, 'This is a valid recent story')
def init_database(database):
example = Story()
example.text = 'lorem ipsum dolor sit amet'
example.likes = 42
example.author_id = 1
example.date = dt.datetime(year=2018, month=12, day=1)
example.is_draft = False
example.deleted = False
example.dice_set = ['a', 'b', 'c']
database.session.add(example)
example = Story()
example.text = 'bird drink coffee baloon'
example.likes = 42
example.author_id = 1
example.date = dt.datetime(year=2019, month=1, day=1)
example.is_draft = False
example.deleted = False
example.dice_set = ['a', 'b', 'c']
database.session.add(example)
example = Story()
example.text = 'lorem Coffee dolor sit amet'
example.likes = 42
example.author_id = 1
example.date = dt.datetime(year=2019, month=3, day=12)
example.is_draft = False
example.deleted = False
example.dice_set = ['a', 'b', 'c']
database.session.add(example)
example = Story()
example.text = 'bird cofFee baloon amet'
example.likes = 42
example.author_id = 1
example.date = dt.datetime(year=2017, month=10, day=1)
example.is_draft = False
example.deleted = False
example.dice_set = ['a', 'b', 'c']
database.session.add(example)
database.session.commit()
def test_get_by_interest(client, auth, database, templates, story_actions):
auth.login()
example1 = Story()
example1.theme = 'halloween'
example1.text = 'Halloween story of test1 user :)'
example1.author_id = 2
example1.is_draft = False
example1.deleted = False
example1.dice_set = ['a', 'b', 'c']
database.session.add(example1)
database.session.commit()
example2 = Story()
example2.theme = 'xmas'
example2.text = 'Xmas story of test2 user :)'
example2.author_id = 3
example2.is_draft = False
example2.deleted = False
example2.dice_set = ['a', 'b', 'c']
database.session.add(example2)
database.session.commit()
example3 = Story()
example3.theme = 'xmas'
example3.theme = 'Old xmas story of test3 user :)'
example3.date = dt.datetime.now() - dt.timedelta(days=6)
example3.author_id = 4
example3.is_draft = False
example3.deleted = False
example3.dice_set = ['a', 'b', 'c']
database.session.add(example3)
database.session.commit()
reply = story_actions.get_all_stories(theme='xmas')
assert reply.status_code == 200
assert templates[-1]['stories'].all() == [example2]
def test_story_digest_format():
u = User()
u.username = '******'
u.email = '*****@*****.**'
u.firstname = 'Firstname'
u.lastname = 'Lastname'
date = dt.datetime.now()
s = Story()
s.author = u
s.text = 'Lorem ipsum dolor sit amet'
s.date = date
s.likes = 42
s.dislikes = 69
s.dice_set = ['lorem', 'ipsum', 'dolor', 'sit', 'amet']
s.deleted = False
s.is_draft = False
lines = digests.story_digest(s).split('\n')
assert lines[0] == 'author: Firstname Lastname (Username)'
assert lines[1] == 'roll: lorem, ipsum, dolor, sit, amet'
assert lines[2] == f'date: {date}'
assert lines[3] == 'story: Lorem ipsum dolor sit amet'
assert lines[4] == 'likes: 42 -- dislikes: 69'
def test_user_digest_format(client, database):
now = dt.datetime.now()
s1 = Story(author_id=2,
text='Story 1',
likes=42,
dislikes=69,
is_draft=False,
deleted=False)
s1.dice_set = ['dice', 'set']
s1.date = now - dt.timedelta(days=1)
database.session.add(s1)
s2 = Story(author_id=2,
text='Story 2',
date=now,
likes=42,
dislikes=69,
is_draft=False,
deleted=False)
s2.dice_set = ['dice', 'set']
s2.date = now
database.session.add(s2)
s3 = Story(author_id=2,
text='Story 3',
likes=42,
dislikes=69,
is_draft=False,
deleted=False)
s3.dice_set = ['dice', 'set']
s3.date = now - dt.timedelta(days=7)
database.session.add(s3)
s4 = Story(author_id=3,
text='Story 4',
likes=42,
dislikes=69,
is_draft=False,
deleted=False)
s4.dice_set = ['dice', 'set']
s4.date = now
database.session.add(s4)
s5 = Story(author_id=2,
text='Story 5',
likes=42,
dislikes=69,
is_draft=True,
deleted=False)
s5.dice_set = ['dice', 'set']
s5.date = now
database.session.add(s5)
s6 = Story(author_id=2,
text='Story 5',
likes=42,
dislikes=69,
is_draft=False,
deleted=True)
s6.dice_set = ['dice', 'set']
s6.date = now
database.session.add(s6)
u = User.query.get(1)
u.follows.append(User.query.get(2))
database.session.commit()
digest = digests.build_user_digest(u, now - dt.timedelta(days=2),
now + dt.timedelta(days=2)) \
.get_content().split('----------\n')
assert len(digest) == 2
assert digest[0] == digests.story_digest(s2)
assert digest[1] == digests.story_digest(s1)