100 lines
2.2 KiB
Python
100 lines
2.2 KiB
Python
from __future__ import annotations
|
|
|
|
import json
|
|
import uuid
|
|
from dataclasses import dataclass
|
|
from enum import Enum
|
|
from typing import Optional, get_type_hints, Dict
|
|
|
|
from megasniff import SchemaDeflatorGenerator
|
|
from src.megasniff import SchemaInflatorGenerator
|
|
|
|
|
|
def test_dicts():
|
|
@dataclass()
|
|
class A:
|
|
a: dict[str, int]
|
|
b: Dict[int, int]
|
|
c: dict[str, list[int]]
|
|
d: dict[str, dict[str, int | str]]
|
|
|
|
infl = SchemaInflatorGenerator(store_sources=True)
|
|
defl = SchemaDeflatorGenerator(store_sources=True)
|
|
|
|
infl_fn = infl.schema_to_inflator(A)
|
|
defl_fn = defl.schema_to_deflator(A)
|
|
|
|
a = infl_fn({
|
|
'a': {
|
|
1: '42',
|
|
2: '123',
|
|
'asdf': 42
|
|
},
|
|
'b': {
|
|
1: 1,
|
|
'2': '2',
|
|
'3': 3,
|
|
4: '4'
|
|
},
|
|
'c': {
|
|
'a': [1, 2, 3, '4']
|
|
},
|
|
'd': {
|
|
'a': {
|
|
'a': 1,
|
|
'b': '1',
|
|
'c': 'asdf'
|
|
}
|
|
}
|
|
})
|
|
|
|
assert a.a['1'] == 42
|
|
assert a.a['2'] == 123
|
|
assert a.a['asdf'] == 42
|
|
|
|
assert a.b[1] == 1
|
|
assert a.b[2] == 2
|
|
assert a.b[3] == 3
|
|
assert a.b[4] == 4
|
|
|
|
assert a.c['a'][0] == 1
|
|
assert a.c['a'][1] == 2
|
|
assert a.c['a'][2] == 3
|
|
assert a.c['a'][3] == 4
|
|
|
|
assert a.d['a']['a'] == 1
|
|
assert a.d['a']['b'] == 1
|
|
assert a.d['a']['c'] == 'asdf'
|
|
|
|
|
|
def test_uuid_dicts():
|
|
@dataclass()
|
|
class A:
|
|
a: dict[uuid.UUID, uuid.UUID]
|
|
|
|
infl = SchemaInflatorGenerator(store_sources=True)
|
|
defl = SchemaDeflatorGenerator(store_sources=True)
|
|
|
|
infl_fn = infl.schema_to_inflator(A)
|
|
defl_fn = defl.schema_to_deflator(A)
|
|
|
|
uuids = [uuid.uuid4() for _ in range(32)]
|
|
|
|
a = infl_fn({
|
|
'a': {
|
|
str(uuids[0]): str(uuids[0]),
|
|
str(uuids[1]): str(uuids[2]),
|
|
str(uuids[3]): str(uuids[4]),
|
|
}
|
|
})
|
|
|
|
assert a.a[uuids[0]] == uuids[0]
|
|
assert a.a[uuids[1]] == uuids[2]
|
|
assert a.a[uuids[3]] == uuids[4]
|
|
|
|
ad = json.loads(json.dumps(defl_fn(a), default=str))
|
|
|
|
assert ad['a'][str(uuids[0])] == str(uuids[0])
|
|
assert ad['a'][str(uuids[1])] == str(uuids[2])
|
|
assert ad['a'][str(uuids[3])] == str(uuids[4])
|