add unit tests for schema

tests/test_schema.py

@@ -0,0 +1,115 @@
+import pytest
+
+import marshmallow
+from marshmallow_dataclass import dataclass
+
+from thinq2 import schema
+
+
+def test_base_schema_ignores_unknown_fields():
+    model = schema.BaseSchema().load(dict(foo="bar"))
+    assert model == {}
+
+
+def test_controller_can_be_constructed_from_dict(Controller, valid_data):
+    controller = Controller(valid_data)
+    assert isinstance(controller, Controller)
+
+
+def test_controller_can_be_constructed_from_kwargs(Controller, valid_data):
+    controller = Controller(**valid_data)
+    assert isinstance(controller, Controller)
+
+
+def test_controller_can_be_constructed_from_model(Controller, Model, valid_data):
+    model = Model(**valid_data)
+    controller = Controller(model)
+    assert isinstance(controller, Controller)
+
+
+def test_controller_as_dict_equals_model_data(Controller, valid_data):
+    controller = Controller(valid_data)
+    assert vars(controller) == valid_data
+
+
+def test_controller_constructor_styles_are_equivalent(Controller, Model, valid_data):
+    controller_from_dict = Controller(valid_data)
+    controller_from_kwargs = Controller(**valid_data)
+    controller_from_model = Controller(Model(**valid_data))
+    assert vars(controller_from_dict) == vars(controller_from_kwargs)
+    assert vars(controller_from_dict) == vars(controller_from_model)
+
+
+def test_controller_instances_with_identical_data_are_not_equal(Controller, valid_data):
+    controller_a = Controller(valid_data)
+    controller_b = Controller(valid_data)
+    assert controller_a != controller_b
+
+
+def test_initialzer_decorator_sets_default_value(Controller, data_with_missing_quux):
+    quux = 43
+
+    class QuuxedController(Controller):
+        @schema.initializer
+        def quux(self):
+            return quux
+
+    controller = QuuxedController(**data_with_missing_quux)
+    assert controller.quux == quux
+
+
+def test_initialzer_ignored_when_value_supplied(Controller, data_with_missing_quux):
+    quux = 43
+    data = {**data_with_missing_quux, **dict(quux=quux + 1)}
+
+    class QuuxedController(Controller):
+        @schema.initializer
+        def quux(self):
+            return quux
+
+    controller = QuuxedController(data)
+    assert controller.quux != quux
+
+
+# XXX - need proper test of child controllers for nested fields
+def test_child_controller_decoration(Controller, valid_data):
+    pass
+
+
+@pytest.fixture
+def Model():
+    @dataclass
+    class Model:
+        foo: str
+        quux: int
+
+    return Model
+
+
+@pytest.fixture
+def Controller(Model):
+    @schema.controller(Model)
+    class Controller:
+        pass
+
+    return Controller
+
+
+@pytest.fixture
+def valid_data():
+    return dict(foo="bar", quux=42)
+
+
+@pytest.fixture
+def data_with_extra_field():
+    return dict(foo="bar", quux=42, boz=True)
+
+
+@pytest.fixture
+def data_with_missing_quux():
+    return dict(foo="bar")
+
+
+@pytest.fixture
+def data_with_wrong_quux_type():
+    return dict(foo="bar", quux="42")

thinq2/schema.py

@@ -85,6 +85,7 @@ def load(cls, data):
         def __dict__(self):
             return schema.dump(self._data)
 
+
         # XXX - should throw attribute exception if attr not in schema
         def __getattr__(self, attr):
             if not attr.startswith("_"):