feat(schematics) Creates ChoicesEnumType

Author: Maiky Guanaes

AUTHORS.rst

@@ -15,6 +15,7 @@ Contributors
 * Leandro Gomes <[email protected]>
 * Tomas Fagerbekk <[email protected]>
 * Lefteris Karapetsas <[email protected]>
+* Maiky Guanaes <[email protected]>
 
 
 Tools

choicesenum/schematics/__init__.py

@@ -0,0 +1,22 @@
+# coding: utf-8
+from __future__ import absolute_import, unicode_literals
+
+from choicesenum import ChoicesEnum
+from schematics.types import BaseType, ValidationError
+
+
+class ChoicesEnumType(BaseType):
+    def __init__(self, type_):
+        if not issubclass(type_, ChoicesEnum):
+            raise ValidationError(self.Messages.INVALID_TYPE)
+        super(ChoicesEnumType, self).__init__(type_)
+        self.type = type_
+
+    def to_native(self, value, context=None):
+        return self.type(value)
+
+    def to_primitive(self, value, context=None):
+        return self.type(value).value
+
+    class Messages:
+        INVALID_TYPE = 'Expected a ChoicesEnum sub type'

choicesenum/schematics/types.py

@@ -4,3 +4,4 @@ pytest==3.10.1
 pytest-cov==2.6.0
 pytest-watch==4.2.0
 pytest-sugar==0.9.2
+schematics==1.1.3

requirements_test.txt

@@ -0,0 +1,18 @@
+# coding: utf-8
+from __future__ import absolute_import, unicode_literals
+
+import pytest
+from schematics.exceptions import ValidationError
+from choicesenum.schematics.types import ChoicesEnumType
+
+
+def test_choices_enum_type(http_statuses):
+    state = ChoicesEnumType(http_statuses)
+    assert state.to_native(200) is http_statuses.OK
+    assert state.to_primitive(200) == http_statuses.OK.value
+
+
+def test_choices_enum_type_should_throw_exception():
+    with pytest.raises(ValidationError) as e:
+        ChoicesEnumType(object)
+        assert e.value.message[0] == ChoicesEnumType.Messages.INVALID_TYPE