Merge pull request #24 from laslabs/feature/0.1/user-type-enum

[ADD] Support for user types

Author: Ted S

carepoint/models/cph/user.py

@@ -2,6 +2,8 @@
 # Copyright 2015-TODAY LasLabs Inc.
 # License MIT (https://opensource.org/licenses/MIT).
 
+import enum
+
 from carepoint import Carepoint
 from sqlalchemy import (Column,
                         Integer,
@@ -12,6 +14,14 @@
                         )
 
 
+class EnumUserType(enum.Enum):
+    """ It provides PEP-0435 compliant Carepoint User Type Enumerable """
+
+    user = 'U'
+    group = 'G'
+    machine = 'M'
+
+
 class User(Carepoint.BASE):
     __tablename__ = 'csuser'
     __dbname__ = 'cph'
@@ -57,3 +67,7 @@ class User(Carepoint.BASE):
         ForeignKey('csuser.user_id'),
     )
     chg_date = Column(DateTime)
+
+    @property
+    def user_type(self):
+        return EnumUserType(self.user_type_cd)

carepoint/tests/models/cph/test_user.py

@@ -5,14 +5,28 @@
 import unittest
 from sqlalchemy.schema import Table
 from carepoint.tests.db.db import DatabaseTest
-from carepoint.models.cph.user import User
+from carepoint.models.cph.user import EnumUserType, User
 
 
 class TestModelsCphUser(DatabaseTest):
 
+    def new_record(self, type_cd='U'):
+        self.type_cd = type_cd
+        obj = User()
+        obj.user_type_cd = type_cd
+        return obj
+
     def test_table_initialization(self, ):
         self.assertIsInstance(User.__table__, Table)
 
+    def test_user_type(self):
+        """ It should return proper Enum for user type cd """
+        obj = self.new_record()
+        self.assertEqual(
+            obj.user_type,
+            EnumUserType(self.type_cd),
+        )
+
 
 if __name__ == '__main__':
     unittest.main()

setup.py

@@ -9,7 +9,7 @@
 
 setup_vals = {
     'name': 'carepoint',
-    'version': '0.1.7',
+    'version': '0.1.8',
     'author': 'LasLabs Inc.',
     'author_email': 'support@laslabs.com',
     'description': 'This library will allow you to interact with CarePoint '