# coding=utf-8

"""

 (c) 2015-2017 - Copyright Red Hat Inc

 Authors:

   Pierre-Yves Chibon <pingou@pingoured.fr></pingou@pingoured.fr>

   Vivek Anand <vivekanand1101@gmail.com></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)