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

"""

Teste le formulaire de recherche avec un groupe d'hôtes.

"""

from nose.tools import assert_true, assert_equal

from datetime import datetime

import transaction

from vigiboard.tests import TestController

from vigilo.models.session import DBSession

from vigilo.models.tables import SupItemGroup, Host, Permission, StateName, \

                                    Event, CorrEvent, User, UserGroup, \

                                    DataPermission

from vigilo.models.tables.grouphierarchy import GroupHierarchy

def insert_deps():

    """Insère les dépendances nécessaires aux tests."""

    timestamp = datetime.now()

    hostgroup = SupItemGroup(

        name=u'foo',

    )

    DBSession.add(hostgroup)

    DBSession.flush()

    DBSession.add(GroupHierarchy(

        parent=hostgroup,

        child=hostgroup,

        hops=0,

    ))

    DBSession.flush()

    host = Host(

        name=u'bar',

        checkhostcmd=u'',

        description=u'',

        hosttpl=u'',

        mainip=u'127.0.0.1',

        snmpport=42,

        snmpcommunity=u'public',

        snmpversion=u'3',

        weight=42,

    )

    DBSession.add(host)

    DBSession.flush()

    hostgroup.supitems.append(host)

    DBSession.flush()

    event = Event(

        supitem=host,

        timestamp=timestamp,

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

        message=u'Hello world',

    )

    DBSession.add(event)

    DBSession.flush()

    correvent = CorrEvent(

        impact=42,

        priority=42,

        trouble_ticket=None,

        status=u'None',

        occurrence=42,

        timestamp_active=timestamp,

        cause=event,

    )

    correvent.events.append(event)

    DBSession.add(correvent)

    DBSession.flush()

    return hostgroup

class TestSearchFormHostGroup(TestController):

    """Teste la récupération d'événements selon le groupe d'hôtes."""

    def setUp(self):

        super(TestSearchFormHostGroup, self).setUp()

        perm = Permission.by_permission_name(u'vigiboard-access')

        user = User(

            user_name=u'user',

            fullname=u'',

            email=u'some.random@us.er',

        )

        usergroup = UserGroup(

            group_name=u'users',

        )

        user.usergroups.append(usergroup)

        usergroup.permissions.append(perm)

        DBSession.add(user)

        DBSession.add(usergroup)

        DBSession.flush()

    def test_search_hostgroup_when_allowed(self):

        """Teste la recherche par hostgroup avec les bons droits d'accès."""

        # On crée un groupe d'hôte appelé 'foo',

        # contenant un hôte 'bar', ainsi qu'un événement

        # et un événement corrélé sur cet hôte.

        # De plus, on donne l'autorisation aux utilisateurs

        # ayant la permission 'edit' de voir cette alerte.

        hostgroup = insert_deps()

        usergroup = UserGroup.by_group_name(u'users')

        DBSession.add(DataPermission(

            group=hostgroup,

            usergroup=usergroup,

            access=u'r',

        ))

        DBSession.flush()

        transaction.commit()

        # On envoie une requête avec recherche sur le groupe d'hôtes créé,

        # on s'attend à recevoir 1 résultat.

        response = self.app.get('/?supitemgroup=foo',

            extra_environ={'REMOTE_USER': 'user'})

        # Il doit y avoir 1 seule ligne de résultats.

        rows = response.lxml.xpath('//table[@class="vigitable"]/tbody/tr')

        print "There are %d rows in the result set" % len(rows)

        assert_equal(len(rows), 1)

        # Il doit y avoir plusieurs colonnes dans la ligne de résultats.

        cols = response.lxml.xpath('//table[@class="vigitable"]/tbody/tr/td')

        print "There are %d columns in the result set" % len(cols)

        assert_true(len(cols) > 1)

    def test_search_inexistent_hostgroup(self):

        """Teste la recherche par hostgroup sur un groupe inexistant."""

        # On envoie une requête avec recherche sur un groupe d'hôtes

        # qui n'existe pas, on s'attend à n'obtenir aucun résultat.

        transaction.commit()

        response = self.app.get('/?supitemgroup=foot',

            extra_environ={'REMOTE_USER': 'user'})

        # Il doit y avoir 1 seule ligne de résultats.

        rows = response.lxml.xpath('//table[@class="vigitable"]/tbody/tr')

        print "There are %d rows in the result set" % len(rows)

        assert_equal(len(rows), 1)

        # Il doit y avoir 1 seule colonne dans la ligne de résultats.

        # (la colonne contient le texte "Il n'y a aucun événément", traduit)

        cols = response.lxml.xpath('//table[@class="vigitable"]/tbody/tr/td')

        print "There are %d columns in the result set" % len(cols)

        assert_equal(len(cols), 1)

    def test_search_hostgroup_when_disallowed(self):

        """Teste la recherche par hostgroup SANS les droits d'accès."""

        # On crée un groupe d'hôte appelé 'foo',

        # contenant un hôte 'bar', ainsi qu'un événement

        # et un événement corrélé sur cet hôte.

        # MAIS, on NE DONNE PAS l'autorisation aux utilisateurs

        # de voir cette alerte, donc elle ne doit jamais apparaître.

        insert_deps()

        transaction.commit()

        # On envoie une requête avec recherche sur le groupe d'hôtes créé,

        # mais avec un utilisateur ne disposant pas des permissions adéquates.

        # On s'attend à n'obtenir aucun résultat.

        response = self.app.get('/?supitemgroup=foo',

            extra_environ={'REMOTE_USER': 'user'})

        # Il doit y avoir 1 seule ligne de résultats.

        rows = response.lxml.xpath('//table[@class="vigitable"]/tbody/tr')

        print "There are %d rows in the result set" % len(rows)

        assert_equal(len(rows), 1)

        # Il doit y avoir 1 seule colonne dans la ligne de résultats.

        # (la colonne contient le texte "Il n'y a aucun événément", traduit)

        cols = response.lxml.xpath('//table[@class="vigitable"]/tbody/tr/td')

        print "There are %d columns in the result set" % len(cols)

        assert_equal(len(cols), 1)

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

"""

Teste le formulaire de recherche avec un groupe de services.

"""

from nose.tools import assert_true, assert_equal

from datetime import datetime

import transaction

from vigiboard.tests import TestController

from vigilo.models.session import DBSession

from vigilo.models.tables import SupItemGroup, Host, Permission, Event, \

                                    LowLevelService, CorrEvent, StateName, \

                                    User, UserGroup, DataPermission

from vigilo.models.tables.grouphierarchy import GroupHierarchy

def insert_deps():

    """Insère les dépendances nécessaires aux tests."""

    timestamp = datetime.now()

    host = Host(

        name=u'bar',

        checkhostcmd=u'',

        description=u'',

        hosttpl=u'',

        mainip=u'127.0.0.1',

        snmpport=42,

        snmpcommunity=u'public',

        snmpversion=u'3',

        weight=42,

    )

    DBSession.add(host)

    DBSession.flush()

    servicegroup = SupItemGroup(

        name=u'foo',

    )

    DBSession.add(servicegroup)

    DBSession.add(GroupHierarchy(

        parent=servicegroup,

        child=servicegroup,

        hops=0,

    ))

    DBSession.flush()

    service = LowLevelService(

        host=host,

        command=u'',

        weight=42,

        servicename=u'baz',

        op_dep=u'&',

    )

    DBSession.add(service)

    DBSession.flush()

    servicegroup.supitems.append(service)

    event = Event(

        supitem=service,

        timestamp=timestamp,

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

        message=u'Hello world',

    )

    DBSession.add(event)

    DBSession.flush()

    correvent = CorrEvent(

        impact=42,

        priority=42,

        trouble_ticket=None,

        status=u'None',

        occurrence=42,

        timestamp_active=timestamp,

        cause=event,

    )

    correvent.events.append(event)

    DBSession.add(correvent)

    DBSession.flush()

    return servicegroup

class TestSearchFormServiceGroup(TestController):

    """Teste la récupération d'événements selon le groupe de services."""

    def setUp(self):

        super(TestSearchFormServiceGroup, self).setUp()

        perm = Permission.by_permission_name(u'vigiboard-access')

        user = User(

            user_name=u'user',

            fullname=u'',

            email=u'some.random@us.er',

        )

        usergroup = UserGroup(

            group_name=u'users',

        )

        user.usergroups.append(usergroup)

        usergroup.permissions.append(perm)

        DBSession.add(user)

        DBSession.add(usergroup)

        DBSession.flush()

    def test_search_servicegroup_when_allowed(self):

        """Teste la recherche par servicegroup avec les bons droits d'accès."""

        # On crée un groupe de services appelé 'foo',

        # contenant un service 'bar', ainsi qu'un événement

        # et un événement corrélé sur ce service.

        # De plus, on donne l'autorisation aux utilisateurs

        # ayant la permission 'edit' de voir cette alerte.

        servicegroup = insert_deps()

        usergroup = UserGroup.by_group_name(u'users')

        DBSession.add(DataPermission(

            group=servicegroup,

            usergroup=usergroup,

            access=u'r',

        ))

        DBSession.flush()

        transaction.commit()

        # On envoie une requête avec recherche sur le groupe

        # de services créé, on s'attend à recevoir 1 résultat.

        response = self.app.get('/?supitemgroup=foo',

            extra_environ={'REMOTE_USER': 'user'})

        # Il doit y avoir 1 seule ligne de résultats.

        rows = response.lxml.xpath('//table[@class="vigitable"]/tbody/tr')

        print "There are %d rows in the result set" % len(rows)

        assert_equal(len(rows), 1)

        # Il doit y avoir plusieurs colonnes dans la ligne de résultats.

        cols = response.lxml.xpath('//table[@class="vigitable"]/tbody/tr/td')

        print "There are %d columns in the result set" % len(cols)

        assert_true(len(cols) > 1)

    def test_search_inexistent_servicegroup(self):

        """Teste la recherche par servicegroup sur un groupe inexistant."""

        transaction.commit()

        # On envoie une requête avec recherche sur un groupe de services

        # qui n'existe pas, on s'attend à n'obtenir aucun résultat.

        response = self.app.get('/?supitemgroup=foot',

            extra_environ={'REMOTE_USER': 'user'})

        # Il doit y avoir 1 seule ligne de résultats.

        rows = response.lxml.xpath('//table[@class="vigitable"]/tbody/tr')

        print "There are %d rows in the result set" % len(rows)

        assert_equal(len(rows), 1)

        # Il doit y avoir 1 seule colonne dans la ligne de résultats.

        # (la colonne contient le texte "Il n'y a aucun événément", traduit)

        cols = response.lxml.xpath('//table[@class="vigitable"]/tbody/tr/td')

        print "There are %d columns in the result set" % len(cols)

        assert_equal(len(cols), 1)

    def test_search_servicegroup_when_disallowed(self):

        """Teste la recherche par servicegroup SANS les droits d'accès."""

        # On crée un groupe de services appelé 'foo',

        # contenant un service 'bar', ainsi qu'un événement

        # et un événement corrélé sur cet hôte.

        # MAIS, on NE DONNE PAS l'autorisation aux utilisateurs

        # de voir cette alerte, donc elle ne doit jamais apparaître.

        insert_deps()

        transaction.commit()

        # On envoie une requête avec recherche sur le groupe de services

        # services créé, mais avec un utilisateur ne disposant pas des

        # permissions adéquates. On s'attend à n'obtenir aucun résultat.

        response = self.app.get('/?supitemgroup=foo',

            extra_environ={'REMOTE_USER': 'user'})

        # Il doit y avoir 1 seule ligne de résultats.

        rows = response.lxml.xpath('//table[@class="vigitable"]/tbody/tr')

        print "There are %d rows in the result set" % len(rows)

        assert_equal(len(rows), 1)

        # Il doit y avoir 1 seule colonne dans la ligne de résultats.

        # (la colonne contient le texte "Il n'y a aucun événément", traduit)

        cols = response.lxml.xpath('//table[@class="vigitable"]/tbody/tr/td')

        print "There are %d columns in the result set" % len(cols)

        assert_equal(len(cols), 1)