Ejemplo n.º 1
0
class TopologyTestHelpers(object):
    @deferred(timeout=60)
    def setUp(self):
        """Initialisation de la BDD préalable à chacun des tests"""
        helpers.setup_db()
        helpers.populate_statename()

        # Création de 5 couples host/service
        self.add_services()

        # On ajoute quelques dépendances entre
        # les services de bas niveau dans la BDD.
        self.add_dependencies()

        # On prépare la topology.
        self.topology = Topology()
        self.topology.generate()
        self.context_factory = helpers.ContextStubFactory()
        self.database = DummyDatabaseWrapper(True, async=False)
        return defer.succeed(None)

    @deferred(timeout=60)
    def tearDown(self):
        """Nettoyage de la BDD à la fin de chaque test"""
        self.context_factory.reset()
        helpers.teardown_db()
        return defer.succeed(None)

    def add_services(self):
        """Création de 5 couples host/service"""
        self.host1 = functions.add_host(u'messagerie')
        self.host2 = functions.add_host(u'firewall')

        self.service1 = functions.add_lowlevelservice(self.host1, u'Processes')
        self.service2 = functions.add_lowlevelservice(self.host1, u'CPU')
        self.service3 = functions.add_lowlevelservice(self.host1, u'RAM')
        self.service4 = functions.add_lowlevelservice(self.host1, u'Interface eth0')
        self.service5 = functions.add_lowlevelservice(self.host2, u'Interface eth1')

    def add_dependencies(self):
        """
        Ajout de quelques dépendances entre services de bas
        niveau dans la BDD, préalable à certains des test.
        """
        dep_group1 = functions.add_dependency_group(None, self.service1, u'topology', u'|')
        dep_group2 = functions.add_dependency_group(None, self.service2, u'topology', u'|')
        dep_group3 = functions.add_dependency_group(None, self.service3, u'topology', u'|')
        dep_group4 = functions.add_dependency_group(None, self.service4, u'topology', u'|')

        self.dependency1 = functions.add_dependency(dep_group1, self.service2, 1)
        self.dependency2 = functions.add_dependency(dep_group1, self.service3, 1)
        self.dependency3 = functions.add_dependency(dep_group2, self.service4, 1)
        self.dependency4 = functions.add_dependency(dep_group3, self.service4, 1)
        self.dependency5 = functions.add_dependency(dep_group4, self.service5, 1)
Ejemplo n.º 2
0
    def setUp(self):
        """Initialisation de la BDD préalable à chacun des tests"""
        helpers.setup_db()
        helpers.populate_statename()

        # Création de 5 couples host/service
        self.add_services()

        # On ajoute quelques dépendances entre
        # les services de bas niveau dans la BDD.
        self.add_dependencies()

        # On prépare la topology.
        self.topology = Topology()
        self.topology.generate()
        self.context_factory = helpers.ContextStubFactory()
        self.database = DummyDatabaseWrapper(True, async=False)
        return defer.succeed(None)