mirror of
https://github.com/arc53/DocsGPT
synced 2024-11-02 03:40:17 +00:00
51 lines
1.5 KiB
Python
51 lines
1.5 KiB
Python
import pytest
|
|
from openapi_parser import parse
|
|
from application.parser.file.openapi3_parser import OpenAPI3Parser
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"urls, expected_base_urls",
|
|
[
|
|
(
|
|
[
|
|
"http://petstore.swagger.io/v1",
|
|
"https://api.example.com/v1/resource",
|
|
"https://api.example.com/v1/another/resource",
|
|
"https://api.example.com/v1/some/endpoint",
|
|
],
|
|
["http://petstore.swagger.io", "https://api.example.com"],
|
|
),
|
|
],
|
|
)
|
|
def test_get_base_urls(urls, expected_base_urls):
|
|
assert OpenAPI3Parser().get_base_urls(urls) == expected_base_urls
|
|
|
|
|
|
def test_get_info_from_paths():
|
|
file_path = "tests/test_openapi3.yaml"
|
|
data = parse(file_path)
|
|
path = data.paths[1]
|
|
assert (
|
|
OpenAPI3Parser().get_info_from_paths(path)
|
|
== "\nget=Expected response to a valid request"
|
|
)
|
|
|
|
|
|
def test_parse_file():
|
|
file_path = "tests/test_openapi3.yaml"
|
|
results_expected = (
|
|
"Base URL:http://petstore.swagger.io,https://api.example.com\nPath1: "
|
|
+ "/pets\ndescription: None\nparameters: []\nmethods: \n"
|
|
+ "get=A paged array of pets\npost=Null "
|
|
+ "response\nPath2: /pets/{petId}\ndescription: None\n"
|
|
+ "parameters: []\nmethods: "
|
|
+ "\nget=Expected response to a valid request\n"
|
|
)
|
|
openapi_parser_test = OpenAPI3Parser()
|
|
results = openapi_parser_test.parse_file(file_path)
|
|
assert results == results_expected
|
|
|
|
|
|
if __name__ == "__main__":
|
|
pytest.main()
|