1 files changed, 41 insertions, 0 deletions
diff --git a/testing-scheduler/server/src/step/step_manager.py b/testing-scheduler/server/src/step/step_manager.py
new file mode 100644
index 00000000..8d76c67c
--- /dev/null
+++ b/testing-scheduler/server/src/step/step_manager.py
@@ -0,0 +1,41 @@
+##############################################################################
+# Copyright (c) 2018 HUAWEI TECHNOLOGIES CO.,LTD and others.
+#
+# All rights reserved. This program and the accompanying materials
+# are made available under the terms of the Apache License, Version 2.0
+# which accompanies this distribution, and is available at
+# http://www.apache.org/licenses/LICENSE-2.0
+##############################################################################
+
+from src.step.test_step import TestStep
+import os
+import sys
+
+
+class TestStepManager(object):
+    def __init__(self, context):
+        self._context = context
+
+        currentDirPath = os.path.dirname(os.path.abspath(__file__))
+        sys.path.append(currentDirPath)
+
+        excludeFiles = ('__init__.py', 'step_manager.py', 'test_step.py')
+        for fileName in os.listdir(currentDirPath):
+            if os.path.isfile(os.path.join(currentDirPath, fileName)) and \
+                os.path.splitext(fileName)[1] == '.py' and \
+                    fileName not in excludeFiles:
+                __import__(os.path.splitext(fileName)[0])
+
+    def getStepObj(self, type, id, name, service, action, args):
+        for subclass in TestStep.__subclasses__():
+            if type == subclass.__step_type__:
+                return subclass(id, name, service, action, args, self._context)
+
+
+if __name__ == "__main__":
+    tsMgr = TestStepManager()
+    args = {'command': 'greet', 'method': 'POST', 'args': {'name': 'leo'}}
+    stepObj = tsMgr.getStepObj('test', 1, 'test_cpu', {
+        'name': 'greet', 'call': 'REST'}, 'start', args)
+    print stepObj
+    print stepObj.__class__.__mro__