from vigilo.models.session import DBSession

from vigilo.models.tables import Permission, DataPermission, StateName, \

                            Event, CorrEvent, ImpactedPath, ImpactedHLS, \

                            User, UserGroup

from vigiboard.tests import TestController

def populate_DB():

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

    # On ajoute un groupe d'hôtes

    # On lui octroie les permissions

    usergroup = UserGroup.by_group_name(u'users_with_access')

    # On crée un hôte de test.

    host = Host(

        checkhostcmd = u'halt',

        snmpcommunity = u'public',

        hosttpl = u'/dev/null',

    # On ajoute un évènement corrélé causé par cet évènement 'brut'.

    aggregate = CorrEvent(

        timestamp_active = datetime.now(),

        priority = 1,

        status = u'None')

    aggregate.events.append(event1)

    DBSession.add(aggregate)

    DBSession.flush()

    transaction.commit()

    return aggregate

def add_paths(path_number, path_length, idsupitem):

    Ajoute path_number chemins de services de haut niveau impactés

    dans la base de donnée. Leur longeur sera égale à path_length.

    La 3ème valeur passée en paramètre est l'id du supitem impactant.

    haut niveau sont créés dans l'opération.

    """

    # Création des chemins de services de haut niveau impactés.

    for j in range(path_number):

        # On crée le chemin en lui-même

        path = ImpactedPath(idsupitem = idsupitem)

        DBSession.add(path)

        DBSession.flush()

        # Pour chaque étage du chemin,

        for i in range(path_length):

            # on ajoute un service de haut niveau dans la BDD,

            hls = HighLevelService(

                **hls_template)

            DBSession.add(hls)

            DBSession.add(

                ImpactedHLS(

                    path = path,

                    hls = hls,

                    distance = i + 1,

                    ))

    DBSession.flush()

    transaction.commit()

class TestHLSPlugin(TestController):

    """

    de haut niveau impactés par un évènement corrélé.

    """

    def setUp(self):

        Teste la valeur de retour du plugin lorsque

        aucun service de haut niveau n'est impacté.

        """

        # On peuple la base de données avant le test.

        DBSession.add(self.aggregate)

        add_paths(0, 0, self.aggregate.events[0].idsupitem)

        DBSession.add(self.aggregate)

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

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

        resp = self.app.post(

            {"idcorrevent" : str(self.aggregate.idcorrevent),

             "plugin_name" : "hls"},

            status = 401,)

        ### 2ème cas : l'utilisateur n'a pas les

        ### droits sur l'hôte ayant causé le correvent.

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

        resp = self.app.post(

            {"idcorrevent" : str(self.aggregate.idcorrevent),

             "plugin_name" : "hls"},

            status = 404,

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

        ### 3ème cas : l'utilisateur a cette fois les droits.

        resp = self.app.post(

            {"idcorrevent" : str(self.aggregate.idcorrevent),

             "plugin_name" : "hls"},

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

        # On vérifie que le plugin ne retourne toujours rien.

        assert_equal(resp.json, {"services": []})

    def test_1_impacted_hls_path(self):

        """

        Retour du plugin HLS pour 1 chemin impacté

        Teste la valeur de retour du plugin lorsqu'un

        chemin de services de haut niveau est impacté.

        """

        # On peuple la base de données avant le test.

        DBSession.add(self.aggregate)

        add_paths(1, 2, self.aggregate.events[0].idsupitem)

        DBSession.add(self.aggregate)

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

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

        resp = self.app.post(

            {"idcorrevent" : str(self.aggregate.idcorrevent),

             "plugin_name" : "hls"},

            status = 401,)

        ### 2ème cas : l'utilisateur n'a pas les droits

        ### sur l'hôte ayant causé le correvent, on doit

        ### obtenir une erreur 404 (pas d'événement trouvé

        ### avec les informations liées à cet utilisateur).

        resp = self.app.post(

            {"idcorrevent" : str(self.aggregate.idcorrevent),

             "plugin_name" : "hls"},

            status = 404,

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

        ### 3ème cas : l'utilisateur a cette fois les droits.

        resp = self.app.post(

            {"idcorrevent" : str(self.aggregate.idcorrevent),

             "plugin_name" : "hls"},

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

        # On vérifie que le plugin retourne bien les 2 HLS impactés.

        assert_equal(resp.json, {"services": ['HLS12']})

    def test_2_impacted_hls_path(self):

        """

        Retour du plugin HLS pour 2 chemins impactés

        Teste la valeur de retour du plugin lorsque deux

        chemins de services de haut niveau sont impactés.

        """

        # On peuple la base de données avant le test.

        DBSession.add(self.aggregate)

        add_paths(2, 2, self.aggregate.events[0].idsupitem)

        DBSession.add(self.aggregate)

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

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

        resp = self.app.post(

            {"idcorrevent" : str(self.aggregate.idcorrevent),

             "plugin_name" : "hls"},

            status = 401,)

        ### 2ème cas : l'utilisateur n'a pas les

        ### droits sur l'hôte ayant causé le correvent.

        resp = self.app.post(

            {"idcorrevent" : str(self.aggregate.idcorrevent),

             "plugin_name" : "hls"},

            status = 404,

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

        ### 3ème cas : l'utilisateur a cette fois les droits.

        resp = self.app.post(

            {"idcorrevent" : str(self.aggregate.idcorrevent),

             "plugin_name" : "hls"},

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

        # On vérifie que le plugin retourne bien les 4 HLS impactés.

        assert_equal(resp.json, {"services": ['HLS12', 'HLS22']})