import os

from sqlalchemy.exc import IntegrityError

from kallithea.tests import *

from kallithea.tests.fixture import Fixture

from kallithea.model.repo_group import RepoGroupModel

from kallithea.model.repo import RepoModel

from kallithea.model.db import RepoGroup

from kallithea.model.meta import Session

fixture = Fixture()

def _update_group(id_, group_name, desc=u'desc', parent_id=None):

    form_data = fixture._get_group_create_params(group_name=group_name,

                                                 group_desc=desc,

                                                 group_parent_id=parent_id)

    gr = RepoGroupModel().update(id_, form_data)

    return gr

def _update_repo(name, **kwargs):

    form_data = fixture._get_repo_create_params(**kwargs)

    if not 'repo_name' in kwargs:

        form_data['repo_name'] = name

    if not 'perms_new' in kwargs:

        form_data['perms_new'] = []

    if not 'perms_updates' in kwargs:

        form_data['perms_updates'] = []

    r = RepoModel().update(name, **form_data)

    return r

class TestRepoGroups(TestControllerPytest):

    def setup_method(self, method):

        self.g1 = fixture.create_repo_group(u'test1', skip_if_exists=True)

        self.g2 = fixture.create_repo_group(u'test2', skip_if_exists=True)

        self.g3 = fixture.create_repo_group(u'test3', skip_if_exists=True)

    def teardown_method(self, method):

        Session.remove()

    def __check_path(self, *path):

        """

        """

        path = [TESTS_TMP_PATH] + list(path)

        path = os.path.join(*path)

        return os.path.isdir(path)

    def _check_folders(self):

        print os.listdir(TESTS_TMP_PATH)

    def __delete_group(self, id_):

        RepoGroupModel().delete(id_)

    def test_create_group(self):

        g = fixture.create_repo_group(u'newGroup')

        Session().commit()

        self.assertEqual(g.full_path, 'newGroup')

        self.assertTrue(self.__check_path('newGroup'))

    def test_create_same_name_group(self):

        self.assertRaises(IntegrityError, lambda: fixture.create_repo_group(u'newGroup'))

        Session().rollback()

    def test_same_subgroup(self):

        sg1 = fixture.create_repo_group(u'sub1', group_parent_id=self.g1.group_id)

        self.assertEqual(sg1.parent_group, self.g1)

        self.assertEqual(sg1.full_path, 'test1/sub1')

        self.assertTrue(self.__check_path('test1', 'sub1'))

        ssg1 = fixture.create_repo_group(u'subsub1', group_parent_id=sg1.group_id)

        self.assertEqual(ssg1.parent_group, sg1)

        self.assertEqual(ssg1.full_path, 'test1/sub1/subsub1')

        self.assertTrue(self.__check_path('test1', 'sub1', 'subsub1'))

    def test_remove_group(self):

        sg1 = fixture.create_repo_group(u'deleteme')

        self.__delete_group(sg1.group_id)

        self.assertEqual(RepoGroup.get(sg1.group_id), None)

        self.assertFalse(self.__check_path('deteteme'))

        sg1 = fixture.create_repo_group(u'deleteme', group_parent_id=self.g1.group_id)

        self.__delete_group(sg1.group_id)

        self.assertEqual(RepoGroup.get(sg1.group_id), None)

        self.assertFalse(self.__check_path('test1', 'deteteme'))

    def test_rename_single_group(self):

        sg1 = fixture.create_repo_group(u'initial')