Implement schema function

features/dict.feature

@@ -0,0 +1,83 @@
+Feature: Test JSON DOM objects
+
+  Scenario: Test good input string
+
+    Given the Python module dict_module.py
+    When we execute the following python code:
+      """
+      example_dict_input = {
+        "first_name": "Marge",
+        "last_name": "Simpson",
+        "current_address": {
+        "first_line": "123 Fake Street",
+        "second_line": "",
+        "city": "Springfield",
+        "postal_code": 58008
+        },
+        "previous_addresses": [{
+          "first_line": "742 Evergreen Terrace",
+          "second_line": "",
+          "city": "Springfield",
+          "postal_code": 58008
+        }],
+        "vehicles": {
+          "eabf04": {
+          "color": "orange",
+          "description": "Station Wagon"
+          }
+        }
+      }
+      example = dict_module.Person(example_dict_input)
+      example_dict_output = example.to_builtin()
+      """
+    Then the following statements are true:
+      """
+      example.first_name == "Marge"
+      example.last_name == "Simpson"
+      example.current_address.first_line == "123 Fake Street"
+      example.current_address.second_line == ""
+      example.current_address.city == "Springfield"
+      example.current_address.postal_code == 58008
+      example["current_address"].first_line == "123 Fake Street"
+      example["current_address"].second_line == ""
+      example["current_address"].city == "Springfield"
+      example["current_address"].postal_code == 58008
+      example.previous_addresses[0].first_line == "742 Evergreen Terrace"
+      example.previous_addresses[0].second_line == ""
+      example.previous_addresses[0].city == "Springfield"
+      example.previous_addresses[0].postal_code == 58008
+      example.vehicles["eabf04"].color == "orange"
+      example.vehicles["eabf04"].description == "Station Wagon"
+      example.vehicles["eabf04"].license == "eabf04"
+      len(example) == 5
+      len(example.current_address) == 4
+      len(example.previous_addresses) == 1
+      len(example.vehicles) == 1
+      parent(example.current_address) is example
+      document(example.current_address) is example
+      key(example.current_address) == "current_address"
+      parent(example.vehicles["eabf04"]) is example.vehicles
+      document(example.vehicles["eabf04"]) is example
+      key(example.vehicles["eabf04"]) == "eabf04"
+      parent(example["vehicles"]["eabf04"]) is example.vehicles
+      document(example["vehicles"]["eabf04"]) is example
+      key(example["vehicles"]["eabf04"]) == "eabf04"
+      schema(example).is_valid(example_dict_input)
+      example_dict_output == example_dict_input
+      """
+
+  Scenario: Test bad input string
+
+    Given the Python module dict_module.py
+    When we execute the following python code:
+      """
+      example_dict_input = {"foo": "bar"}
+      """
+    Then the following statements are true:
+      """
+      not(schema(example).is_valid(example_dict_input))
+      """
+    And the following statement raises ValidationError
+      """
+      dict_module.Person(example_dict_input)
+      """

features/examples/modules/dict_module.py

@@ -0,0 +1,27 @@
+from typing import Dict, List
+
+from wysdom import UserObject, UserProperty, SchemaArray, SchemaDict, key
+
+
+class Vehicle(UserObject):
+    color: str = UserProperty(str)
+    description: str = UserProperty(str)
+
+    @property
+    def license(self):
+        return key(self)
+
+
+class Address(UserObject):
+    first_line: str = UserProperty(str)
+    second_line: str = UserProperty(str)
+    city: str = UserProperty(str)
+    postal_code: str = UserProperty(int)
+
+
+class Person(UserObject):
+    first_name: str = UserProperty(str)
+    last_name: str = UserProperty(str)
+    current_address: Address = UserProperty(Address)
+    previous_addresses: List[Address] = UserProperty(SchemaArray(Address))
+    vehicles: Dict[str, Vehicle] = UserProperty(SchemaDict(Vehicle))

features/steps/steps.py

@@ -1,6 +1,6 @@
 from behave import *
 
-from wysdom import document, parent, key
+from wysdom import document, parent, key, schema
 
 import os
 import importlib.util
@@ -38,6 +38,7 @@ def step_impl(context):
     assert callable(document)
     assert callable(parent)
     assert callable(key)
+    assert callable(schema)
     for line in context.text.splitlines():
         result = eval(line)
         if not result:

wysdom/__init__.py

@@ -1,7 +1,7 @@
 from .__version__ import __version__
 from . import mixins
 from .exceptions import ValidationError
-from .dom.functions import dom, document, parent, key
+from .dom.functions import dom, document, parent, key, schema
 from .dom import DOMInfo as DOMInfo
 from .dom import DOMElement as Element
 from .base_schema import Schema, SchemaAnything, SchemaConst

wysdom/dom/functions.py

@@ -1,5 +1,6 @@
 from typing import Optional
 
+from ..base_schema import Schema
 from .DOMElement import DOMElement
 from . import DOMInfo
 
@@ -18,3 +19,7 @@ def parent(element: DOMElement) -> Optional[DOMElement]:
 
 def key(element: DOMElement) -> Optional[str]:
     return dom(element).element_key
+
+
+def schema(element: DOMElement) -> Schema:
+    return element.__json_schema__()