NFVBENCH-102 NFVBench won't work with external chain

[nfvbench.git] / nfvbench / chain_managers.py

diff --git a/nfvbench/chain_managers.py b/nfvbench/chain_managers.py
index 211f9bb..5882913 100644 (file)
--- a/nfvbench/chain_managers.py
+++ b/nfvbench/chain_managers.py
@@ -56,7 +56,7 @@ class StageManager(object):
         return self.client.ports
 
     def get_compute_nodes(self):
-        return self.client.compute_nodes
+        return self.client.compute_nodes if self.client else {}
 
     def set_vm_macs(self):
         if self.client and self.config.service_chain != ChainType.EXT:
@@ -81,7 +81,10 @@ class PVPStatsManager(object):
         self._setup()
 
     def set_vlan_tag(self, device, vlan):
-        self.worker.set_vlan_tag(device, vlan)
+        if self.worker:
+            self.worker.set_vlan_tag(device, vlan)
+        else:
+            device.set_vlan_tag(vlan)
 
     def _setup(self):
         WORKER_CLASS = self.factory.get_chain_worker(self.specs.openstack.encaps,