phshare/tests/test_main.py

# -*- coding: utf-8 -*-
from flask import url_for

from phshare.extensions import db
from phshare.models import User, Photo, Comment, Notification, Tag
from tests.base import BaseTestCase


class MainTestCase(BaseTestCase):

    def test_index_page(self):
        response = self.client.get(url_for('main.index'))
        data = response.get_data(as_text=True)
        self.assertIn('Join Now', data)

        self.login()
        response = self.client.get(url_for('main.index'))
        data = response.get_data(as_text=True)
        self.assertNotIn('Join Now', data)
        self.assertIn('My Home', data)

    def test_explore_page(self):
        response = self.client.get(url_for('main.explore'))
        data = response.get_data(as_text=True)
        self.assertIn('Change', data)

    def test_search(self):
        response = self.client.get(url_for('main.search', q=''), follow_redirects=True)
        data = response.get_data(as_text=True)
        self.assertIn('Enter keyword about photo, user or tag.', data)

        response = self.client.get(url_for('main.search', q='normal'))
        data = response.get_data(as_text=True)
        self.assertNotIn('Enter keyword about photo, user or tag.', data)
        self.assertIn('No results.', data)

        response = self.client.get(url_for('main.search', q='normal', category='tag'))
        data = response.get_data(as_text=True)
        self.assertNotIn('Enter keyword about photo, user or tag.', data)
        self.assertIn('No results.', data)

        response = self.client.get(url_for('main.search', q='normal', category='user'))
        data = response.get_data(as_text=True)
        self.assertNotIn('Enter keyword about photo, user or tag.', data)
        self.assertNotIn('No results.', data)
        self.assertIn('Normal User', data)

    def test_show_notifications(self):
        user = User.query.get(2)
        notification1 = Notification(message='test 1', is_read=True, receiver=user)
        notification2 = Notification(message='test 2', is_read=False, receiver=user)
        db.session.add_all([notification1, notification2])
        db.session.commit()

        self.login()
        response = self.client.get(url_for('main.show_notifications'))
        data = response.get_data(as_text=True)
        self.assertIn('test 1', data)
        self.assertIn('test 2', data)

        response = self.client.get(url_for('main.show_notifications', filter='unread'))
        data = response.get_data(as_text=True)
        self.assertNotIn('test 1', data)
        self.assertIn('test 2', data)

    def test_read_notification(self):
        user = User.query.get(2)
        notification1 = Notification(message='test 1', receiver=user)
        notification2 = Notification(message='test 2', receiver=user)
        db.session.add_all([notification1, notification2])
        db.session.commit()

        self.login(email='admin@helloflask.com', password='123')
        response = self.client.post(url_for('main.read_notification', notification_id=1))
        self.assertEqual(response.status_code, 403)

        self.logout()
        self.login()

        response = self.client.post(url_for('main.read_notification', notification_id=1), follow_redirects=True)
        data = response.get_data(as_text=True)
        self.assertIn('Notification archived.', data)

        self.assertTrue(Notification.query.get(1).is_read)

    def test_read_all_notification(self):
        user = User.query.get(2)
        notification1 = Notification(message='test 1', receiver=user)
        notification2 = Notification(message='test 2', receiver=user)
        db.session.add_all([notification1, notification2])
        db.session.commit()

        self.login()

        response = self.client.post(url_for('main.read_all_notification'), follow_redirects=True)
        data = response.get_data(as_text=True)
        self.assertIn('All notifications archived.', data)

        self.assertTrue(Notification.query.get(1).is_read)
        self.assertTrue(Notification.query.get(2).is_read)

    def test_show_photo(self):
        response = self.client.get(url_for('main.show_photo', photo_id=1), follow_redirects=True)
        data = response.get_data(as_text=True)
        self.assertNotIn('Delete', data)
        self.assertIn('test tag', data)
        self.assertIn('test comment body', data)

        self.login(email='admin@helloflask.com', password='123')
        response = self.client.get(url_for('main.show_photo', photo_id=1), follow_redirects=True)
        data = response.get_data(as_text=True)
        self.assertIn('Delete', data)

    def test_photo_next(self):
        user = User.query.get(1)
        photo2 = Photo(filename='test.jpg', filename_s='test_s.jpg', filename_m='test_m.jpg',
                       description='Photo 2', author=user)
        photo3 = Photo(filename='test.jpg', filename_s='test_s.jpg', filename_m='test_m.jpg',
                       description='Photo 3', author=user)
        photo4 = Photo(filename='test.jpg', filename_s='test_s.jpg', filename_m='test_m.jpg',
                       description='Photo 4', author=user)
        db.session.add_all([photo2, photo3, photo4])
        db.session.commit()

        response = self.client.get(url_for('main.photo_next', photo_id=5), follow_redirects=True)
        data = response.get_data(as_text=True)
        self.assertIn('Photo 3', data)

        response = self.client.get(url_for('main.photo_next', photo_id=4), follow_redirects=True)
        data = response.get_data(as_text=True)
        self.assertIn('Photo 2', data)

        response = self.client.get(url_for('main.photo_next', photo_id=3), follow_redirects=True)
        data = response.get_data(as_text=True)
        self.assertIn('Photo 1', data)

        response = self.client.get(url_for('main.photo_next', photo_id=1), follow_redirects=True)
        data = response.get_data(as_text=True)
        self.assertIn('This is already the last one.', data)

    def test_photo_prev(self):
        user = User.query.get(1)
        photo2 = Photo(filename='test.jpg', filename_s='test_s.jpg', filename_m='test_m.jpg',
                       description='Photo 2', author=user)
        photo3 = Photo(filename='test.jpg', filename_s='test_s.jpg', filename_m='test_m.jpg',
                       description='Photo 3', author=user)
        photo4 = Photo(filename='test.jpg', filename_s='test_s.jpg', filename_m='test_m.jpg',
                       description='Photo 4', author=user)
        db.session.add_all([photo2, photo3, photo4])
        db.session.commit()

        response = self.client.get(url_for('main.photo_previous', photo_id=1), follow_redirects=True)
        data = response.get_data(as_text=True)
        self.assertIn('Photo 2', data)

        response = self.client.get(url_for('main.photo_previous', photo_id=3), follow_redirects=True)
        data = response.get_data(as_text=True)
        self.assertIn('Photo 3', data)

        response = self.client.get(url_for('main.photo_previous', photo_id=4), follow_redirects=True)
        data = response.get_data(as_text=True)
        self.assertIn('Photo 4', data)

        response = self.client.get(url_for('main.photo_previous', photo_id=5), follow_redirects=True)
        data = response.get_data(as_text=True)
        self.assertIn('This is already the first one.', data)

    def test_collect(self):
        photo = Photo(filename='test.jpg', filename_s='test_s.jpg', filename_m='test_m.jpg',
                      description='Photo 3', author=User.query.get(2))
        db.session.add(photo)
        db.session.commit()
        self.assertEqual(Photo.query.get(3).collectors, [])

        self.login()
        response = self.client.post(url_for('main.collect', photo_id=3), follow_redirects=True)
        data = response.get_data(as_text=True)
        self.assertIn('Photo collected.', data)

        self.assertEqual(Photo.query.get(3).collectors[0].collector.name, 'Normal User')

        response = self.client.post(url_for('main.collect', photo_id=3), follow_redirects=True)
        data = response.get_data(as_text=True)
        self.assertIn('Already collected.', data)

    def test_uncollect(self):
        self.login()
        self.client.post(url_for('main.collect', photo_id=1), follow_redirects=True)

        response = self.client.post(url_for('main.uncollect', photo_id=1), follow_redirects=True)
        data = response.get_data(as_text=True)
        self.assertIn('Photo uncollected.', data)

        response = self.client.post(url_for('main.uncollect', photo_id=1), follow_redirects=True)
        data = response.get_data(as_text=True)
        self.assertIn('Not collect yet.', data)

    def test_report_comment(self):
        self.assertEqual(Comment.query.get(1).flag, 0)

        self.login()
        response = self.client.post(url_for('main.report_comment', comment_id=1), follow_redirects=True)
        data = response.get_data(as_text=True)
        self.assertIn('Comment reported.', data)
        self.assertEqual(Comment.query.get(1).flag, 1)

    def test_report_photo(self):
        self.assertEqual(Photo.query.get(1).flag, 0)

        self.login()
        response = self.client.post(url_for('main.report_photo', photo_id=1), follow_redirects=True)
        data = response.get_data(as_text=True)
        self.assertIn('Photo reported.', data)
        self.assertEqual(Photo.query.get(1).flag, 1)

    def test_show_collectors(self):
        user = User.query.get(2)
        user.collect(Photo.query.get(1))
        response = self.client.get(url_for('main.show_collectors', photo_id=1), follow_redirects=True)
        data = response.get_data(as_text=True)
        self.assertIn('1 Collectors', data)
        self.assertIn('Normal User', data)

    def test_edit_description(self):
        self.assertEqual(Photo.query.get(2).description, 'Photo 2')

        self.login()
        response = self.client.post(url_for('main.edit_description', photo_id=2), data=dict(
            description='test description.'
        ), follow_redirects=True)
        data = response.get_data(as_text=True)
        self.assertIn('Description updated.', data)
        self.assertEqual(Photo.query.get(2).description, 'test description.')

    def test_new_comment(self):
        self.login()
        response = self.client.post(url_for('main.new_comment', photo_id=1), data=dict(
            body='test comment from normal user.'
        ), follow_redirects=True)
        data = response.get_data(as_text=True)
        self.assertIn('Comment published.', data)
        self.assertEqual(Photo.query.get(1).comments[1].body, 'test comment from normal user.')

    def test_new_tag(self):
        self.login(email='admin@helloflask.com', password='123')

        response = self.client.post(url_for('main.new_tag', photo_id=1), data=dict(
            tag='hello dog pet happy'
        ), follow_redirects=True)
        data = response.get_data(as_text=True)
        self.assertEqual(response.status_code, 200)
        self.assertIn('Tag added.', data)
        self.assertEqual(Photo.query.get(1).tags[1].name, 'hello')
        self.assertEqual(Photo.query.get(1).tags[2].name, 'dog')
        self.assertEqual(Photo.query.get(1).tags[3].name, 'pet')
        self.assertEqual(Photo.query.get(1).tags[4].name, 'happy')

    def test_set_comment(self):
        self.login()
        response = self.client.post(url_for('main.set_comment', photo_id=1), follow_redirects=True)
        self.assertEqual(response.status_code, 403)

        self.logout()
        self.login(email='admin@helloflask.com', password='123')
        response = self.client.post(url_for('main.set_comment', photo_id=1), follow_redirects=True)
        data = response.get_data(as_text=True)
        self.assertEqual(response.status_code, 200)
        self.assertIn('Comment disabled', data)
        self.assertFalse(Photo.query.get(1).can_comment)

        response = self.client.post(url_for('main.set_comment', photo_id=1), follow_redirects=True)
        data = response.get_data(as_text=True)
        self.assertEqual(response.status_code, 200)
        self.assertIn('Comment enabled', data)
        self.assertTrue(Photo.query.get(1).can_comment)

    def test_reply_comment(self):
        self.login()
        response = self.client.get(url_for('main.reply_comment', comment_id=1), follow_redirects=True)
        data = response.get_data(as_text=True)
        self.assertIn('Reply to', data)

    def test_delete_photo(self):
        self.login()
        response = self.client.post(url_for('main.delete_photo', photo_id=2), follow_redirects=True)
        data = response.get_data(as_text=True)
        self.assertIn('Photo deleted.', data)
        self.assertIn('Normal User', data)

    def test_delete_comment(self):
        self.login()
        response = self.client.post(url_for('main.delete_comment', comment_id=1), follow_redirects=True)
        data = response.get_data(as_text=True)
        self.assertIn('Comment deleted.', data)

    def test_show_tag(self):
        response = self.client.get(url_for('main.show_tag', tag_id=1))
        data = response.get_data(as_text=True)
        self.assertIn('Order by time', data)

        response = self.client.get(url_for('main.show_tag', tag_id=1, order='by_collects'))
        data = response.get_data(as_text=True)
        self.assertIn('Order by collects', data)

    def test_delete_tag(self):
        photo = Photo.query.get(2)
        tag = Tag(name='test')
        photo.tags.append(tag)
        db.session.commit()

        self.login()
        response = self.client.post(url_for('main.delete_tag', photo_id=2, tag_id=2), follow_redirects=True)
        data = response.get_data(as_text=True)
        self.assertIn('Tag deleted.', data)

        self.assertEqual(photo.tags, [])
        self.assertIsNone(Tag.query.get(2))