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
|
from . import base
from genson import SchemaBuilder
SCHEMA_URI = 'https://json-schema.org/draft/2020-12/schema'
class TestParams(base.SchemaBuilderTestCase):
def test_uri(self):
self.builder = SchemaBuilder(schema_uri=SCHEMA_URI)
self.assertResult({"$schema": SCHEMA_URI})
def test_null_uri(self):
self.builder = SchemaBuilder(schema_uri=None)
self.assertResult({})
class TestMethods(base.SchemaBuilderTestCase):
def test_add_schema(self):
self.add_schema({"type": "null"})
self.assertResult({
"$schema": SchemaBuilder.DEFAULT_URI,
"type": "null"})
def test_add_object(self):
self.add_object(None)
self.assertResult({
"$schema": SchemaBuilder.DEFAULT_URI,
"type": "null"})
def test_to_json(self):
self.assertEqual(
self.builder.to_json(),
'{"$schema": "%s"}' % SchemaBuilder.DEFAULT_URI)
def test_add_schema_with_uri_default(self):
self.add_schema({"$schema": SCHEMA_URI, "type": "null"})
self.assertResult({"$schema": SCHEMA_URI, "type": "null"})
def test_add_schema_with_uri_not_defuult(self):
self.builder = SchemaBuilder(schema_uri=SCHEMA_URI)
self.add_schema({"$schema": 'BAD_URI', "type": "null"})
self.assertResult({"$schema": SCHEMA_URI, "type": "null"})
def test_empty_falsy(self):
self.assertIs(bool(self.builder), False)
def test_full_truty(self):
self.add_object(None)
self.assertIs(bool(self.builder), True)
class TestInteraction(base.SchemaBuilderTestCase):
def test_add_other(self):
other = SchemaBuilder(schema_uri=SCHEMA_URI)
other.add_object(1)
self.add_object('one')
self.add_schema(other)
self.assertResult({
"$schema": SCHEMA_URI,
"type": ["integer", "string"]})
def test_add_other_no_uri_overwrite(self):
other = SchemaBuilder()
other.add_object(1)
self.add_object('one')
self.add_schema(other)
self.add_schema({'$schema': SCHEMA_URI})
self.assertResult({
"$schema": SCHEMA_URI,
"type": ["integer", "string"]})
def test_eq(self):
b1 = SchemaBuilder()
b1.add_object(1)
b2 = SchemaBuilder()
b2.add_object(1)
self.assertEqual(b1, b2)
def test_ne(self):
b1 = SchemaBuilder()
b1.add_object(1)
b2 = SchemaBuilder()
b2.add_object('one')
self.assertNotEqual(b1, b2)
def test_eq_after_serialization(self):
b1 = SchemaBuilder()
b1.add_object({"bar": 10, "foo": 20})
b2 = SchemaBuilder()
b2.add_schema(b1)
self.assertEqual(b1, b2)
def test_eq_empty_required(self):
b1 = SchemaBuilder()
b1.add_schema({
"type": "object",
"properties": {
"bar": {"type": "integer"},
"foo": {"type": "integer"}},
"required": []})
b2 = SchemaBuilder()
b2.add_schema(b1)
self.assertEqual(b1, b2)
|
