Add: 添加WechatConf测试文件

doraemonext · Jan 27, 2016 · 7daca54 · 7daca54
1 parent 9223e0a
commit 7daca54
Show file tree

Hide file tree

Showing 30 changed files with 392 additions and 19 deletions.
diff --git a/setup.py b/setup.py
@@ -18,4 +18,7 @@
     packages=find_packages(),
     include_package_data=True,
     install_requires=map(lambda x: x.replace('==', '>='), open("requirements.txt").readlines()),
+
+    tests_require=['nose', 'httmock'],
+    test_suite='tests',
 )
diff --git a/tests/__init__.py b/tests/__init__.py
diff --git a/tests/core/__init__.py b/tests/core/__init__.py
diff --git a/tests/core/test_conf.py b/tests/core/test_conf.py
@@ -0,0 +1,166 @@
+# -*- coding: utf-8 -*-
+
+from __future__ import unicode_literals
+
+import os
+import json
+import time
+import unittest
+
+import xmltodict
+from httmock import urlmatch, HTTMock, response
+
+from wechat_sdk.core.conf import WechatConf
+from wechat_sdk.exceptions import NeedParamError
+from wechat_sdk.lib.crypto.base import BaseCrypto
+from wechat_sdk.utils import to_binary
+from tests.utils import api_weixin_mock
+
+
+class TestBaseCrypto(BaseCrypto):
+
+    def get_random_str(self):
+        return to_binary('doraemonext')
+
+
+class CoreConfTestCase(unittest.TestCase):
+    token = 'test_token'
+    appid = 'wx9354b770fe837911'
+    appsecret = 'c994da14dca2047bb51caaedaf16f249'
+    encoding_aes_key = 'hXMHI0zd8S4Hc14NRoSKvX8lw8jQhJ02N4bEXKh9tJc'
+
+    fixtures_access_token = 'HoVFaIslbrofqJgkR0Svcx2d4za0RJKa3H6A_NjzhBbm96Wtg_a3ifUYQvOfJmV76QTcCpNubcsnOLmDopu2hjWfFeQSCE4c8QrsxwE_N3w'
+    fixtures_jsapi_ticket = 'bxLdikRXVbTPdHSM05e5u5sUoXNKd8-41ZO3MhKoyN5OfkWITDGgnr2fwJ0m9E8NYzWKVZvdVtaUgWvsdshFKA'
+
+    normal_request_message = """<xml>
+    <ToUserName><![CDATA[gh_1b2959761a7d]]></ToUserName>
+    <FromUserName><![CDATA[oJNCEjt3uIphaC1DrpB030QxMV_w]]></FromUserName>
+    <CreateTime>1442129896</CreateTime>
+    <MsgType><![CDATA[text]]></MsgType>
+    <Content><![CDATA[xxx]]></Content>
+    <MsgId>6193900740131666054</MsgId>
+    </xml>"""
+    compatible_request_message = """<xml>
+    <ToUserName><![CDATA[gh_1b2959761a7d]]></ToUserName>
+    <FromUserName><![CDATA[oJNCEjt3uIphaC1DrpB030QxMV_w]]></FromUserName>
+    <CreateTime>1442129896</CreateTime>
+    <MsgType><![CDATA[text]]></MsgType>
+    <Content><![CDATA[xxx]]></Content>
+    <MsgId>6193900740131666054</MsgId>
+    <Encrypt><![CDATA[JCqIUfe7GJ84SQdp7M4y2xrnsqUp9ihcNZrAErcA8KSc5ewTAYCX1CtAiZqntANyYmFdQqwNAQXMU3vQGbJms/rXSjz4vN77kc9SbIDphNKRYXyaEI5P8iuQHlAjyfUE2kXnWjusYEPnKaB5RLkanwbNwhWVzgYbV4OVst9hSMepv3WpVzOBZlj8CpqhklqwHJuOda4X7unQzwRWyLA8M/lzPDCdN3rcRqz2wH9SlEc0JfjKXKk1RmBFYhHbJ2boeLFc7dIg0Sc3JNMCBw7MU4tBWAEuD+wedUSy45zZ0u1HeQSe13d6QSlVOoC9Ta0InwnlL28VpngEMru4YGeDGoyRKgDOYbDr71dI0sp/kInAKwPXWbOcqd0LFQ9P9Y+BxW4z5BNkYo9LFubXbilnO/VzU8++zuAhFSED11baYFs=]]></Encrypt>
+    </xml>"""
+    safe_request_message = """<xml>
+    <ToUserName><![CDATA[gh_1b2959761a7d]]></ToUserName>
+    <Encrypt><![CDATA[JCqIUfe7GJ84SQdp7M4y2xrnsqUp9ihcNZrAErcA8KSc5ewTAYCX1CtAiZqntANyYmFdQqwNAQXMU3vQGbJms/rXSjz4vN77kc9SbIDphNKRYXyaEI5P8iuQHlAjyfUE2kXnWjusYEPnKaB5RLkanwbNwhWVzgYbV4OVst9hSMepv3WpVzOBZlj8CpqhklqwHJuOda4X7unQzwRWyLA8M/lzPDCdN3rcRqz2wH9SlEc0JfjKXKk1RmBFYhHbJ2boeLFc7dIg0Sc3JNMCBw7MU4tBWAEuD+wedUSy45zZ0u1HeQSe13d6QSlVOoC9Ta0InwnlL28VpngEMru4YGeDGoyRKgDOYbDr71dI0sp/kInAKwPXWbOcqd0LFQ9P9Y+BxW4z5BNkYo9LFubXbilnO/VzU8++zuAhFSED11baYFs=]]></Encrypt>
+    </xml>"""
+
+    response_message = """<xml>
+    <ToUserName><![CDATA[gh_1b2959761a7d]]></ToUserName>
+    <FromUserName><![CDATA[oJNCEjt3uIphaC1DrpB030QxMV_w]]></FromUserName>
+    <CreateTime>1442129896</CreateTime>
+    <MsgType><![CDATA[text]]></MsgType>
+    <Content><![CDATA[测试消息]]></Content>
+    <MsgId>6193900740131666054</MsgId>
+    </xml>"""
+    response_encrypted_message = """<xml>
+    <Encrypt><![CDATA[manHcaGm3P7cOzkyWaigw1nNiIohhn3S7xogsE0ftGFIWncP56fLoI8oVv5HS8D44X5EuQjS+/0rGMNDblrdiiflGzrTjv1CtyT0jEYsgSdPCPFvGf/pOv+timCn+YAQDby0vSxzdJ3WKQB7kubElnxVrQUTIKfe4YKSxaGuhr4m6O7C6YH2uEGaZYQyUeuq5GVPcd9ZS45YOFzAxOyK8mWp7tCkuItqt5hElZ6dMaFOnp6cEGN7113ONtPSS2US7thd9u0+EVuiNEvuF5ErqJtc0saBK18LL/PeZ1xH1ZI6ZQZiQiX3uSeQ7SUDzOCNPV0BZ9NCdOnhHwKGS/F5PkU/sG0LsZVkobeBYApOTS0MY3Da/UVR3Zznr0Enab4YM4VjGdNBclBdkbukT/t/cYQaha0zGrYpOpzqNOQpxSuM5jB/lSQo6B+niT/PftLCR6J+hmRAXPbeVRZx1pJcUQ==]]></Encrypt>
+    <MsgSignature><![CDATA[e458c417d5aee4c3a9abed78da4020404e9b6cf9]]></MsgSignature>
+    <TimeStamp>1442129896</TimeStamp>
+    <Nonce><![CDATA[1641605039]]></Nonce>
+    </xml>
+    """
+    signature = 'cc7c1221903889730bee4e92aba106a7bdb382f8'
+    timestamp = 1442129896
+    nonce = 1641605039
+    encrypt_type = 'aes'
+    msg_signature = 'd4e1e604ec9c7f88aeb0faea11ebb83d6f6f20a0'
+
+    def test_init_with_token(self):
+        """ 测试只有 Token 的初始化 """
+        conf = WechatConf(token=self.token)
+        self.assertEqual(conf.token, self.token)
+        self.assertIsNone(conf.appid)
+        self.assertIsNone(conf.appsecret)
+        self.assertIsNone(conf.encoding_aes_key)
+        self.assertIsNone(conf.crypto)
+        with self.assertRaises(NeedParamError):
+            getattr(conf, 'access_token')
+        with self.assertRaises(NeedParamError):
+            getattr(conf, 'jsapi_ticket')
+
+    def test_init_with_token_and_appid_appsecret(self):
+        """ 测试 Token AppID AppSecret 的初始化 """
+        conf = WechatConf(token=self.token, appid=self.appid, appsecret=self.appsecret)
+        self.assertEqual(conf.token, self.token)
+        self.assertEqual(conf.appid, self.appid)
+        self.assertEqual(conf.appsecret, self.appsecret)
+        self.assertIsNone(conf.encoding_aes_key)
+        self.assertIsNone(conf.crypto)
+        with HTTMock(api_weixin_mock):
+            access_token = conf.access_token
+            access_token_expires_at = int(time.time()) + 7200
+            self.assertEqual(access_token, self.fixtures_access_token)
+            self.assertEqual(conf._WechatConf__access_token, self.fixtures_access_token)
+            self.assertIn(conf._WechatConf__access_token_expires_at,
+                          range(access_token_expires_at-10, access_token_expires_at+11))
+
+            jsapi_ticket = conf.jsapi_ticket
+            jsapi_ticket_expires_at = int(time.time()) + 7200
+            self.assertEqual(jsapi_ticket, self.fixtures_jsapi_ticket)
+            self.assertEqual(conf._WechatConf__jsapi_ticket, self.fixtures_jsapi_ticket)
+            self.assertIn(conf._WechatConf__access_token_expires_at,
+                          range(jsapi_ticket_expires_at-10, jsapi_ticket_expires_at+11))
+
+    def test_init_with_normal_encrypt_mode(self):
+        """ 测试明文模式下的初始化 """
+        conf = WechatConf(token=self.token, appid=self.appid, appsecret=self.appsecret, encrypt_mode='normal')
+        self.assertEqual(conf.token, self.token)
+        self.assertEqual(conf.appid, self.appid)
+        self.assertEqual(conf.appsecret, self.appsecret)
+        self.assertIsNone(conf.encoding_aes_key)
+        self.assertIsNone(conf.crypto)
+
+    def test_init_with_compatible_encrypt_mode(self):
+        """ 测试兼容模式下的初始化 """
+        conf = WechatConf(token=self.token, appid=self.appid, appsecret=self.appsecret,
+                          encrypt_mode='compatible', encoding_aes_key=self.encoding_aes_key)
+        self.assertEqual(conf.token, self.token)
+        self.assertEqual(conf.appid, self.appid)
+        self.assertEqual(conf.appsecret, self.appsecret)
+        self.assertEqual(conf.encoding_aes_key, self.encoding_aes_key)
+        self.assertIsNotNone(conf.crypto)
+
+        # 测试解密微信服务器的请求消息
+        req = conf.crypto.decrypt_message(msg=self.compatible_request_message, msg_signature=self.msg_signature,
+                                          timestamp=self.timestamp, nonce=self.nonce)
+        self.assertEqual(xmltodict.parse(req), xmltodict.parse(self.normal_request_message))
+
+        # 测试加密返回信息
+        origin_crypto = conf.crypto._WechatBaseCrypto__pc
+        conf.crypto._WechatBaseCrypto__pc = TestBaseCrypto(key=conf.crypto._WechatBaseCrypto__key)
+        resp = conf.crypto.encrypt_message(msg=self.response_message, nonce=self.nonce, timestamp=self.timestamp)
+        self.assertEqual(xmltodict.parse(resp), xmltodict.parse(self.response_encrypted_message))
+        conf.crypto._WechatBaseCrypto__pc = origin_crypto
+
+    def test_init_with_safe_encrypt_mode(self):
+        """ 测试安全模式下的初始化 """
+        conf = WechatConf(token=self.token, appid=self.appid, appsecret=self.appsecret,
+                          encrypt_mode='safe', encoding_aes_key=self.encoding_aes_key)
+        self.assertEqual(conf.token, self.token)
+        self.assertEqual(conf.appid, self.appid)
+        self.assertEqual(conf.appsecret, self.appsecret)
+        self.assertEqual(conf.encoding_aes_key, self.encoding_aes_key)
+        self.assertIsNotNone(conf.crypto)
+
+        # 测试解密微信服务器的请求消息
+        req = conf.crypto.decrypt_message(msg=self.safe_request_message, msg_signature=self.msg_signature,
+                                          timestamp=self.timestamp, nonce=self.nonce)
+        self.assertEqual(xmltodict.parse(req), xmltodict.parse(self.normal_request_message))
+
+        # 测试加密返回信息
+        origin_crypto = conf.crypto._WechatBaseCrypto__pc
+        conf.crypto._WechatBaseCrypto__pc = TestBaseCrypto(key=conf.crypto._WechatBaseCrypto__key)
+        resp = conf.crypto.encrypt_message(msg=self.response_message, nonce=self.nonce, timestamp=self.timestamp)
+        self.assertEqual(xmltodict.parse(resp), xmltodict.parse(self.response_encrypted_message))
+        conf.crypto._WechatBaseCrypto__pc = origin_crypto
+
diff --git a/tests/fixtures/customservice_kfaccount_add.json b/tests/fixtures/customservice_kfaccount_add.json
@@ -0,0 +1,4 @@
+{
+    "errcode": 0,
+    "errmsg": "ok"
+}
diff --git a/tests/fixtures/customservice_kfaccount_del.json b/tests/fixtures/customservice_kfaccount_del.json
@@ -0,0 +1,4 @@
+{
+    "errcode": 0,
+    "errmsg": "ok"
+}
diff --git a/tests/fixtures/customservice_kfaccount_update.json b/tests/fixtures/customservice_kfaccount_update.json
@@ -0,0 +1,4 @@
+{
+    "errcode": 0,
+    "errmsg": "ok"
+}
diff --git a/tests/fixtures/customservice_kfaccount_uploadheadimg.json b/tests/fixtures/customservice_kfaccount_uploadheadimg.json
@@ -0,0 +1,4 @@
+{
+    "errcode": 0,
+    "errmsg": "ok"
+}
diff --git a/tests/fixtures/groups_create.json b/tests/fixtures/groups_create.json
@@ -0,0 +1,6 @@
+{
+    "group": {
+        "id": 107,
+        "name": "test"
+    }
+}
diff --git a/tests/fixtures/groups_get.json b/tests/fixtures/groups_get.json
@@ -0,0 +1,29 @@
+{
+    "groups": [
+        {
+            "id": 0,
+            "name": "未分组",
+            "count": 72596
+        },
+        {
+            "id": 1,
+            "name": "黑名单",
+            "count": 36
+        },
+        {
+            "id": 2,
+            "name": "星标组",
+            "count": 8
+        },
+        {
+            "id": 104,
+            "name": "华东媒",
+            "count": 4
+        },
+        {
+            "id": 106,
+            "name": "★不测试组★",
+            "count": 1
+        }
+    ]
+}
diff --git a/tests/fixtures/groups_getid.json b/tests/fixtures/groups_getid.json
@@ -0,0 +1,3 @@
+{
+    "groupid": 102
+}
diff --git a/tests/fixtures/groups_members_update.json b/tests/fixtures/groups_members_update.json
@@ -0,0 +1,4 @@
+{
+    "errcode": 0,
+    "errmsg": "ok"
+}
diff --git a/tests/fixtures/groups_update.json b/tests/fixtures/groups_update.json
@@ -0,0 +1,4 @@
+{
+    "errcode": 0,
+    "errmsg": "ok"
+}
diff --git a/tests/fixtures/menu_create.json b/tests/fixtures/menu_create.json
@@ -0,0 +1,4 @@
+{
+    "errcode": 0,
+    "errmsg": "ok"
+}
diff --git a/tests/fixtures/menu_delete.json b/tests/fixtures/menu_delete.json
@@ -0,0 +1,4 @@
+{
+    "errcode": 0,
+    "errmsg": "ok"
+}
diff --git a/tests/fixtures/menu_get.json b/tests/fixtures/menu_get.json
@@ -0,0 +1,41 @@
+{
+    "menu": {
+        "button": [
+            {
+                "type": "click",
+                "name": "今日歌曲",
+                "key": "V1001_TODAY_MUSIC",
+                "sub_button": []
+            },
+            {
+                "type": "click",
+                "name": "歌手简介",
+                "key": "V1001_TODAY_SINGER",
+                "sub_button": []
+            },
+            {
+                "name": "菜单",
+                "sub_button": [
+                    {
+                        "type": "view",
+                        "name": "搜索",
+                        "url": "http://www.soso.com/",
+                        "sub_button": []
+                    },
+                    {
+                        "type": "view",
+                        "name": "视频",
+                        "url": "http://v.qq.com/",
+                        "sub_button": []
+                    },
+                    {
+                        "type": "click",
+                        "name": "赞一下我们",
+                        "key": "V1001_GOOD",
+                        "sub_button": []
+                    }
+                ]
+            }
+        ]
+    }
+}
diff --git a/tests/fixtures/message_custom_send.json b/tests/fixtures/message_custom_send.json
@@ -0,0 +1,4 @@
+{
+    "errcode": 0,
+    "errmsg": "ok"
+}
diff --git a/tests/fixtures/message_template_send.json b/tests/fixtures/message_template_send.json
@@ -0,0 +1,5 @@
+{
+    "errcode": 0,
+    "errmsg": "ok",
+    "msgid": 200228332
+}
diff --git a/tests/fixtures/qrcode_create.json b/tests/fixtures/qrcode_create.json
@@ -0,0 +1,5 @@
+{
+    "ticket": "gQH47joAAAAAAAAAASxodHRwOi8vd2VpeGluLnFxLmNvbS9xL2taZ2Z3TVRtNzJXV1Brb3ZhYmJJAAIEZ23sUwMEmm3sUw==",
+    "expire_seconds": 60,
+    "url": "http:\/\/weixin.qq.com\/q\/kZgfwMTm72WWPkovabbI"
+}
diff --git a/tests/fixtures/template_api_add_template.json b/tests/fixtures/template_api_add_template.json
@@ -0,0 +1,5 @@
+{
+    "errcode": 0,
+    "errmsg": "ok",
+    "template_id": "Doclyl5uP7Aciu-qZ7mJNPtWkbkYnWBWVja26EGbNyk"
+}
diff --git a/tests/fixtures/ticket_getticket.json b/tests/fixtures/ticket_getticket.json
@@ -0,0 +1,6 @@
+{
+    "errcode":0,
+    "errmsg":"ok",
+    "ticket":"bxLdikRXVbTPdHSM05e5u5sUoXNKd8-41ZO3MhKoyN5OfkWITDGgnr2fwJ0m9E8NYzWKVZvdVtaUgWvsdshFKA",
+    "expires_in":7200
+}
diff --git a/tests/fixtures/token.json b/tests/fixtures/token.json
@@ -0,0 +1,4 @@
+{
+    "access_token": "HoVFaIslbrofqJgkR0Svcx2d4za0RJKa3H6A_NjzhBbm96Wtg_a3ifUYQvOfJmV76QTcCpNubcsnOLmDopu2hjWfFeQSCE4c8QrsxwE_N3w",
+    "expires_in": 7200
+}
diff --git a/tests/fixtures/user_get.json b/tests/fixtures/user_get.json
@@ -0,0 +1,12 @@
+{
+    "total": 2,
+    "count": 2,
+    "data": {
+        "openid": [
+            "",
+            "OPENID1",
+            "OPENID2"
+        ]
+    },
+    "next_openid": "NEXT_OPENID"
+}
diff --git a/tests/fixtures/user_info.json b/tests/fixtures/user_info.json
@@ -0,0 +1,15 @@
+{
+    "subscribe": 1,
+    "openid": "o6_bmjrPTlm6_2sgVt7hMZOPfL2M",
+    "nickname": "Band",
+    "sex": 1,
+    "language": "zh_CN",
+    "city": "广州",
+    "province": "广东",
+    "country": "中国",
+    "headimgurl": "http://wx.qlogo.cn/mmopen/g3MonUZtNHkdmzicIlibx6iaFqAc56vxLSUfpb6n5WKSYVY0ChQKkiaJSgQ1dZuTOgvLLrhJbERQQ4eMsv84eavHiaiceqxibJxCfHe/0",
+    "subscribe_time": 1382694957,
+    "unionid": "o6_bmasdasdsad6_2sgVt7hMZOPfL",
+    "remark": "",
+    "groupid": 0
+}