mirror of
https://github.com/open-metadata/OpenMetadata.git
synced 2025-07-04 07:33:56 +00:00
23 lines
556 B
Python
23 lines
556 B
Python
|
"""Utils functions for load testing."""
|
||
|
|
|
||
|
|
import sys
|
||
|
|
from typing import List
|
||
|
|
|
||
|
|
import pytest
|
||
|
|
from locust import main
|
||
|
|
|
||
|
|
TEST_CASE_RESOURCE_PATH = "/api/v1/dataQuality/testCases"
|
||
|
|
TEST_CASE_RESULT_RESOURCE_PATH = "/api/v1/dataQuality/testCases/testCaseResults"
|
||
|
|
|
||
|
|
|
||
|
|
def run_load_test(args: List[str]):
|
||
|
|
"""Test test case result resource"""
|
||
|
|
original_argv = sys.argv
|
||
|
|
try:
|
||
|
|
sys.argv = args
|
||
|
|
with pytest.raises(SystemExit) as excinfo:
|
||
|
|
main.main()
|
||
|
|
assert excinfo.value.code == 0
|
||
|
|
finally:
|
||
|
|
sys.argv = original_argv
|