更新类的测试

zhanghe06 · zhanghe06 · commit 8b6ddb64bb06 · 2016-01-13T17:57:15.000+08:00
diff --git a/test/test_class.py b/test/test_class.py
@@ -38,16 +38,114 @@ class BaseResume(object):
 
     # 定义构造方法
     def __init__(self, **kwargs):
+        self.data = kwargs
         for key, value in kwargs.iteritems():
+            if not hasattr(self, key):
+                raise Exception(u'存在不允许的属性')
             self.__setattr__(key, value)
 
+    # 字典方式获取元素
+    def __getitem__(self, key):
+        return self.data.get(key)
 
-if __name__ == '__main__':
+    # 字典方式设置元素
+    def __setitem__(self, key, value):
+        self.data[key] = value
+
+
+class DetailResume(object):
+    """
+    简历详细信息
+    """
+    # 定义构造方法
+    def __init__(self, **kwargs):
+        for key, value in kwargs.iteritems():
+            self.__setattr__(key, value)
+
+
+def test_01():
+    print '--------测试 01'
     user_info = {
         'name': 'Tom',
         'age': 20
     }
     # base_resume = BaseResume(name='Tom', age=20)  # 方式一
     base_resume = BaseResume(**user_info)  # 方式二
-    print base_resume.name
-    print base_resume.age
+    print base_resume.name, base_resume['name']
+    print base_resume.age, base_resume['age']
+    # print base_resume.ff
+    print isinstance(base_resume, BaseResume)  # True
+    print isinstance(base_resume, DetailResume)  # False
+
+
+def test_02():
+    print '--------测试 02'
+    try:
+        base_resume = BaseResume(name='Tom', age=20, ff=33)  # 方式一 塞入错误的key
+        print base_resume.name
+        print base_resume.age
+        print base_resume.ff
+        print isinstance(base_resume, BaseResume)  # True
+        print isinstance(base_resume, DetailResume)  # False
+    except Exception, e:
+        print e.message
+
+
+def test_03():
+    print '--------测试 03'
+    try:
+        data = {
+            'base': {
+                'uid': 100,
+                'name': 'Tom',
+                'sex': 'M'
+            },
+            'edu': [
+                {
+                    'school': 'S-ABC',
+                    'degree': u'学历1',
+                    'major': u'专业1'
+                },
+                {
+                    'school': 'S-DEF',
+                    'degree': u'学历2',
+                    'major': u'专业2'
+                }
+            ],
+            'work': [
+                {
+                    'company': 'C-DGG',
+                    'job': 'J-DGG'
+                },
+                {
+                    'company': 'C-FGG',
+                    'job': 'J-FGG'
+                },
+            ]
+        }
+        detail_resume = DetailResume(**data)
+        print isinstance(detail_resume, DetailResume)
+        print type(detail_resume.work), detail_resume.work
+    except Exception, e:
+        print e.message
+
+
+if __name__ == '__main__':
+    test_01()
+    test_02()
+    test_03()
+
+
+"""
+测试结果：
+--------测试 01
+Tom
+20
+True
+False
+--------测试 02
+存在不允许的属性
+--------测试 03
+True
+<type 'list'> [{'company': 'C-DGG', 'job': 'J-DGG'}, {'company': 'C-FGG', 'job': 'J-FGG'}]
+"""
