# -*- coding: utf-8 -*-

"""

 (c) 2017 - Copyright Red Hat Inc

 Authors:

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

"""

from __future__ import unicode_literals, absolute_import

import json

import unittest

import sys

import os

import uuid

sys.path.insert(0, os.path.join(os.path.dirname(

    os.path.abspath(__file__)), '..'))

import pagure.lib.model as model

import pagure.lib.query

import tests

class DeleteProjectTests(tests.Modeltests):

    """ Tests for flask issues controller of pagure """

    def test_delete_project_with_group(self):

        """ Test the model when we delete a project with a group. """

        # Create a project

        item = model.Project(

            user_id=1,  # pingou

            name='test',

            description='test project #1',

            hook_token='aaabbbiii',

        )

        self.session.add(item)

        self.session.commit()

        # Create a group

        grp = model.PagureGroup(

            group_name='testgrp',

            display_name='Test group',

            description=None,

            group_type='user',

            user_id=1,  # pingou

        )

        self.session.add(grp)

        self.session.commit()

        # Add group to project

        project_group = model.ProjectGroup(

            project_id=1,

            group_id=1,

            access='admin',

        )

        self.session.add(project_group)

        self.session.commit()

        self.assertEqual(

            self.session.query(model.Project).count(), 1)

        self.assertEqual(

            self.session.query(model.ProjectGroup).count(), 1)

        project = self.session.query(

            model.Project

        ).filter(

            model.Project.id==1

        ).one()

        self.session.delete(project)

        self.session.commit()

        self.assertEqual(

            self.session.query(model.Project).count(), 0)

        self.assertEqual(

            self.session.query(model.ProjectGroup).count(), 0)

    def test_delete_project_with_user(self):

        """ Test the model when we delete a project with users. """

        # Create a project

        item = model.Project(

            user_id=1,  # pingou

            name='test',

            description='test project #1',

            hook_token='aaabbbiii',

        )

        self.session.add(item)

        self.session.commit()

        # Add user #2 to project

        project_user = model.ProjectUser(

            project_id=1,

            user_id=2,

            access='admin',

        )

        self.session.add(project_user)

        self.session.commit()

        self.assertEqual(

            self.session.query(model.Project).count(), 1)

        self.assertEqual(

            self.session.query(model.ProjectUser).count(), 1)

        self.assertEqual(

            self.session.query(model.User).count(), 2)

        project = self.session.query(

            model.Project

        ).filter(

            model.Project.id==1

        ).one()

        self.session.delete(project)

        self.session.commit()

        self.assertEqual(

            self.session.query(model.Project).count(), 0)

        self.assertEqual(

            self.session.query(model.ProjectUser).count(), 0)

        self.assertEqual(

            self.session.query(model.User).count(), 2)

    def test_delete_project_with_coloredtags(self):

        """ Test the model when we delete a project with Colored tags. """

        # Create a project

        item = model.Project(

            user_id=1,  # pingou

            name='test',

            description='test project #1',

            hook_token='aaabbbiii',

        )

        self.session.add(item)

        self.session.commit()

        # Create two ColoredTags

        tagobj = model.TagColored(

            tag='Tag#1',

            project_id=1

        )

        self.session.add(tagobj)

        self.session.flush()

        tagobj = model.TagColored(

            tag='Tag#2',

            project_id=1

        )

        self.session.add(tagobj)

        self.session.flush()

        self.assertEqual(

            self.session.query(model.Project).count(), 1)

        self.assertEqual(

            self.session.query(model.TagColored).count(), 2)

        project = self.session.query(

            model.Project

        ).filter(

            model.Project.id==1

        ).one()

        self.session.delete(project)

        self.session.commit()

        self.assertEqual(

            self.session.query(model.Project).count(), 0)

        self.assertEqual(

            self.session.query(model.TagColored).count(), 0)

    def test_delete_project_with_coloredtags_and_issues(self):

        """ Test the model when we delete a project with Colored tags and

        issues. """

        # Create a project

        item = model.Project(

            user_id=1,  # pingou

            name='test',

            description='test project #1',

            hook_token='aaabbbiii',

        )

        self.session.add(item)

        self.session.commit()

        # Create two ColoredTags

        tagobj = model.TagColored(

            tag='Tag#1',

            project_id=1

        )

        self.session.add(tagobj)

        self.session.flush()

        tagobj = model.TagColored(

            tag='Tag#2',

            project_id=1

        )

        self.session.add(tagobj)

        self.session.flush()

        # Create issues

        issue = model.Issue(

            id=pagure.lib.query.get_next_id(self.session, 1),

            project_id=1,

            title='Issue #1',

            content='Description #1',

            user_id=1,

            uid=uuid.uuid4().hex,

            private=False,

        )

        self.session.add(issue)

        self.session.commit()

        issue = model.Issue(

            id=pagure.lib.query.get_next_id(self.session, 1),

            project_id=1,

            title='Issue #2',

            content='Description #2',

            user_id=1,

            uid=uuid.uuid4().hex,

            private=False,

        )

        self.session.add(issue)

        self.session.commit()

        self.assertEqual(

            self.session.query(model.Project).count(), 1)

        self.assertEqual(

            self.session.query(model.TagColored).count(), 2)

        self.assertEqual(

            self.session.query(model.Issue).count(), 2)

        project = self.session.query(

            model.Project

        ).filter(

            model.Project.id==1

        ).one()

        self.session.delete(project)

        self.session.commit()

        self.assertEqual(

            self.session.query(model.Project).count(), 0)

        self.assertEqual(

            self.session.query(model.TagColored).count(), 0)

        self.assertEqual(

            self.session.query(model.Issue).count(), 0)

    def test_delete_project_with_coloredtags_and_tagged_issues(self):

        """ Test the model when we delete a project with Colored tags and

        tagged issues. """

        # Create a project

        item = model.Project(

            user_id=1,  # pingou

            name='test',

            description='test project #1',

            hook_token='aaabbbiii',

        )

        self.session.add(item)

        self.session.commit()

        # Create two ColoredTags

        tagobj = model.TagColored(

            tag='Tag#1',

            project_id=1

        )

        self.session.add(tagobj)

        self.session.flush()

        tagobj = model.TagColored(

            tag='Tag#2',

            project_id=1

        )

        self.session.add(tagobj)

        self.session.flush()

        # Create issues

        issue = model.Issue(

            id=pagure.lib.query.get_next_id(self.session, 1),

            project_id=1,

            title='Issue #1',

            content='Description #1',

            user_id=1,

            uid='Issue#1',

            private=False,

        )

        self.session.add(issue)

        self.session.commit()

        issue = model.Issue(

            id=pagure.lib.query.get_next_id(self.session, 1),

            project_id=1,

            title='Issue #2',

            content='Description #2',

            user_id=1,

            uid='Issue#2',

            private=False,

        )

        self.session.add(issue)

        self.session.commit()

        # Tag the issue

        tagissue = model.TagIssueColored(

            issue_uid='Issue#1',

            tag_id=1

        )

        self.session.add(tagissue)

        self.session.commit()

        tagissue = model.TagIssueColored(

            issue_uid='Issue#2',

            tag_id=2

        )

        self.session.add(tagissue)

        self.session.commit()

        self.assertEqual(

            self.session.query(model.Project).count(), 1)

        self.assertEqual(

            self.session.query(model.TagColored).count(), 2)

        self.assertEqual(

            self.session.query(model.Issue).count(), 2)

        project = self.session.query(

            model.Project

        ).filter(

            model.Project.id==1

        ).one()

        self.session.delete(project)

        self.session.commit()

        self.assertEqual(

            self.session.query(model.Project).count(), 0)

        self.assertEqual(

            self.session.query(model.TagColored).count(), 0)

        self.assertEqual(

            self.session.query(model.Issue).count(), 0)

if __name__ == '__main__':

    unittest.main(verbosity=2)