File: test_params_models.py

package info (click to toggle)
django-ninja 1.6.0-1
  • links: PTS, VCS
  • area: main
  • in suites:
  • size: 11,280 kB
  • sloc: python: 15,956; javascript: 1,689; makefile: 39; sh: 25
file content (23 lines) | stat: -rw-r--r-- 609 bytes parent folder | download | duplicates (2)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
from typing import Optional

from ninja.params.models import DictStrAny, ParamModel


class _NestedParamModel(ParamModel):
    outer: DictStrAny
    leaf: Optional[int]

    __ninja_flatten_map__ = {
        "foo": ("outer", "foo"),
        "bar": ("outer", "bar"),
        "leaf": ("leaf",),
    }


def test_map_data_paths_creates_parent_for_missing_nested_values():
    assert _NestedParamModel._map_data_paths({}) == {"outer": {}}


def test_map_data_paths_sets_values_when_present():
    data = _NestedParamModel._map_data_paths({"foo": 1, "leaf": 2})
    assert data == {"outer": {"foo": 1}, "leaf": 2}