from vigilo.models.session import DBSession

from vigilo.models.tables import Event, EventHistory, CorrEvent, \

                            Permission, StateName, Host, \

                            Permission, DataPermission, User, \

from vigiboard.tests import TestController

def populate_DB():

    """ Peuple la base de données. """

    # On ajoute un groupe d'hôtes et un groupe de services.

    usergroup = UserGroup.by_group_name(u'users_with_access')

    DBSession.add(DataPermission(

    # On crée un hôte de test, et on l'ajoute au groupe d'hôtes.

    managerhost = Host(

        checkhostcmd = u'halt',

        snmpcommunity = u'public',

        hosttpl = u'/dev/null',

    DBSession.add(managerservice)

    supitemmanagers.supitems.append(managerservice)

    DBSession.flush()

    return (managerhost, managerservice)

def add_correvent_caused_by(supitem):

    """

    par un incident survenu sur l'item passé en paramètre.

    Génère un historique pour les tests.

    """

    # Ajout d'un événement

    event = Event(

        message = u'foo',

        current_state = StateName.statename_to_value(u"WARNING"),

        timestamp = datetime.now(),

    # Ajout des historiques

    DBSession.add(EventHistory(

        type_action=u'Nagios update state',

        timestamp=datetime.now()))

    DBSession.add(EventHistory(

        type_action=u'Acknowlegement change state',

        timestamp=datetime.now()))

    DBSession.flush()

    # Ajout d'un événement corrélé

    aggregate = CorrEvent(

        timestamp_active = datetime.now(),

        priority = 1,

        status = u"None")

    aggregate.events.append(event)

    DBSession.add(aggregate)

    DBSession.flush()

    return event.idevent

class TestHistoryTable(TestController):

    """

        # On peuple la BDD avec un hôte, un service de bas niveau,

        # et un groupe d'hôtes et de services associés à ces items.

        (managerhost, managerservice) = populate_DB()

        # On ajoute un évènement corrélé causé par l'hôte

        idevent = add_correvent_caused_by(managerhost)

        transaction.commit()

        # L'utilisateur N'A PAS les bonnes permissions.

        environ = {'REMOTE_USER': 'limited_access'}

        # On s'attend à ce qu'une erreur 302 soit renvoyée, et à

        # ce qu'un message d'erreur précise à l'utilisateur qu'il

        # n'a pas accès aux informations concernant cet évènement.

        response = self.app.get(

            '/event/%d' % idevent,

            extra_environ = environ)

        # On suit la redirection.

        # L'utilisateur a les bonnes permissions.

        environ = {'REMOTE_USER': 'access'}

        # On s'attend à ce que le statut de la requête soit 200.

        response = self.app.get(

            '/event/%d' % idevent,

            extra_environ = environ)

        # Il doit y avoir 2 lignes de résultats.

        # On peuple la BDD avec un hôte, un service de bas niveau,

        # et un groupe d'hôtes et de services associés à ces items.

        (managerhost, managerservice) = populate_DB()

        # On ajoute un évènement corrélé causé par le service

        idevent = add_correvent_caused_by(managerservice)

        transaction.commit()

        # L'utilisateur n'est pas authentifié.

        # L'utilisateur N'A PAS les bonnes permissions.

        environ = {'REMOTE_USER': 'limited_access'}

        # On s'attend à ce qu'une erreur 302 soit renvoyée, et à

        # ce qu'un message d'erreur précise à l'utilisateur qu'il

        # n'a pas accès aux informations concernant cet évènement.

        response = self.app.get(

            '/event/%d' % idevent,

            extra_environ = environ)

        # On suit la redirection.

        # L'utilisateur a les bonnes permissions.

        environ = {'REMOTE_USER': 'access'}

        # On s'attend à ce que le statut de la requête soit 200.

        response = self.app.get(

            '/event/%d' % idevent,

            extra_environ = environ)

        # Il doit y avoir 2 lignes de résultats.

        rows = response.lxml.xpath(

            '//table[contains(@class,"vigitable")]/tbody/tr')

        assert_equal(len(rows), 2)