def populate_DB():
""" Peuple la base de données en vue des tests. """
# On crée quatre hôtes de test.
host1 = functions.add_host(u"host1")
host2 = functions.add_host(u"host2")
host3 = functions.add_host(u"host3")
host4 = functions.add_host(u"host4")
DBSession.flush()
# On ajoute un service sur chaque hôte.
service1 = functions.add_lowlevelservice(host2, u"service1")
service2 = functions.add_lowlevelservice(host1, u"service2")
service3 = functions.add_lowlevelservice(host3, u"service3")
functions.add_lowlevelservice(host4, u"service4")
DBSession.flush()
# On crée un groupe de supitems et on y ajoute 'host1' et 'host4' en vue des
# tests portant sur les permissions
group1 = functions.add_supitemgroup(u"group1")
DBSession.add(group1)
functions.add_host2group(u"host1", u"group1")
functions.add_host2group(u"host4", u"group1")
DBSession.flush()
# On ajoute 2 utilisateurs.
functions.add_user(u"no_rights", u"*****@*****.**", u"no_rights", u"no_rights", u"no_rights")
functions.add_user(u"limited_rights", u"*****@*****.**", u"limited_rights", u"limited_rights", u"limited_rights")
functions.add_usergroup_permission(u"limited_rights", u"vigiboard-silence")
functions.add_supitemgrouppermission(u"group1", u"limited_rights")
DBSession.flush()
# On ajoute 4 règles de mise en silence.
functions.add_silence(
states=[u"UNKNOWN"], host=host1, service=None, user=u"manager", comment=u"foo", date=u"2000-01-01 00:00:00"
)
functions.add_silence(
states=[u"DOWN"],
host=host1,
service=service2,
user=u"unrestricted",
comment=u"bar",
date=u"2000-01-02 00:00:00",
)
functions.add_silence(
states=[u"WARNING", "CRITICAL"],
host=host2,
service=service1,
user=u"unrestricted",
comment=u"baz",
date=u"2000-01-03 00:00:00",
)
functions.add_silence(
states=[u"DOWN"], host=host3, service=None, user=u"manager", comment=u"qux", date=u"2000-01-04 00:00:00"
)
DBSession.flush()
transaction.commit()
def setUp(self):
"""Initialisation avant chaque test."""
super(TestDetailsPluginMapsHostLimited, self).setUp()
# On fait manuellement ce que l'initialisation de VigiMap ferait
# (car on est dans les tests de VigiBoard, pas ceux de VigiMap).
root = functions.add_mapgroup(u'Root')
DBSession.add(Permission(permission_name=u'vigimap-access'))
print "Creation hote, service et cartes"
host = functions.add_host(u'localhost éçà')
functions.add_lowlevelservice(host, u'lls éçà')
sig = functions.add_supitemgroup(u'supitemgroup éçà')
functions.add_host2group(host, sig)
mg = functions.add_mapgroup(u'éçà', root)
m1 = functions.add_map(u'M1', root)
# La seconde carte appartient à "/Root/éçà"
# et permet de tester les accès indirects.
m2 = functions.add_map(u'M2', mg)
m3 = functions.add_map(u'M3', root)
# On ajoute l'hôte 2 fois sur M1 pour vérifier
# l'absense de doublons dans les liens.
print "Preparation cartes"
functions.add_node_host(host, 'h1', m1)
functions.add_node_host(host, 'h2', m1)
functions.add_node_host(host, 'h', m2)
functions.add_node_host(host, 'h', m3)
# Création de quelques utilisateurs.
print "Creation des comptes utilisateurs et reglages permissions"
functions.add_user(u'restricted', u'*****@*****.**',
u'restricted', u'restricted',
u'restricted')
functions.add_user(u'unrestricted', u'*****@*****.**',
u'unrestricted', u'unrestricted',
u'unrestricted')
functions.add_user(u'no_rights', u'*****@*****.**',
u'no_rights', u'no_rights',
u'no_rights')
# Les 3 utilisateurs ont accès à VigiBoard.
functions.add_usergroup_permission(u'no_rights', u'vigiboard-access')
functions.add_usergroup_permission(u'restricted', u'vigiboard-access')
functions.add_usergroup_permission(u'unrestricted', u'vigiboard-access')
# Mais seuls "restricted" et "unrestricted" ont accès à VigiMap.
functions.add_usergroup_permission(u'restricted', u'vigimap-access')
functions.add_usergroup_permission(u'unrestricted', u'vigimap-access')
# Ils voient tous les trois les événements dans VigiBoard...
functions.add_supitemgrouppermission(sig, 'no_rights')
functions.add_supitemgrouppermission(sig, 'restricted')
functions.add_supitemgrouppermission(sig, 'unrestricted')
# ... mais "restricted" ne peut voir que "M2" ...
functions.add_MapGroupPermission(mg, 'restricted')
# ... tandis que "unrestricted" voit les 3 cartes (par héritage).
functions.add_MapGroupPermission(root, 'unrestricted')
DBSession.flush()
transaction.commit()
def populateDB():
"""
Peuple la base de données en ajoutant :
- 3 groupes d'hôtes ;
- 3 hôtes ;
- 3 utilisateurs.
Les objets construits respectent les matrices suivantes.
Matrice des utilisateurs et groupes d'utilisateurs :
+-------------------+-----------+---------------+-------+-----------+
| util. \ groupe | managers | powerusers | users | visitor |
+===================+===========+===============+=======+===========+
| manager | X | | | |
+-------------------+-----------+---------------+-------+-----------+
| poweruser | | X | | |
+-------------------+-----------+---------------+-------+-----------+
| user | | | X | |
+-------------------+-----------+---------------+-------+-----------+
| visitor | | | | X |
+-------------------+-----------+---------------+-------+-----------+
Matrice des groupes d'utilisateurs et des droits
sur les groupes d'objets supervisés :
+---------------------------+-----------+-----------+-----------+
| usergroup \ supitemgroup | mhg | hg1 | hg2 |
| | (host1) | (host2) | (host3) |
+===========================+===========+===========+===========+
| managers | / | / | / |
+---------------------------+-----------+-----------+-----------+
| powerusers | X | / | / |
+---------------------------+-----------+-----------+-----------+
| users | | X | |
+---------------------------+-----------+-----------+-----------+
| visitors | | | |
+---------------------------+-----------+-----------+-----------+
(X = accès explicite, / = accès implicite)
@return: Tuple contenant les trois hôtes créés.
@rtype: C{tuple} of C{vigilo.models.tables.Host}
"""
# Ajout d'un groupe d'hôtes principal
mainhostgroup = add_supitemgroup(u'mhg', None)
# Ajout d'un premier groupe d'hôtes de second niveau
hostgroup1 = add_supitemgroup(u'hg1', mainhostgroup)
# Ajout d'un second groupe d'hôtes de second niveau
hostgroup2 = add_supitemgroup(u'hg2', mainhostgroup)
# Ajout de trois hôtes
# On ajoute des caractères spéciaux pour détecter les
# conversions implicites Unicode <-> ASCII (et leurs erreurs).
host1 = add_host(u'host1 éà')
host2 = add_host(u'host2 éà')
host3 = add_host(u'host3 éà')
# Ajout du premier hôte dans le groupe d'hôtes principal.
add_host2group(host1, mainhostgroup)
# Ajout du deuxième hôte dans le premier
# groupe d'hôtes de second niveau.
add_host2group(host2, hostgroup1)
# Ajout du troisième hôte dans le second
# groupe d'hôtes de second niveau.
add_host2group(host3, hostgroup2)
# Ajout de trois groupes d'utilisateurs
poweruser_group = add_usergroup(u'powerusers')
user_group = add_usergroup(u'users')
visitor_group = add_usergroup(u'visitor')
# Ajout de trois utilisateurs
add_user(u'poweruser', u'*****@*****.**',
u'Power User', u'poweruserpass', u'powerusers')
add_user(u'user', u'*****@*****.**',
u'User', u'userpass', u'users')
add_user(u'visitor', u'*****@*****.**',
u'', u'visitorpass', u'visitor')
# Ajout des permissions sur le groupe d'hôtes
# principal pour le premier groupe d'utilisateurs
add_supitemgrouppermission(mainhostgroup, poweruser_group)
# Ajout des permissions sur le premier groupe d'hôtes
# secondaire pour le second groupe d'utilisateurs
add_supitemgrouppermission(hostgroup1, user_group)
# Ajout de la permission 'vigigraph-access' aux groupes d'utilisateurs
perm = Permission.by_permission_name(u'vigigraph-access')
add_usergroup_permission(poweruser_group, perm)
add_usergroup_permission(user_group, perm)
add_usergroup_permission(visitor_group, perm)
return (host1, host2, host3)
