Ejemplo n.º 1
0
        def on_mngr_msg(client, body, routing_key):

            self.mngr_msg_count += 1

            report = Entity.from_json(body)
            on_report(report)
            self.assertEqual(report.reporter_oid, self.node_oid)

            if self.mngr_msg_count == 1: # got NodeOnline Report
                self.assertEqual(report.__class__.__name__, 'NodeOnlineReport')
                client.send_task(TestConnectTask(node_oid=self.node_oid))

            if self.mngr_msg_count == 2:
                # TaskUpdated reports
                self.assertEqual(report.task.node_oid, self.node_oid)
                self.assertEqual(report.task.status, 'accepted')

            if self.mngr_msg_count == 3:
                self.assertEqual(report.task.status, 'inprogress')
                self.assertEqual(report.task.progress, 50)

            if self.mngr_msg_count == 4:
                self.assertEqual(report.__class__.__name__, 'TaskSuccess')
                self.assertEqual(report.task.status, 'success')
                self.stop()
Ejemplo n.º 2
0
def on_msg(client, body, routing_key):
    try:
        entity = Entity.from_json(body)
        assert isinstance(entity, BaseTask)
        worker = TaskThreadWorker(client, entity)
        worker.start()
    except Exception:
        log.error("on msg processing", exc_info=True)
Ejemplo n.º 3
0
def on_mngr_msg(client, body, routing_key):
    from swarm.reports.base_report import BaseReport
    global CLIENT
    CLIENT = client
    entity = Entity.from_json(body)
    log.debug("got msg %s, client is %s" % (entity.__class__.__name__, CLIENT))
    if isinstance(entity, BaseReport):
        return on_report(entity)
Ejemplo n.º 4
0
 def on_mngr_msg(client, body, routing_key):
     default_mngr_callback(client, body, routing_key)
     event = Entity.from_json(body)
     if isinstance(event, NodeOnlineReport):
         node = Cluster.instance().get(self.node_oid)
         self.assertEqual(len(node.storages), 2)
         for mount_point in node.storages:
             if mount_point.storage_oid == self.storage1_oid:
                 self.assertEqual(mount_point.path, 
                                  self.storage1_path)
         self.stop()
Ejemplo n.º 5
0
 def on_mngr_msg(client, body, routing_key):
     report = Entity.from_json(body)
     on_report(report)
     self.assertEqual(report.reporter_oid, self.node_oid)
     if report.__class__.__name__ == 'NodeOnlineReport':
         on_report(IFConfigReport.create(self.node_oid, 
                                         raw_data=IFCONFIG_DATA))
         client.send_task(VMInventoryTask(node_oid=self.node_oid))
     if report.__class__.__name__ == 'VmXMLReport':
         cluster = Cluster.instance()
         vm_config = cluster.entities_by_class('VmConfig')[0]
         self.assertEqual(vm_config.oid, 'c2127a40-eb4c-4e3c-af5b-ab455fd8bb40')
         self.stop()
Ejemplo n.º 6
0
 def on_mngr_msg(client, body, routing_key):
     event = Entity.from_json(body)
     self.assertEqual(event.__class__.__name__,
                      'NodeOnlineReport')
     self.assertEqual(event.node_oid, self.node_oid)
     self.stop()
Ejemplo n.º 7
0
 def entity_from_json(self, json_str):
     "Return Entity instance for given json str"
     return Entity.from_json(json_str)
Ejemplo n.º 8
0
 def on_node_msg(client, body, routing_key):
     task = Entity.from_json(body)
     self.assertEqual(task.node_oid, self.node_oid)
     worker = TaskThreadWorker(client, task)
     worker.start()