1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146
|
import sys
from unittest import TestCase
from xsdata.codegen.models import Restrictions
class RestrictionsTests(TestCase):
def setUp(self) -> None:
self.restrictions = Restrictions(
min_occurs=1,
max_occurs=1,
min_exclusive="1.1",
min_inclusive="1",
min_length=1,
max_exclusive="1",
max_inclusive="1.1",
max_length=10,
total_digits=333,
fraction_digits=2,
length=5,
white_space="collapse",
pattern=r"[A-Z]",
explicit_timezone="+1",
nillable=True,
path=[("s", 0, 1, 1)],
process_contents="skip",
)
def test_property_is_list(self):
restrictions = Restrictions()
self.assertFalse(restrictions.is_list)
restrictions.max_occurs = 1
self.assertFalse(restrictions.is_list)
restrictions.max_occurs = 2
self.assertTrue(restrictions.is_list)
def test_property_is_prohibited(self):
self.assertFalse(Restrictions().is_prohibited)
self.assertTrue(Restrictions(max_occurs=0).is_prohibited)
def test_merge(self):
source = Restrictions(min_length=2, max_length=10, format="base16")
target = Restrictions(min_length=1, pattern=r"[A-Z]")
target.merge(source)
self.assertEqual(2, target.min_length)
self.assertEqual(10, target.max_length)
self.assertEqual(r"[A-Z]", target.pattern)
self.assertEqual("base16", target.format)
def test_merge_ignore_nillable(self):
parent = Restrictions(nillable=True)
child = Restrictions()
child.merge(parent)
self.assertIsNone(child.nillable)
child.nillable = False
child.merge(parent)
self.assertFalse(child.nillable)
def test_merge_occurs(self):
a = Restrictions()
b = Restrictions()
a.merge(b)
self.assertIsNone(a.min_occurs)
self.assertIsNone(a.max_occurs)
b.min_occurs = 0
b.max_occurs = 1
a.merge(b)
self.assertEqual(0, a.min_occurs)
self.assertEqual(1, a.max_occurs)
def test_asdict(self):
expected = {
"explicit_timezone": "+1",
"fraction_digits": 2,
"length": 5,
"max_exclusive": "1",
"max_inclusive": "1.1",
"max_length": 10,
"min_exclusive": "1.1",
"min_inclusive": "1",
"min_length": 1,
"nillable": True,
"pattern": "[A-Z]",
"process_contents": "skip",
"total_digits": 333,
"white_space": "collapse",
}
self.assertEqual(expected, self.restrictions.asdict())
self.restrictions.nillable = None
del expected["nillable"]
expected["required"] = True
self.assertEqual(expected, self.restrictions.asdict())
def test_asdict_with_types(self):
expected = {
"explicit_timezone": "+1",
"fraction_digits": 2,
"length": 5,
"max_exclusive": 1.0, # str -> float
"max_inclusive": 1.1, # str -> float
"max_length": 10,
"min_exclusive": 1.1, # str -> float
"min_inclusive": 1.0, # str -> float
"min_length": 1,
"nillable": True,
"pattern": "[A-Z]",
"process_contents": "skip",
"total_digits": 333,
"white_space": "collapse",
}
self.assertEqual(expected, self.restrictions.asdict(types=[float]))
def test_asdict_with_implied_types(self):
restrictions = Restrictions(min_occurs=1, max_occurs=4)
self.assertEqual({"max_occurs": 4, "min_occurs": 1}, restrictions.asdict())
restrictions.min_occurs = 0
self.assertEqual({"max_occurs": 4}, restrictions.asdict())
restrictions.max_occurs = sys.maxsize
self.assertEqual({}, restrictions.asdict())
def test_asdict_with_process_contents(self):
restrictions = Restrictions(process_contents="skip")
self.assertEqual({"process_contents": "skip"}, restrictions.asdict())
restrictions.process_contents = "strict"
self.assertEqual({}, restrictions.asdict())
def test_clone(self):
restrictions = Restrictions(max_occurs=2)
clone = restrictions.clone()
self.assertEqual(clone, restrictions)
self.assertIsNot(clone, restrictions)
|