import tempfile
import warnings
from pathlib import Path
from unittest import TestCase
from xsdata import __version__
from xsdata.codegen.exceptions import CodegenError
from xsdata.exceptions import ParserError
from xsdata.models.config import (
ExtensionType,
GeneratorConfig,
GeneratorExtension,
GeneratorOutput,
OutputFormat,
)
class GeneratorConfigTests(TestCase):
def setUp(self) -> None:
self.maxDiff = None
def test_create(self) -> None:
file_path = Path(tempfile.mktemp())
obj = GeneratorConfig.create()
with file_path.open("w") as fp:
obj.write(fp, obj)
expected = (
'\n'
f'\n'
' \n"
" \n"
' \n'
' \n'
' \n'
' \n'
' \n'
" \n"
" \n"
' \n'
' \n'
' \n'
' \n'
' \n'
' \n'
' \n'
' \n'
' \n'
' \n'
' \n'
" \n"
" \n"
"\n"
)
self.assertEqual(expected, file_path.read_text())
file_path.unlink()
def test_read(self) -> None:
existing = (
'\n'
'\n'
' \n"
" \n"
' \n'
" \n"
" \n"
"\n"
)
file_path = Path(tempfile.mktemp())
file_path.write_text(existing, encoding="utf-8")
config = GeneratorConfig.read(file_path)
with file_path.open("w") as fp:
GeneratorConfig.write(fp, config)
expected = (
'\n'
f'\n'
' \n"
" \n"
' \n'
' \n'
' \n'
' \n'
' \n'
" \n"
" \n"
" \n"
"\n"
)
self.assertEqual(expected, file_path.read_text())
def test_read_with_wrong_value(self) -> None:
existing = (
'\n'
'\n'
' \n"
"\n"
)
file_path = Path(tempfile.mktemp())
file_path.write_text(existing, encoding="utf-8")
with self.assertRaises(ParserError):
GeneratorConfig.read(file_path)
def test_format_with_invalid_eq_config(self) -> None:
with warnings.catch_warnings(record=True) as w:
OutputFormat(eq=False, order=True)
self.assertEqual("Enabling eq because order is true", str(w[-1].message))
def test_generic_collections_requires_frozen_false(self) -> None:
with warnings.catch_warnings(record=True) as w:
output = GeneratorOutput(
generic_collections=True, format=OutputFormat(frozen=True)
)
self.assertFalse(output.generic_collections)
self.assertEqual(
"Generic Collections, requires frozen=False, reverting...",
str(w[-1].message),
)
def test_extension_with_invalid_import_string(self) -> None:
cases = [None, "", "bar"]
for case in cases:
with self.assertRaises(CodegenError):
GeneratorExtension(
type=ExtensionType.DECORATOR, import_string=case, class_name="Test"
)
def test_extension_with_invalid_class_name_pattern(self) -> None:
with self.assertRaises(CodegenError):
GeneratorExtension(
type=ExtensionType.DECORATOR, import_string="a.b", class_name="*Foo"
)
def test_extension_with_parent_path(self) -> None:
GeneratorExtension(
type=ExtensionType.DECORATOR,
import_string="a.b",
class_name="Foo",
parent_path=r"Grandpa\.Papa$",
)
def test_extension_with_invalid_parent_path(self) -> None:
with self.assertRaises(CodegenError):
GeneratorExtension(
type=ExtensionType.DECORATOR,
import_string="a.b",
class_name="Foo",
parent_path=r"*Grandpa\.Papa$",
)