from kallithea.model.permission import PermissionModel

fixture = Fixture()

class TestPermissions(TestController):

    @classmethod

    def setup_class(cls):

        #recreate default user to get a clean start

        PermissionModel().create_default_permissions(user=User.DEFAULT_USER,

                                                     force=True)

        Session().commit()

    def setup_method(self, method):

        self.u1 = UserModel().create_or_update(

            username=u'u1', password=u'qweqwe',

            email=u'u1@example.com', firstname=u'u1', lastname=u'u1'

        )

        self.u2 = UserModel().create_or_update(

            username=u'u2', password=u'qweqwe',

            email=u'u2@example.com', firstname=u'u2', lastname=u'u2'

        )

        self.u3 = UserModel().create_or_update(

            username=u'u3', password=u'qweqwe',

            email=u'u3@example.com', firstname=u'u3', lastname=u'u3'

        )

        self.anon = User.get_default_user()

        self.a1 = UserModel().create_or_update(

            username=u'a1', password=u'qweqwe',

            email=u'a1@example.com', firstname=u'a1', lastname=u'a1', admin=True

        )

        Session().commit()

    def teardown_method(self, method):

        if hasattr(self, 'test_repo'):

            RepoModel().delete(repo=self.test_repo)

        UserModel().delete(self.u1)

        UserModel().delete(self.u2)

        UserModel().delete(self.u3)

        UserModel().delete(self.a1)

        if hasattr(self, 'g1'):

            RepoGroupModel().delete(self.g1.group_id)

        if hasattr(self, 'g2'):

            RepoGroupModel().delete(self.g2.group_id)

        if hasattr(self, 'ug1'):

            UserGroupModel().delete(self.ug1, force=True)

        Session().commit()

    def test_default_perms_set(self):

        u1_auth = AuthUser(user_id=self.u1.user_id)

        perms = {

            'repositories_groups': {},

            'global': set(['hg.create.repository', 'repository.read',

                           'hg.register.manual_activate']),

            'repositories': {HG_REPO: 'repository.read'}

        }

        assert u1_auth.permissions['repositories'][HG_REPO] == perms['repositories'][HG_REPO]

        new_perm = 'repository.write'

        RepoModel().grant_user_permission(repo=HG_REPO, user=self.u1,

                                          perm=new_perm)

        Session().commit()

        u1_auth = AuthUser(user_id=self.u1.user_id)

        assert u1_auth.permissions['repositories'][HG_REPO] == new_perm

    def test_default_admin_perms_set(self):

        a1_auth = AuthUser(user_id=self.a1.user_id)

        perms = {

            'repositories_groups': {},

            'global': set(['hg.admin', 'hg.create.write_on_repogroup.true']),

            'repositories': {HG_REPO: 'repository.admin'}

        }

        assert a1_auth.permissions['repositories'][HG_REPO] == perms['repositories'][HG_REPO]

        new_perm = 'repository.write'

        RepoModel().grant_user_permission(repo=HG_REPO, user=self.a1,

                                          perm=new_perm)

        Session().commit()

        # cannot really downgrade admins permissions !? they still gets set as

        # admin !

        u1_auth = AuthUser(user_id=self.a1.user_id)

        assert u1_auth.permissions['repositories'][HG_REPO] == perms['repositories'][HG_REPO]

    def test_default_group_perms(self):

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

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

        u1_auth = AuthUser(user_id=self.u1.user_id)

        perms = {

            'repositories_groups': {u'test1': 'group.read', u'test2': 'group.read'},

            'global': set(Permission.DEFAULT_USER_PERMISSIONS),

            'repositories': {HG_REPO: 'repository.read'}

        }

        assert u1_auth.permissions['repositories'][HG_REPO] == perms['repositories'][HG_REPO]

        assert u1_auth.permissions['repositories_groups'] == perms['repositories_groups']