1
|
|
|
from fastest.fuzzers.rest_fuzzer.json_schema import get_schema_for |
2
|
|
|
from fastest.fuzzers.rest_fuzzer.json_schema import get_schema_for_dict |
3
|
|
|
from fastest.fuzzers.rest_fuzzer.json_schema import get_schema_for_list |
4
|
|
|
from fastest.fuzzers.rest_fuzzer.json_schema import get_schema_for_primitive |
5
|
|
|
from fastest.fuzzers.rest_fuzzer.json_schema import make_schema_object |
6
|
|
|
from fastest.fuzzers.rest_fuzzer.json_schema import schema_fn |
7
|
|
|
from fastest.fuzzers.rest_fuzzer.json_schema import type_name |
8
|
|
|
import unittest |
9
|
|
|
|
10
|
|
|
|
11
|
|
|
class TestJsonSchemaTypeName(unittest.TestCase): |
12
|
|
|
def test__type_name__321A000F63(self): |
13
|
|
|
self.assertEqual(type_name('apple'), 'str') |
14
|
|
|
|
15
|
|
|
|
16
|
|
|
class TestJsonSchemaSchemaFn(unittest.TestCase): |
17
|
|
|
def test__schema_fn__76ED5A360F(self): |
18
|
|
|
self.assertEqual(schema_fn("str"), get_schema_for_primitive) |
19
|
|
|
|
20
|
|
|
|
21
|
|
|
class TestJsonSchemaGetSchemaForPrimitive(unittest.TestCase): |
22
|
|
|
def test__get_schema_for_primitive__A070E8DF88(self): |
23
|
|
|
self.assertEqual(get_schema_for_primitive('', ''), []) |
24
|
|
|
|
25
|
|
|
|
26
|
|
|
class TestJsonSchemaGetSchemaForList(unittest.TestCase): |
27
|
|
|
def test__get_schema_for_list__76EB9EE50E(self): |
28
|
|
|
schema_object = { |
29
|
|
|
'inner': [{ |
30
|
|
|
'inner': [], |
31
|
|
|
'name': 'key', |
32
|
|
|
'type': 'int' |
33
|
|
|
}], |
34
|
|
|
'name': '__auto__', |
35
|
|
|
'type': 'dict' |
36
|
|
|
} |
37
|
|
|
|
38
|
|
|
self.assertEqual(get_schema_for_list([{"key": 1}], "list"), schema_object) |
39
|
|
|
|
40
|
|
|
|
41
|
|
|
class TestJsonSchemaGetSchemaForDict(unittest.TestCase): |
42
|
|
|
def test__get_schema_for_dict__7F4D19694B(self): |
43
|
|
|
self.assertEqual(get_schema_for_dict({'key': 1}, 'dict'), [{'type': 'int', 'inner': [], 'name': 'key'}]) |
44
|
|
|
|
45
|
|
|
|
46
|
|
|
class TestJsonSchemaGetSchemaFor(unittest.TestCase): |
47
|
|
|
def test__get_schema_for__E131A99969(self): |
48
|
|
|
self.assertEqual(get_schema_for({"key": 1}, 'dict'), [{'type': 'int', 'inner': [], 'name': 'key'}]) |
49
|
|
|
|
50
|
|
|
|
51
|
|
|
class TestJsonSchemaMakeSchemaObject(unittest.TestCase): |
52
|
|
|
def test__make_schema_object__C0A5822480(self): |
53
|
|
|
schema_object = { |
54
|
|
|
'inner': [{'inner': [], 'name': 'key', 'type': 'int'}], |
55
|
|
|
'name': '__root__', |
56
|
|
|
'type': 'dict' |
57
|
|
|
} |
58
|
|
|
|
59
|
|
|
self.assertEqual(make_schema_object({"key": 1}), schema_object) |
60
|
|
|
|