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

# vim:set expandtab tabstop=4 shiftwidth=4:

"""

Test de la classe User Utils

"""

from nose.tools import assert_true

from vigiboard.model import * 

from vigiboard.tests import TestController

from vigiboard.controllers.vigiboard_ctl import GetUserGroups

from vigiboard.tests import setup_db, teardown_db

import tg

import transaction

#Create an empty database before we start our tests for this module

def setup():

    """Function called by nose on module load"""

    setup_db()

#Teardown that database 

def teardown():

    """Function called by nose after all tests in this module ran"""

    teardown_db()

    f = open('/home/tandreja/moimoi','a')

    try:

       f.write("%d\n" % DBSession.query(Group).count())

    except:

        pass

    f.close()

class TestUserUtils(TestController):

    def test_get_user_groups(self):

        """Manager est dans le group hostmanagers et hosteditors et Editor seulement dans hosteditors"""

        #On commence par peupler la base

        DBSession.add(Groups(name="hostmanagers"))

        DBSession.add(Groups(name="hosteditors",parent="hostmanagers"))

        idmanagers = DBSession.query(Permission).filter(Permission.permission_name=='manage')[0].permission_id

        ideditors = DBSession.query(Permission).filter(Permission.permission_name=='edit')[0].permission_id

        DBSession.add(GroupPermissions(groupname="hostmanagers",idpermission=idmanagers))

        DBSession.add(GroupPermissions(groupname="hosteditors",idpermission=ideditors))

        # On commit car app.get fait un rollback ou équivalent

        transaction.commit()

        # On obtient les variables de sessions comme si on était loggué en tant que manager

        environ = {'REMOTE_USER': 'manager'}

        response = self.app.get('/', extra_environ=environ)

        tg.request = response.request

        #On récupère la liste des groups auquel on appartient

        grp = GetUserGroups()

        #On vérifi que la liste est correcte (vérifi la gestion des groupes sous forme d'arbre)

        assert_true( 'hostmanagers' in grp and 'hosteditors' in grp)

        #On recommence pour l'utilisateur editor

        environ = {'REMOTE_USER': 'editor'}

        response = self.app.get('/', extra_environ=environ)

        tg.request = response.request

        grp = GetUserGroups()

        assert_true( not('hostmanagers' in grp) and 'hosteditors' in grp)