openapi-python-client/tests/test_config.py at main · e2b-dev/openapi-python-client · GitHub

History

61 lines (50 loc) · 2 KB

Raw

import json

import os

from io import StringIO

from pathlib import Path

from typing import Any

import pytest

from ruamel.yaml import YAML as _YAML

from openapi_python_client.config import ConfigFile

class YAML(_YAML):

def dump_to_string(self, data: Any, **kwargs: Any) -> str:

stream = StringIO()

self.dump(data=data, stream=stream, **kwargs)

return stream.getvalue()

yaml = YAML(typ=["safe", "string"])

def json_with_tabs(d: dict) -> str:

return json.dumps(d, indent=4).replace(" ", "\t")

@pytest.mark.parametrize(

"filename,dump",

[

("example.yml", yaml.dump_to_string),

("example.json", json.dumps),

("example.yaml", yaml.dump_to_string),

("example.json", json_with_tabs),

)

@pytest.mark.parametrize("relative", (True, False), ids=("relative", "absolute"))

def test_load_from_path(tmp_path: Path, filename, dump, relative) -> None:

yml_file = tmp_path.joinpath(filename)

if relative:

if not os.getenv("TEST_RELATIVE"):

pytest.skip("Skipping relative path checks")

return

yml_file = yml_file.relative_to(Path.cwd())

override1 = {"class_name": "ExampleClass", "module_name": "example_module"}

override2 = {"class_name": "DifferentClass", "module_name": "different_module"}

data = {

"field_prefix": "blah",

"class_overrides": {"Class1": override1, "Class2": override2},

"project_name_override": "project-name",

"package_name_override": "package_name",

"package_version_override": "package_version",

}

yml_file.write_text(dump(data))

config = ConfigFile.load_from_path(yml_file)

assert config.field_prefix == "blah"

assert config.class_overrides["Class1"].model_dump() == override1

assert config.class_overrides["Class2"].model_dump() == override2

assert config.project_name_override == "project-name"

assert config.package_name_override == "package_name"

assert config.package_version_override == "package_version"

openapi-python-client/tests/test_config.py at main · e2b-dev/openapi-python-client · GitHub

Uh oh!

Expand file tree

Breadcrumbs

test_config.py

Latest commit

History

Breadcrumbs

test_config.py

File metadata and controls