From e32043f58a2450b6a5986dc2a01f64f8b22c3992 Mon Sep 17 00:00:00 2001
From: "Yang (Gabriel) Yu" <Gabriel.yuyang@huawei.com>
Date: Tue, 4 Sep 2018 21:02:27 +0800
Subject: Change naming and veriy test-scheduler function

Changes:
1. Testing-scheduler -> Test-scheduler
2. lots of windows breaks '\r' and '^M' in files,
   batch changes to unix breaks '$'
3. Add ui/build

Change-Id: I1f2c98ab9348460d4e68bfbfab664dae82b761ba
Signed-off-by: Yang (Gabriel) Yu <Gabriel.yuyang@huawei.com>
---
 test-scheduler/server/src/step/workload.py | 46 ++++++++++++++++++++++++++++++
 1 file changed, 46 insertions(+)
 create mode 100644 test-scheduler/server/src/step/workload.py

(limited to 'test-scheduler/server/src/step/workload.py')

diff --git a/test-scheduler/server/src/step/workload.py b/test-scheduler/server/src/step/workload.py
new file mode 100644
index 00000000..b35819e1
--- /dev/null
+++ b/test-scheduler/server/src/step/workload.py
@@ -0,0 +1,46 @@
+##############################################################################
+# 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
+##############################################################################
+
+import json
+import os
+from src.step.test_step import TestStep
+
+
+class WorkloadStep(TestStep):
+    __step_type__ = 'workload'
+
+    def __init__(self, id, name, service, action, args):
+        super(WorkloadStep, self).__init__(
+            self.__step_type__, id, name, service, action, args)
+        self._argsParse()
+        self._action()
+
+    def _argsParse(self):
+        if self._callType == "REST":
+            currentDirPath = os.path.dirname(os.path.abspath(__file__))
+            envDirPath = os.path.abspath(
+                os.path.join(currentDirPath, os.pardir, os.pardir, 'env'))
+            envFilePath = os.path.join(
+                envDirPath, "%s.json" % self._service['name'])
+            with open(envFilePath) as f:
+                propDict = json.load(f)
+                self._args['ip'] = propDict['ip']
+                self._args['port'] = propDict['port']
+                self._args['api'] = "%s/%s" % (
+                    propDict['api_map']['workload'], self._args['command'])
+                exclude = {'ip', 'port', 'api', 'command', 'method'}
+                self._args['req_body'] = {
+                    key: value for key, value in
+                    self._args.items() if key not in exclude}
+
+    def _start(self):
+        print "workload start"
+
+    def _stop(self):
+        print "workload stop"
-- 
cgit 1.2.3-korg