Added basic marshalling tests

Author: Joel Collins

tests/test_marshalling.py

@@ -0,0 +1,29 @@
+from labthings.marshalling import marshalling as ms
+from labthings.schema import Schema
+from labthings import fields
+
+
+def test_schema_to_converter_schema():
+    class TestSchema(Schema):
+        foo = fields.String()
+
+    test_schema = TestSchema()
+    converter = ms.schema_to_converter(test_schema)
+    assert converter == test_schema.dump
+    assert converter({"foo": 5}) == {"foo": "5"}
+
+
+def test_schema_to_converter_map():
+    test_schema = {"foo": fields.String()}
+    converter = ms.schema_to_converter(test_schema)
+    assert converter({"foo": 5}) == {"foo": "5"}
+
+
+def test_schema_to_converter_field():
+    field = fields.String()
+    converter = ms.schema_to_converter(field)
+    assert converter(5) == "5"
+
+
+def test_schema_to_converter_none():
+    assert ms.schema_to_converter(object()) is None

tests/test_marshalling_args.py

@@ -0,0 +1,97 @@
+from labthings import views, fields
+from labthings.schema import Schema
+from labthings.marshalling.args import use_args, use_body
+
+
+def test_use_body_string(app, client):
+    class Index(views.MethodView):
+        @use_body(fields.String(required=True))
+        def post(self, args):
+            return args
+
+    app.add_url_rule("/", view_func=Index.as_view("index"))
+
+    with client:
+        res = client.post("/", data="string", content_type="application/json")
+        assert res.data.decode() == "string"
+
+
+def test_use_body_no_data_error(app, client):
+    class Index(views.MethodView):
+        @use_body(fields.String(required=True))
+        def post(self, args):
+            return args
+
+    app.add_url_rule("/", view_func=Index.as_view("index"))
+
+    with client:
+        res = client.post("/", content_type="application/json")
+        assert res.status_code == 400
+
+
+def test_use_body_no_data_missing(app, client):
+    class Index(views.MethodView):
+        @use_body(fields.String(missing="default"))
+        def post(self, args):
+            return args
+
+    app.add_url_rule("/", view_func=Index.as_view("index"))
+
+    with client:
+        res = client.post("/", content_type="application/json")
+        assert res.data.decode() == "default"
+
+
+def test_use_body_validation_error(app, client):
+    class Index(views.MethodView):
+        @use_body(fields.String(required=True))
+        def post(self, args):
+            return args
+
+    app.add_url_rule("/", view_func=Index.as_view("index"))
+
+    with client:
+        res = client.post("/", json={"foo": "bar"}, content_type="application/json")
+        assert res.status_code == 400
+
+
+def test_use_args_field(app, client):
+    class Index(views.MethodView):
+        @use_args(fields.String(required=True))
+        def post(self, args):
+            return args
+
+    app.add_url_rule("/", view_func=Index.as_view("index"))
+
+    with client:
+        res = client.post("/", data="string", content_type="application/json")
+        assert res.data.decode() == "string"
+
+
+def test_use_args_map(app, client):
+    class Index(views.MethodView):
+        @use_args({"foo": fields.String(required=True)})
+        def post(self, args):
+            return args
+
+    app.add_url_rule("/", view_func=Index.as_view("index"))
+
+    with client:
+        res = client.post("/", json={"foo": "bar"}, content_type="application/json")
+        assert res.json == {"foo": "bar"}
+
+
+def test_use_args_schema(app, client):
+    class TestSchema(Schema):
+        foo = fields.String(required=True)
+
+    class Index(views.MethodView):
+        @use_args(TestSchema())
+        def post(self, args):
+            return args
+
+    app.add_url_rule("/", view_func=Index.as_view("index"))
+
+    with client:
+        res = client.post("/", json={"foo": "bar"}, content_type="application/json")
+        assert res.json == {"foo": "bar"}