chore(schematics) Improves ChoicesEnumType to receive kwargs

loggi · Sep 2, 2019 · f191c08 · f191c08
1 parent 6342b6e
commit f191c08
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/choicesenum/schematics/types.py b/choicesenum/schematics/types.py
@@ -6,10 +6,10 @@
 
 
 class ChoicesEnumType(BaseType):
-    def __init__(self, type_):
+    def __init__(self, type_, **kwargs):
         if not issubclass(type_, ChoicesEnum):
             raise ValidationError(self.Messages.INVALID_TYPE)
-        super(ChoicesEnumType, self).__init__(type_)
+        super(ChoicesEnumType, self).__init__(type_, kwargs)
         self.type = type_
 
     def to_native(self, value, context=None):

diff --git a/tests/test_schematics_types.py b/tests/test_schematics_types.py
@@ -12,6 +12,11 @@ def test_choices_enum_type(http_statuses):
     assert state.to_primitive(200) == http_statuses.OK.value
 
 
+def test_choices_enum_type_with_extra_params(http_statuses):
+    state = ChoicesEnumType(http_statuses, required=True)
+    assert state.to_native(200) is http_statuses.OK
+
+
 def test_choices_enum_type_should_throw_exception():
     with pytest.raises(ValidationError) as e:
         ChoicesEnumType(object)