pagure/tests/test_pagure_lib_star_project.py

# coding=utf-8
"""
 (c) 2015-2017 - Copyright Red Hat Inc

 Authors:
   Pierre-Yves Chibon <pingou@pingoured.fr>
   Vivek Anand <vivekanand1101@gmail.com>

"""

from __future__ import unicode_literals, absolute_import

import unittest
import sys
import os

sys.path.insert(0, os.path.join(os.path.dirname(
    os.path.abspath(__file__)), '..'))

import pagure.lib.model
import pagure.lib.query
import tests


class TestStarProjectLib(tests.SimplePagureTest):
    ''' Test the star project feature of pagure '''

    def setUp(self):
        """ Set up the environnment for running each star project lib tests """
        super(TestStarProjectLib, self).setUp()
        tests.create_projects(self.session)

    def test_update_star_project(self):
        ''' Test the update_star_project endpoint of pagure.lib '''

        repo_obj = pagure.lib.query._get_project(self.session, 'test')
        # test with invalud Star object, should return None
        msg = pagure.lib.query.update_star_project(
            self.session,
            repo_obj,
            None,
            'pingou',
        )
        self.session.commit()
        self.assertEqual(msg, None)
        user_obj = pagure.lib.query.get_user(self.session, 'pingou')
        self.assertEqual(len(user_obj.stars), 0)

        # test starring the project
        repo_obj = pagure.lib.query._get_project(self.session, 'test')
        msg = pagure.lib.query.update_star_project(
            self.session,
            repo_obj,
            '1',
            'pingou',
        )

        self.session.commit()
        self.assertEqual(msg, 'You starred this project')
        user_obj = pagure.lib.query.get_user(self.session, 'pingou')
        self.assertEqual(len(user_obj.stars), 1)

        # test unstarring the project
        repo_obj = pagure.lib.query._get_project(self.session, 'test')
        msg = pagure.lib.query.update_star_project(
            self.session,
            repo_obj,
            '0',
            'pingou',
        )

        self.session.commit()
        self.assertEqual(msg, 'You unstarred this project')
        user_obj = pagure.lib.query.get_user(self.session, 'pingou')
        self.assertEqual(len(user_obj.stars), 0)

    def test_star_project(self):
        ''' Test the _star_project endpoint of pagure.lib '''

        # test with not all arguments present
        user_obj = pagure.lib.query.get_user(self.session, 'pingou')
        repo_obj = pagure.lib.query._get_project(self.session, 'test')
        msg = pagure.lib.query._star_project(
            self.session,
            repo_obj,
            None
        )
        self.session.commit()
        self.assertEqual(msg, None)

        repo_obj = pagure.lib.query._get_project(self.session, 'test')
        user_obj = pagure.lib.query.get_user(self.session, 'pingou')
        msg = pagure.lib.query._star_project(
            self.session,
            repo_obj,
            user_obj,
        )
        self.session.commit()
        self.assertEqual(msg, 'You starred this project')
        user_obj = pagure.lib.query.get_user(self.session, 'pingou')
        self.assertEqual(len(user_obj.stars), 1)

    def test_unstar_project(self):
        ''' Test the _unstar_project endpoint of pagure.lib '''

        # test with not all arguments present
        user_obj = pagure.lib.query.get_user(self.session, 'pingou')
        repo_obj = pagure.lib.query._get_project(self.session, 'test')
        msg = pagure.lib.query._unstar_project(
            self.session,
            repo_obj,
            None
        )
        self.session.commit()
        self.assertEqual(msg, None)

        # the user hasn't starred the project before
        repo_obj = pagure.lib.query._get_project(self.session, 'test')
        user_obj = pagure.lib.query.get_user(self.session, 'pingou')
        msg = pagure.lib.query._unstar_project(
            self.session,
            repo_obj,
            user_obj,
        )
        self.assertEqual(msg, 'You never starred the project')
        user_obj = pagure.lib.query.get_user(self.session, 'pingou')
        self.assertEqual(len(user_obj.stars), 0)

        # star it for testing
        msg = pagure.lib.query._star_project(
            self.session,
            repo_obj,
            user_obj,
        )
        self.session.commit()
        self.assertEqual(msg, 'You starred this project')
        user_obj = pagure.lib.query.get_user(self.session, 'pingou')
        self.assertEqual(len(user_obj.stars), 1)

        # the user starred and wishes to unstar
        repo_obj = pagure.lib.query._get_project(self.session, 'test')
        user_obj = pagure.lib.query.get_user(self.session, 'pingou')
        msg = pagure.lib.query._unstar_project(
            self.session,
            repo_obj,
            user_obj,
        )
        self.session.commit()
        self.assertEqual(msg, 'You unstarred this project')
        user_obj = pagure.lib.query.get_user(self.session, 'pingou')
        self.assertEqual(len(user_obj.stars), 0)

    def test_get_stargazer_obj(self):
        ''' Test the _get_stargazer_obj test of pagure.lib '''

        # star the project first
        repo_obj = pagure.lib.query._get_project(self.session, 'test')
        user_obj = pagure.lib.query.get_user(self.session, 'pingou')
        msg = pagure.lib.query._star_project(
            self.session,
            repo_obj,
            user_obj,
        )
        self.session.commit()
        self.assertEqual(msg, 'You starred this project')
        user_obj = pagure.lib.query.get_user(self.session, 'pingou')
        self.assertEqual(len(user_obj.stars), 1)

        # get the object now
        repo_obj = pagure.lib.query._get_project(self.session, 'test')
        star_obj = pagure.lib.query._get_stargazer_obj(
            self.session,
            repo_obj,
            user_obj
        )
        self.assertEqual(isinstance(star_obj, pagure.lib.model.Star), True)

        # unstar it and then try to get the object
        repo_obj = pagure.lib.query._get_project(self.session, 'test')
        user_obj = pagure.lib.query.get_user(self.session, 'pingou')
        msg = pagure.lib.query._unstar_project(
            self.session,
            repo_obj,
            user_obj,
        )
        self.session.commit()
        self.assertEqual(msg, 'You unstarred this project')
        user_obj = pagure.lib.query.get_user(self.session, 'pingou')
        self.assertEqual(len(user_obj.stars), 0)

        # we don't store if the user has unstarred, we delete the obj
        # so, we should get anything back in the query
        repo_obj = pagure.lib.query._get_project(self.session, 'test')
        star_obj = pagure.lib.query._get_stargazer_obj(
            self.session,
            repo_obj,
            user_obj
        )
        self.assertEqual(star_obj is None, True)

    def test_has_starred(self):
        ''' Test the has_starred endpoint of pagure.lib '''

        # star the project
        repo_obj = pagure.lib.query._get_project(self.session, 'test')
        user_obj = pagure.lib.query.get_user(self.session, 'pingou')
        msg = pagure.lib.query._star_project(
            self.session,
            repo_obj,
            user_obj,
        )
        self.session.commit()
        self.assertEqual(msg, 'You starred this project')
        user_obj = pagure.lib.query.get_user(self.session, 'pingou')
        self.assertEqual(len(user_obj.stars), 1)

        has_starred = pagure.lib.query.has_starred(
            self.session,
            repo_obj,
            'pingou'
        )
        self.assertEqual(has_starred is True, True)

        # unstar it and then test for has_starred
        repo_obj = pagure.lib.query._get_project(self.session, 'test')
        user_obj = pagure.lib.query.get_user(self.session, 'pingou')
        msg = pagure.lib.query._unstar_project(
            self.session,
            repo_obj,
            user_obj,
        )
        self.session.commit()
        self.assertEqual(msg, 'You unstarred this project')
        user_obj = pagure.lib.query.get_user(self.session, 'pingou')
        self.assertEqual(len(user_obj.stars), 0)

        # check now, it should return False
        has_starred = pagure.lib.query.has_starred(
            self.session,
            repo_obj,
            'pingou'
        )
        self.assertEqual(has_starred is False, True)


if __name__ == '__main__':
    unittest.main(verbosity=2)