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

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

"""

Test de la classe Vigiboard Request pour des requêtes concernant les hôtes

"""

from nose.tools import assert_true

from datetime import datetime

import tg

import transaction

from vigilo.models.configure import DBSession

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

                            Permission, User, StateName, \

                            Host, HostGroup

from vigiboard.tests import TestController

from vigiboard.controllers.vigiboardrequest import VigiboardRequest

class TestHostVigiboardRequest(TestController):

    """ Préparation de la base de données en vue des tests. """

    def setUp(self):

        super(TestHostVigiboardRequest, self).setUp()

        # On peuple la base de données.

        # On ajoute les groupes et leurs dépendances

        self.hosteditors = HostGroup(name=u'editorsgroup')

        DBSession.add(self.hosteditors)

        self.hostmanagers = HostGroup(name=u'managersgroup', 

                                        parent=self.hosteditors)

        DBSession.add(self.hostmanagers)

        DBSession.flush()

        manage_perm = Permission.by_permission_name(u'manage')

        edit_perm = Permission.by_permission_name(u'edit')

        self.hostmanagers.permissions.append(manage_perm)

        self.hosteditors.permissions.append(edit_perm)

        DBSession.flush()

        # Création des hôtes de test.

        host_template = {

            'checkhostcmd': u'halt',

            'snmpcommunity': u'public',

            'hosttpl': u'/dev/null',

            'mainip': u'192.168.1.1',

            'snmpport': 42,

            'weight': 42,

        }

        managerhost = Host(name=u'managerhost', **host_template)

        editorhost = Host(name=u'editorhost', **host_template)

        DBSession.add(managerhost)

        DBSession.add(editorhost)

        # Affectation des hôtes aux groupes.

        self.hosteditors.hosts.append(editorhost)

        self.hostmanagers.hosts.append(managerhost)

        DBSession.flush()

        # Ajout des événements eux-mêmes

        event_template = {

            'message': u'foo',

            'current_state': StateName.statename_to_value(u'WARNING'),

        }

        event1 = Event(supitem=managerhost, **event_template)

        event2 = Event(supitem=editorhost, **event_template)

        DBSession.add(event1)

        DBSession.add(event2)

        DBSession.flush()

        # Ajout des historiques

        DBSession.add(EventHistory(type_action=u'Nagios update state',

            idevent=event1.idevent, timestamp=datetime.now()))

        DBSession.add(EventHistory(type_action=u'Acknowledgement change state',

            idevent=event1.idevent, timestamp=datetime.now()))

        DBSession.add(EventHistory(type_action=u'Nagios update state',

            idevent=event2.idevent, timestamp=datetime.now()))

        DBSession.add(EventHistory(type_action=u'Acknowledgement change state',

            idevent=event2.idevent, timestamp=datetime.now()))

        DBSession.flush()

        # Ajout des événements corrélés

        aggregate_template = {

            'timestamp_active': datetime.now(),

            'priority': 1,

            'status': u'None',

        }

        self.aggregate1 = CorrEvent(

            idcause=event1.idevent, **aggregate_template)

        self.aggregate2 = CorrEvent(

            idcause=event2.idevent, **aggregate_template)

        self.aggregate1.events.append(event1)

        self.aggregate2.events.append(event2)

        DBSession.add(self.aggregate1)

        DBSession.add(self.aggregate2)

        DBSession.flush()

        transaction.commit()

    def tearDown(self):

        """ Nettoyage de la base de données après les tests. """

        super(TestHostVigiboardRequest, self).tearDown()

    def test_request_creation(self):

        """Génération d'une requête avec plugin et permissions."""

        # On indique qui on est et on envoie une requête sur 

        # l'index pour obtenir toutes les variables de sessions.

        environ = {'REMOTE_USER': 'editor'}

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

        tg.request = response.request

        # Derrière, VigiboardRequest doit charger le plugin de tests tout seul

        vigi_req = VigiboardRequest(User.by_user_name(u'editor'))

        vigi_req.add_table(

            CorrEvent,

            vigi_req.items.c.hostname,

            vigi_req.items.c.servicename,

        )

        vigi_req.add_join((Event, CorrEvent.idcause == Event.idevent))

        vigi_req.add_join((vigi_req.items, 

            Event.idsupitem == vigi_req.items.c.idsupitem))

        # On vérifie que le nombre d'événements corrélés 

        # trouvés par la requête est bien égal à 1.

        num_rows = vigi_req.num_rows()

        assert_true(num_rows == 1, 

            msg = "One history should be available for " +

            "the user 'editor' but there are %d" % num_rows)

        vigi_req.format_events(0, 10)

        vigi_req.format_history()

        assert_true(len(vigi_req.events) == 1 + 1,

            msg = "One history should be available for user " +

            "'editor' but there are %d" % (len(vigi_req.events) - 1))

        # On recommence les tests précédents avec l'utilisateur

        # manager (qui dispose de plus de droits).

        environ = {'REMOTE_USER': 'manager'}

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

        tg.request = response.request

        vigi_req = VigiboardRequest(User.by_user_name(u'manager'))

        vigi_req.add_plugin(MonPlugin)

        vigi_req.add_table(

            CorrEvent,

            vigi_req.items.c.hostname,

            vigi_req.items.c.servicename,

        )

        vigi_req.add_join((Event, CorrEvent.idcause == Event.idevent))

        vigi_req.add_join((vigi_req.items, 

            Event.idsupitem == vigi_req.items.c.idsupitem))

        # On vérifie que le nombre d'événements corrélés 

        # trouvés par la requête est bien égal à 2.

        num_rows = vigi_req.num_rows()

        assert_true(num_rows == 2,  

            msg = "2 histories should be available for " +

            "the user 'manager' but there are %d" % num_rows)

        vigi_req.format_events(0, 10)

        vigi_req.format_history()

        assert_true(len(vigi_req.events) == 2 + 1,

            msg = "2 histories should be available for user " +

            "'manager' but there are %d" % (len(vigi_req.events) - 1))