# -*- 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)