fix locust in testsuite

tests/locust_tests/demo_locust_with_layer.yml

@@ -0,0 +1,18 @@
+config:
+    name: create users with uid
+    variables:
+        - device_sn: ${gen_random_string(15)}
+    base_url: "http://127.0.0.1:5000"
+
+testcases:
+    create user 1000 and check result.:
+        testcase: testcases/create_and_check.yml
+        weight: 2
+        variables:
+            uid: 1000
+
+    create user 1001 and check result.:
+        testcase: testcases/create_and_check.yml
+        weight: 3
+        variables:
+            uid: 1001

tests/test_api.py

@@ -598,3 +598,12 @@ class TestLocust(unittest.TestCase):
         self.assertEqual(len(locust_tests["tests"]), 10)
         self.assertEqual(locust_tests["tests"][0]["name"], "index")
         self.assertEqual(locust_tests["tests"][9]["name"], "user-agent")
+
+    def test_prepare_locust_tests_with_layer(self):
+        path = os.path.join(
+            os.getcwd(), 'tests/locust_tests/demo_locust_with_layer.yml')
+        locust_tests = prepare_locust_tests(path)
+        self.assertIn("gen_md5", locust_tests["functions"])
+        self.assertEqual(len(locust_tests["tests"]), 4 * 2 + 4 * 3)
+        self.assertIn("setup and reset all (override)", locust_tests["tests"][0]["config"]["name"])
+        self.assertIn("check if user ", locust_tests["tests"][19]["name"])