def runTest(self):
"""
Test de la récupération dans la BDD de l'identifiant d'un
item (hôte, service de haut niveau, ou service de bas niveau).
"""
setup_db()
DBSession.flush()
host1 = functions.add_host(u'messagerie')
lls1 = functions.add_lowlevelservice(host1, u'Processes')
hls1 = functions.add_highlevelservice(u'Connexion')
# On vérifie que la fonction get_supitem renvoie bien l'identifiant
# du host1 lorsqu'on lui passe son nom en paramètre.
self.assertEqual(host1.idhost, SupItem.get_supitem(host1.name, None))
# On vérifie que la fonction get_supitem renvoie bien l'identifiant
# du hls1 lorsqu'on lui passe son nom en paramètre.
self.assertEqual(hls1.idservice,
SupItem.get_supitem(None, hls1.servicename))
# On vérifie que la fonction get_supitem renvoie bien l'identifiant
# du lls1 lorsqu'on lui passe son nom en paramètre.
self.assertEqual(lls1.idservice,
SupItem.get_supitem(host1.name, lls1.servicename))
#Nettoyage de la BDD à la fin du test
del host1
del lls1
del hls1
DBSession.rollback()
DBSession.expunge_all()
teardown_db()
def setup(self):
"""Set up the fixture used to test the model."""
setup_db()
try:
print "Class being tested:", self.klass
new_attrs = {}
new_attrs.update(self.attrs)
new_attrs.update(self.do_get_dependencies())
self.obj = self.klass(**new_attrs)
self.obj = DBSession.merge(self.obj)
DBSession.flush()
return self.obj
except:
DBSession.rollback()
raise
def tearDown(self):
"""Nettoyage à l'issue des tests."""
DBSession.rollback()
DBSession.expunge_all()
teardown_db()
super(TestSupItemAbstraction, self).tearDown()
def tearDown(self):
DBSession.rollback()
DBSession.expunge_all()
teardown_db()
transaction.begin()
def teardown_db():
"""Supprime toutes les tables du modèle de la BDD."""
DBSession.expunge_all()
DBSession.rollback()
DBSession.flush()
metadata.drop_all()
def tearDown(self):
"""Tear down the fixture used to test the model."""
del self.obj
DBSession.rollback()
DBSession.expunge_all()
teardown_db()
def tearDown(self):
"""Call after every test case."""
DBSession.rollback()
DBSession.expunge_all()
teardown_db()
