test: update test with data_key

fuhrysteve · Feb 22, 2021 · 160c85a · 160c85a
1 parent 0a0856b
commit 160c85a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tests/__init__.py b/tests/__init__.py
@@ -32,7 +32,7 @@ class UserSchema(Schema):
     email = fields.Email()
     balance = fields.Decimal()
     registered = fields.Boolean()
-    hair_colors = fields.List(fields.Raw)
+    hair_colors = fields.List(fields.Raw, data_key="hairColors", required=True)
     sex_choices = fields.List(fields.Raw)
     finger_count = fields.Integer()
     uid = fields.UUID()

diff --git a/tests/test_dump.py b/tests/test_dump.py
@@ -18,7 +18,7 @@ def test_dump_schema():
 
     props = dumped["definitions"]["UserSchema"]["properties"]
     for field_name, field in schema.fields.items():
-        assert field_name in props
+        assert field.data_key or field_name in props
 
 
 def test_default():