from vigilo.models.session import DBSession

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

                            Permission, StateName, Host, UserGroup, \

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.

    DBSession.add(supitemmanagers)

    DBSession.flush()

    # On crée un 2 hôtes, et on les ajoute au groupe d'hôtes.

    host1 = Host(

        checkhostcmd = u'halt',

        snmpcommunity = u'public',

        hosttpl = u'/dev/null',

    DBSession.add(host1)

    supitemmanagers.supitems.append(host1)

    host2 = Host(

        checkhostcmd = u'halt',

        snmpcommunity = u'public',

        hosttpl = u'/dev/null',

    DBSession.add(service2)

    supitemmanagers.supitems.append(service2)

    DBSession.flush()

    return ([host1, host2], [service1, service2])

def add_correvent_caused_by(supitem, timestamp,

        correvent_status=u"None", event_status=u"WARNING"):

    """

    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(event_status),

        timestamp = datetime.now(),

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

    aggregate = CorrEvent(

        timestamp_active = timestamp,

        priority = 1,

        status = correvent_status)

    aggregate.events.append(event)

    DBSession.add(aggregate)

    DBSession.flush()

    return aggregate.idcorrevent

        # On peuple la BDD avec 2 hôtes, 2 services de bas niveau,

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

        (hosts, services) = populate_DB()

        # On ajoute 2 évènements corrélés causés par ces hôtes

        timestamp = datetime.now()

        correvent1_id = add_correvent_caused_by(hosts[0], timestamp)

        correvent2_id = add_correvent_caused_by(hosts[1], timestamp)

        transaction.commit()

        ### 1er cas : L'utilisateur n'est pas connecté.

        # On vérifie que le plugin retourne bien une erreur 401.

        response = self.app.post(

                "trouble_ticket" : u"foo",

                "ack" : u'NoChange',

            }, status = 401)

        ### 2ème cas : L'utilisateur utilisé pour

        # se connecter à Vigiboard est 'limited_access'.

        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 la permission de modifier ces évènements.

                "trouble_ticket" : u"foo",

                "last_modification": mktime(timestamp.timetuple()),

            }, status = 302, extra_environ = environ)

        response = response.follow(status=200, extra_environ = environ)

        assert_true(response.lxml.xpath(

            '//div[@id="flash"]/div[@class="error"]'))

        ### 3ème cas : L'utilisateur utilisé pour

        # se connecter à Vigiboard est 'access'.

        environ = {'REMOTE_USER': 'access'}

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

        # et à ce qu'un message informe l'utilisateur que les

        # évènements corrélés sélectionnées ont bien été mis à jour.

                "trouble_ticket" : u"foo",

                "ack" : u'NoChange',

            }, status = 302, extra_environ = environ)

        response = response.follow(status=200, extra_environ = environ)

        assert_false(response.lxml.xpath(

            '//div[@id="flash"]/div[@class="error"]'))

        assert_true(response.lxml.xpath(

            '//div[@id="flash"]/div[@class="ok"]'))

        # On s'assure que le ticket de l'évènement corrélé

        # a bien été mis à jour dans la base de données.

        correvents = DBSession.query(

            CorrEvent.trouble_ticket

            ).filter(CorrEvent.idcorrevent.in_([correvent1_id, correvent2_id])

            ).all()

        assert_equal(correvents[0].trouble_ticket, u"foo")

        assert_equal(correvents[1].trouble_ticket, u"foo")

        # On peuple la BDD avec 2 hôtes, 2 services de bas niveau,

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

        (hosts, services) = populate_DB()

        # On ajoute 2 évènements corrélés causés par ces hôtes

        timestamp = datetime.now()

        correvent1_id = add_correvent_caused_by(services[0], timestamp)

        correvent2_id = add_correvent_caused_by(services[1], timestamp)

        transaction.commit()

        ### 1er cas : L'utilisateur n'est pas connecté.

        # On vérifie que le plugin retourne bien une erreur 401.

        response = self.app.post(

                "trouble_ticket" : u"foo",

                "ack" : u'NoChange',

            }, status = 401)

        ### 2ème cas : L'utilisateur utilisé pour

        # se connecter à Vigiboard est 'limited_access'.

        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 la permission de modifier ces évènements.

                "trouble_ticket" : u"foo",

                "ack" : u'NoChange',

            }, status = 302, extra_environ = environ)

        response = response.follow(status=200, extra_environ = environ)

        assert_true(response.lxml.xpath(

            '//div[@id="flash"]/div[@class="error"]'))

        ### 3ème cas : L'utilisateur utilisé pour

        # se connecter à Vigiboard est 'access'.

        environ = {'REMOTE_USER': 'access'}

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

        # et à ce qu'un message informe l'utilisateur que les

        # évènements corrélés sélectionnées ont bien été mis à jour.

                "trouble_ticket" : u"foo",

                "ack" : u'NoChange',

            }, status = 302, extra_environ = environ)

        response = response.follow(status=200, extra_environ = environ)

        assert_false(response.lxml.xpath(

            '//div[@id="flash"]/div[@class="error"]'))

        assert_true(response.lxml.xpath(

            '//div[@id="flash"]/div[@class="ok"]'))

        # On s'assure que le ticket de l'évènement corrélé

        # a bien été mis à jour dans la base de données.

        correvents = DBSession.query(

        # On peuple la BDD avec 2 hôtes, 2 services de bas niveau,

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

        (hosts, services) = populate_DB()

        # On ajoute 2 évènements corrélés causés par ces hôtes

        timestamp = datetime.now()

        correvent1_id = add_correvent_caused_by(hosts[0], timestamp)

        correvent2_id = add_correvent_caused_by(hosts[1], timestamp)

        transaction.commit()

        ### 1er cas : L'utilisateur n'est pas connecté.

        # On vérifie que le plugin retourne bien une erreur 401.

        response = self.app.post(

                "trouble_ticket" : "",

                "ack" : u'Acknowledged',

            }, status = 401)

        ### 2ème cas : L'utilisateur utilisé pour

        # se connecter à Vigiboard est 'limited_access'.

        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 la permission de modifier ces évènements.

                "trouble_ticket" : "",

                "ack" : u'Acknowledged',

            }, status = 302, extra_environ = environ)

        response = response.follow(status=200, extra_environ = environ)

        assert_true(response.lxml.xpath(

            '//div[@id="flash"]/div[@class="error"]'))

        ### 3ème cas : L'utilisateur utilisé pour

        # se connecter à Vigiboard est 'access'.

        environ = {'REMOTE_USER': 'access'}

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

        # et à ce qu'un message informe l'utilisateur que les

        # évènements corrélés sélectionnées ont bien été mis à jour.

                "trouble_ticket" : "",

                "ack" : u'Acknowledged',

            }, status = 302, extra_environ = environ)

        response = response.follow(status=200, extra_environ = environ)

        assert_false(response.lxml.xpath(

            '//div[@id="flash"]/div[@class="error"]'))

        assert_true(response.lxml.xpath(

            '//div[@id="flash"]/div[@class="ok"]'))

        # On s'assure que le statut de l'évènement corrélé

        # a bien été mis à jour dans la base de données.

        correvents = DBSession.query(

        # On peuple la BDD avec 2 hôtes, 2 services de bas niveau,

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

        (hosts, services) = populate_DB()

        # On ajoute 2 évènements corrélés causés par ces hôtes

        timestamp = datetime.now()

        correvent1_id = add_correvent_caused_by(services[0], timestamp)

        correvent2_id = add_correvent_caused_by(services[1], timestamp)

        transaction.commit()

        ### 1er cas : L'utilisateur n'est pas connecté.

        # On vérifie que le plugin retourne bien une erreur 401.

        response = self.app.post(

                "trouble_ticket" : "",

                "ack" : u'Acknowledged',

            }, status = 401)

        ### 2ème cas : L'utilisateur utilisé pour

        # se connecter à Vigiboard est 'limited_access'.

        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 la permission de modifier ces évènements.

                "trouble_ticket" : "",

                "ack" : u'Acknowledged',

            }, status = 302, extra_environ = environ)

        response = response.follow(status=200, extra_environ = environ)

        assert_true(response.lxml.xpath(

            '//div[@id="flash"]/div[@class="error"]'))

        ### 3ème cas : L'utilisateur utilisé pour

        # se connecter à Vigiboard est 'access'.

        environ = {'REMOTE_USER': 'access'}

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

        # et à ce qu'un message informe l'utilisateur que les

        # évènements corrélés sélectionnées ont bien été mis à jour.

                "trouble_ticket" : "",

                "ack" : u'Acknowledged',

            }, status = 302, extra_environ = environ)

        response = response.follow(status=200, extra_environ = environ)

        assert_false(response.lxml.xpath(

            '//div[@id="flash"]/div[@class="error"]'))

        assert_true(response.lxml.xpath(

            '//div[@id="flash"]/div[@class="ok"]'))

        # On s'assure que le statut de l'évènement corrélé

        # a bien été mis à jour dans la base de données.

        correvents = DBSession.query(

        # On peuple la BDD avec 2 hôtes, 2 services de bas niveau,

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

        (hosts, services) = populate_DB()

        # On ajoute 2 évènements corrélés causés par ces hôtes

        timestamp = datetime.now()

        correvent1_id = add_correvent_caused_by(services[0], timestamp)

        correvent2_id = add_correvent_caused_by(services[1], timestamp)

        later_date = datetime.now()

        # Date du chargement de la page

        date = mktime(later_date.timetuple()) - 42

        # On ajoute une entrée dans l'historique de l'évènement brut

        # causant le premier évènement corrélé, portant pour timestamp

        # une date postérieure à celle du chargement de la page.

            idevent = correvent1.idcause,

            timestamp = later_date))

        DBSession.flush()

        transaction.commit()

        # L'utilisateur utilisé pour se connecter à Vigiboard est 'access'.

        environ = {'REMOTE_USER': 'access'}

        # On s'attend à ce que le statut de la requête soit 302, et

        # à ce qu'un message d'erreur avise l'utilisateur que des

        # changements sont intervenus depuis le chargement de la page.

                "trouble_ticket" : "",

                "ack" : u'Acknowledged',

            }, status = 302, extra_environ = environ)

        response = response.follow(status=200, extra_environ = environ)

        assert_true(response.lxml.xpath(

            '//div[@id="flash"]/div[@class="warning"]'))

        # On s'assure que le statut de l'évènement corrélé

        # n'a pas été modifié dans la base de données.

        status = DBSession.query(

            ).filter(CorrEvent.idcorrevent == correvent1_id

            ).scalar()

        assert_equal(status, u'None')