pagure/tests/test_pagure_lib_model_delete_project.py

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

"""
 (c) 2017 - Copyright Red Hat Inc

 Authors:
   Pierre-Yves Chibon <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)