update locusts with HttpRunner 2.0

This commit is contained in:
httprunner
2018-11-23 17:14:43 +08:00
parent e2c5cacc2b
commit b103fd0d2b
7 changed files with 73 additions and 101 deletions

View File

@@ -1,6 +1,7 @@
import os
import shutil
import time
import unittest
from httprunner import HttpRunner, api, loader, parser
from locust import HttpLocust
@@ -564,4 +565,16 @@ class TestApi(ApiServerUnittest):
self.assertEqual(
results.records[1]["name"],
"create user 1001 and check result."
)
)
class TestLocust(unittest.TestCase):
def test_prepare_locust_tests(self):
path = os.path.join(
os.getcwd(), 'tests/locust_tests/demo_simple_locust.yml')
locust_tests = api.prepare_locust_tests(path)
self.assertIn("gen_md5", locust_tests["functions"])
self.assertEqual(len(locust_tests["tests"]), 10)
self.assertEqual(locust_tests["tests"][0]["name"], "index")
self.assertEqual(locust_tests["tests"][9]["name"], "user-agent")