X-Git-Url: https://gerrit.opnfv.org/gerrit/gitweb?a=blobdiff_plain;f=nfvbench%2Ffactory.py;h=0d4b0420d8459ac009d0b4e80e08a7d21d5dc105;hb=87d88eab6fe50101abc607857f13af9273aef80a;hp=35a8c1b9ca32f8d98d671e904f2fbf6f56a1547a;hpb=04a7de082bd221eae3c7004f4e0b99dfa4f8be91;p=nfvbench.git

diff --git a/nfvbench/factory.py b/nfvbench/factory.py
index 35a8c1b..0d4b042 100644
--- a/nfvbench/factory.py
+++ b/nfvbench/factory.py
@@ -13,58 +13,19 @@
 #    License for the specific language governing permissions and limitations
 #    under the License.
 #
+"""Factory for creating worker and config plugin instances."""
 
-from chain_clients import EXTStageClient
-from chain_clients import PVPStageClient
-from chain_clients import PVVPStageClient
-from chain_managers import EXTStatsManager
-from chain_managers import PVPStatsManager
-from chain_managers import PVVPStatsManager
-import chain_workers as workers
-from config_plugin import ConfigPlugin
-from specs import ChainType
-import tor_client
+from . import chain_workers as workers
+from .config_plugin import ConfigPlugin
 
 
 class BasicFactory(object):
-
-    chain_classes = [ChainType.EXT, ChainType.PVP, ChainType.PVVP]
-
-    chain_stats_classes = {
-        ChainType.EXT: EXTStatsManager,
-        ChainType.PVP: PVPStatsManager,
-        ChainType.PVVP: PVVPStatsManager,
-    }
-
-    stage_clients_classes = {
-        ChainType.EXT: EXTStageClient,
-        ChainType.PVP: PVPStageClient,
-        ChainType.PVVP: PVVPStageClient,
-    }
-
-    def get_stats_class(self, service_chain):
-        CLASS = self.chain_stats_classes.get(service_chain, None)
-        if CLASS is None:
-            raise Exception("Service chain '{}' not supported.".format(service_chain))
-
-        return CLASS
-
-    def get_stage_class(self, service_chain):
-        CLASS = self.stage_clients_classes.get(service_chain, None)
-        if CLASS is None:
-            raise Exception("VM Client for chain '{}' not supported.".format(service_chain))
-
-        return CLASS
+    """Basic factory class to be overridden for advanced customization."""
 
     def get_chain_worker(self, encaps, service_chain):
+        """Get a chain worker based on encaps and service chain type."""
         return workers.BasicWorker
 
-    def get_tor_class(self, tor_type, no_tor_access):
-        if no_tor_access or not tor_type:
-            # if no TOR access is required, use basic no-op client
-            tor_type = 'BasicTORClient'
-
-        return getattr(tor_client, tor_type)
-
     def get_config_plugin_class(self):
+        """Get a config plugin."""
         return ConfigPlugin