class HuoBiMessage:
def __init__(self,
name='huobi',
webhook=None,
agent_id=None,
secret=None,
company_id=None):
self.name = name
# 机器人webhook
self.webhook = read_secret("wechat",
"notechats",
"huobi",
name,
"webhook",
value=webhook)
# 应用ID
self.agent_id = read_secret("wechat",
"notechats",
"huobi",
name,
"agent_id",
value=agent_id)
# 企业ID
self.company_id = read_secret("wechat",
"notechats",
"huobi",
name,
"company_id",
value=company_id)
# 应用Secret
self.secret = read_secret("wechat",
"notechats",
"huobi",
name,
"secret",
value=secret)
self.client = WeChatClient(corp_id=self.company_id, secret=self.secret)
def send_msg(self, msg):
self.check_token()
data = {"msgtype": "text", "text": {"content": msg}}
self.client.message.send(agent_id=self.agent_id,
user_ids=['NiuLiangTao', 'GuoYe'],
party_ids=['3'],
msg=data)
self.send_to_qywx(msg)
def send_to_qywx(self, msg):
headers = {'Content-Type': 'application/json'}
data = {"msgtype": "text", "text": {"content": msg}}
data = json.dumps(data)
r = requests.post(url=self.webhook, headers=headers, data=data)
def check_token(self):
# 通行密钥
access_token = None
access_token = read_secret("wechat", "notechats", "huobi", self.name,
"access_token")
if not access_token:
response = self.client.fetch_access_token()
access_token = read_secret("wechat",
"notechats",
"huobi",
self.name,
"access_token",
value=response['access_token'],
expire_time=response['expires_in'])
print(response)
self.client.session.set(self.client.access_token_key, access_token)
return access_token
class WeChatClientTestCase(unittest.TestCase):
corp_id = '123456'
secret = '123456'
def setUp(self):
self.client = WeChatClient(self.corp_id, self.secret)
def test_fetch_access_token(self):
with HTTMock(wechat_api_mock):
token = self.client.fetch_access_token()
self.assertEqual('1234567890', token['access_token'])
self.assertEqual(7200, token['expires_in'])
self.assertEqual('1234567890', self.client.access_token)
def test_get_wechat_ips(self):
with HTTMock(wechat_api_mock):
res = self.client.misc.get_wechat_ips()
self.assertEqual(['127.0.0.1'], res)
def test_department_create(self):
with HTTMock(wechat_api_mock):
res = self.client.department.create('Test')
self.assertEqual(2, res['id'])
def test_department_update(self):
with HTTMock(wechat_api_mock):
res = self.client.department.update(2, 'Test 1')
self.assertEqual(0, res['errcode'])
def test_department_delete(self):
with HTTMock(wechat_api_mock):
res = self.client.department.delete(2)
self.assertEqual(0, res['errcode'])
def test_department_get(self):
with HTTMock(wechat_api_mock):
res = self.client.department.get()
self.assertEqual(2, len(res))
def test_department_get_users(self):
with HTTMock(wechat_api_mock):
res = self.client.department.get_users(2)
self.assertEqual(1, len(res))
def test_tag_create(self):
with HTTMock(wechat_api_mock):
res = self.client.tag.create('test')
self.assertEqual('1', res['tagid'])
def test_tag_update(self):
with HTTMock(wechat_api_mock):
res = self.client.tag.update(1, 'test')
self.assertEqual(0, res['errcode'])
def test_tag_delete(self):
with HTTMock(wechat_api_mock):
res = self.client.tag.delete(1)
self.assertEqual(0, res['errcode'])
def test_tag_get_users(self):
with HTTMock(wechat_api_mock):
res = self.client.tag.get_users(1)
self.assertEqual(1, len(res['userlist']))
self.assertEqual(1, len(res['partylist']))
def test_tag_add_users(self):
with HTTMock(wechat_api_mock):
res = self.client.tag.add_users(1, [1, 2, 3])
self.assertEqual(0, res['errcode'])
def test_tag_delete_users(self):
with HTTMock(wechat_api_mock):
res = self.client.tag.delete_users(1, [1, 2, 3])
self.assertEqual(0, res['errcode'])
def test_tag_list(self):
with HTTMock(wechat_api_mock):
res = self.client.tag.list()
self.assertEqual(2, len(res))
def test_batch_invite_user(self):
with HTTMock(wechat_api_mock):
res = self.client.batch.invite_user(
'http://example.com',
'123456',
'123456',
'123|456',
[123, 456],
(12, 34),
''
)
self.assertEqual(0, res['errcode'])
def test_batch_sync_user(self):
with HTTMock(wechat_api_mock):
res = self.client.batch.sync_user(
'http://example.com',
'123456',
'123456',
'12345678'
)
self.assertEqual(0, res['errcode'])
def test_batch_replace_user(self):
with HTTMock(wechat_api_mock):
res = self.client.batch.replace_user(
'http://example.com',
'123456',
'123456',
'12345678'
)
self.assertEqual(0, res['errcode'])
def test_batch_replace_party(self):
with HTTMock(wechat_api_mock):
res = self.client.batch.replace_party(
'http://example.com',
'123456',
'123456',
'12345678'
)
self.assertEqual(0, res['errcode'])
def test_batch_get_result(self):
with HTTMock(wechat_api_mock):
res = self.client.batch.get_result('123456')
self.assertEqual(0, res['errcode'])
self.assertEqual(1, res['status'])
def test_jsapi_get_ticket(self):
with HTTMock(wechat_api_mock):
result = self.client.jsapi.get_ticket()
self.assertEqual(
'bxLdikRXVbTPdHSM05e5u5sUoXNKd8-41ZO3MhKoyN5OfkWITDGgnr2fwJ0m9E8NYzWKVZvdVtaUgWvsdshFKA', # NOQA
result['ticket']
)
self.assertEqual(7200, result['expires_in'])
def test_jsapi_get_jsapi_signature(self):
noncestr = 'Wm3WZYTPz0wzccnW'
ticket = 'sM4AOVdWfPE4DxkXGEs8VMCPGGVi4C3VM0P37wVUCFvkVAy_90u5h9nbSlYy3-Sl-HhTdfl2fzFy1AOcHKP7qg' # NOQA
timestamp = 1414587457
url = 'http://mp.weixin.qq.com?params=value'
signature = self.client.jsapi.get_jsapi_signature(
noncestr,
ticket,
timestamp,
url
)
self.assertEqual(
'0f9de62fce790f9a083d5c99e95740ceb90c27ed',
signature
)
def test_user_convert_to_openid(self):
with HTTMock(wechat_api_mock):
res = self.client.user.convert_to_openid('zhangsan')
self.assertEqual('oDOGms-6yCnGrRovBj2yHij5JL6E', res['openid'])
self.assertEqual('wxf874e15f78cc84a7', res['appid'])
def test_user_convert_to_user_id(self):
with HTTMock(wechat_api_mock):
user_id = self.client.user.convert_to_user_id(
'oDOGms-6yCnGrRovBj2yHij5JL6E'
)
self.assertEqual('zhangsan', user_id)
def test_upload_media(self):
media_file = six.StringIO('nothing')
with HTTMock(wechat_api_mock):
media = self.client.media.upload('image', media_file)
self.assertEqual('image', media['type'])
self.assertEqual('12345678', media['media_id'])
def test_material_get_count(self):
with HTTMock(wechat_api_mock):
res = self.client.material.get_count(1)
self.assertEqual(37, res['total_count'])
self.assertEqual(3, res['video_count'])
self.assertEqual(10, res['voice_count'])
self.assertEqual(12, res['image_count'])
self.assertEqual(3, res['file_count'])
self.assertEqual(6, res['mpnews_count'])
def test_material_batchget_mpnews(self):
with HTTMock(wechat_api_mock):
res = self.client.material.batchget(1, 'mpnews')
self.assertEqual('mpnews', res['type'])
self.assertEqual(20, res['total_count'])
self.assertEqual(3, res['item_count'])
self.assertEqual(
'2-G6nrLmr5EC3MMb_-zK1dDdzmd0p7cNliYu',
res['itemlist'][0]['media_id']
)
def test_material_delete(self):
media_id = '2-G6nrLmr5EC3MMb_-zK1dDdzmd0p7cNliYu'
with HTTMock(wechat_api_mock):
res = self.client.material.delete(1, media_id)
self.assertEqual('deleted', res['errmsg'])
def test_material_get_mpnews(self):
media_id = '2-G6nrLmr5EC3MMb_-zK1dDdzmd0p7cNliYu'
with HTTMock(wechat_api_mock):
res = self.client.material.get_articles(1, media_id)
self.assertEqual('mpnews', res['type'])
self.assertEqual(
'2-G6nrLmr5EC3MMb_-zK1dDdzmd0' +
'p7cNliYu9V5w7o8K0HuucGBZCzw4HmLa5C',
res['mpnews']['articles'][0]['thumb_media_id']
)
self.assertEqual(
'2-G6nrLmr5EC3MMb_-zK1dDdzmd0' +
'p7cNliYu9V5w7oovsUPf3wG4t9N3tE',
res['mpnews']['articles'][1]['thumb_media_id']
)
def test_reraise_requests_exception(self):
@urlmatch(netloc=r'(.*\.)?qyapi\.weixin\.qq\.com$')
def _wechat_api_mock(url, request):
return {'status_code': 404, 'content': '404 not found'}
try:
with HTTMock(_wechat_api_mock):
self.client.material.get_count(1)
except WeChatClientException as e:
self.assertEqual(404, e.response.status_code)
def test_shakearound_get_shake_info(self):
with HTTMock(wechat_api_mock):
res = self.client.shakearound.get_shake_info('123456')
self.assertEqual(14000, res['page_id'])
self.assertEqual('zhangsan', res['userid'])
def test_service_get_provider_token(self):
with HTTMock(wechat_api_mock):
res = self.client.service.get_provider_token('provider_secret')
self.assertEqual(7200, res['expires_in'])
self.assertEqual('enLSZ5xxxxxxJRL', res['provider_access_token'])
def test_service_get_login_info(self):
with HTTMock(wechat_api_mock):
res = self.client.service.get_login_info(
'enLSZ5xxxxxxJRL',
'auth_code'
)
self.assertTrue(res['is_sys'])
self.assertTrue(res['is_inner'])
def test_chat_create(self):
with HTTMock(wechat_api_mock):
res = self.client.chat.create(
'1', 'chat', 'zhangsan', ['zhangsan', 'lisi', 'wangwu']
)
self.assertEqual(0, res['errcode'])
def test_chat_get(self):
with HTTMock(wechat_api_mock):
chat = self.client.chat.get('235364212115767297')
self.assertEqual('235364212115767297', chat['chatid'])
self.assertEqual('zhangsan', chat['owner'])
def test_chat_update(self):
with HTTMock(wechat_api_mock):
res = self.client.chat.update(
'235364212115767297',
'lisi',
'企业应用中心',
'zhangsan',
['zhaoli'],
['zhangsan']
)
self.assertEqual(0, res['errcode'])
def test_chat_quit(self):
with HTTMock(wechat_api_mock):
res = self.client.chat.quit('235364212115767297', 'lisi')
self.assertEqual(0, res['errcode'])
def test_chat_clear_notify(self):
with HTTMock(wechat_api_mock):
res = self.client.chat.clear_notify('zhangsan', 'single', 'lisi')
self.assertEqual(0, res['errcode'])
def test_chat_set_mute(self):
mute_list = [
{'userid': 'zhangsan', 'status': 0},
{'userid': 'lisi', 'status': 1},
]
with HTTMock(wechat_api_mock):
res = self.client.chat.set_mute(mute_list)
self.assertEqual(0, res['errcode'])
self.assertEqual(['zhangsan'], res['invaliduser'])
def test_chat_send_text(self):
with HTTMock(wechat_api_mock):
res = self.client.chat.send_text(
'zhangsan', 'single', 'lisi', 'hello'
)
self.assertEqual(0, res['errcode'])
def test_chat_send_image(self):
with HTTMock(wechat_api_mock):
res = self.client.chat.send_image(
'zhangsan', 'single', 'lisi', 'media_id'
)
self.assertEqual(0, res['errcode'])
def test_chat_send_file(self):
with HTTMock(wechat_api_mock):
res = self.client.chat.send_file(
'zhangsan', 'single', 'lisi', 'media_id'
)
self.assertEqual(0, res['errcode'])
class WeChatClientTestCase(unittest.TestCase):
corp_id = '123456'
secret = '123456'
def setUp(self):
self.client = WeChatClient(self.corp_id, self.secret)
def test_fetch_access_token(self):
with HTTMock(wechat_api_mock):
token = self.client.fetch_access_token()
self.assertEqual('1234567890', token['access_token'])
self.assertEqual(7200, token['expires_in'])
self.assertEqual('1234567890', self.client.access_token)
def test_get_wechat_ips(self):
with HTTMock(wechat_api_mock):
res = self.client.misc.get_wechat_ips()
self.assertEqual(['127.0.0.1'], res)
def test_department_create(self):
with HTTMock(wechat_api_mock):
res = self.client.department.create('Test')
self.assertEqual(2, res['id'])
def test_department_update(self):
with HTTMock(wechat_api_mock):
res = self.client.department.update(2, 'Test 1')
self.assertEqual(0, res['errcode'])
def test_department_delete(self):
with HTTMock(wechat_api_mock):
res = self.client.department.delete(2)
self.assertEqual(0, res['errcode'])
def test_department_get(self):
with HTTMock(wechat_api_mock):
res = self.client.department.get()
self.assertEqual(2, len(res))
def test_department_get_users(self):
with HTTMock(wechat_api_mock):
res = self.client.department.get_users(2)
self.assertEqual(1, len(res))
def test_tag_create(self):
with HTTMock(wechat_api_mock):
res = self.client.tag.create('test')
self.assertEqual('1', res['tagid'])
def test_tag_update(self):
with HTTMock(wechat_api_mock):
res = self.client.tag.update(1, 'test')
self.assertEqual(0, res['errcode'])
def test_tag_delete(self):
with HTTMock(wechat_api_mock):
res = self.client.tag.delete(1)
self.assertEqual(0, res['errcode'])
def test_tag_get_users(self):
with HTTMock(wechat_api_mock):
res = self.client.tag.get_users(1)
self.assertEqual(1, len(res['userlist']))
self.assertEqual(1, len(res['partylist']))
def test_tag_add_users(self):
with HTTMock(wechat_api_mock):
res = self.client.tag.add_users(1, [1, 2, 3])
self.assertEqual(0, res['errcode'])
def test_tag_delete_users(self):
with HTTMock(wechat_api_mock):
res = self.client.tag.delete_users(1, [1, 2, 3])
self.assertEqual(0, res['errcode'])
def test_tag_list(self):
with HTTMock(wechat_api_mock):
res = self.client.tag.list()
self.assertEqual(2, len(res))
def test_batch_invite_user(self):
with HTTMock(wechat_api_mock):
res = self.client.batch.invite_user('http://example.com', '123456',
'123456', '123|456',
[123, 456], (12, 34), '')
self.assertEqual(0, res['errcode'])
def test_batch_sync_user(self):
with HTTMock(wechat_api_mock):
res = self.client.batch.sync_user('http://example.com', '123456',
'123456', '12345678')
self.assertEqual(0, res['errcode'])
def test_batch_replace_user(self):
with HTTMock(wechat_api_mock):
res = self.client.batch.replace_user('http://example.com',
'123456', '123456',
'12345678')
self.assertEqual(0, res['errcode'])
def test_batch_replace_party(self):
with HTTMock(wechat_api_mock):
res = self.client.batch.replace_party('http://example.com',
'123456', '123456',
'12345678')
self.assertEqual(0, res['errcode'])
def test_batch_get_result(self):
with HTTMock(wechat_api_mock):
res = self.client.batch.get_result('123456')
self.assertEqual(0, res['errcode'])
self.assertEqual(1, res['status'])
def test_jsapi_get_ticket(self):
with HTTMock(wechat_api_mock):
result = self.client.jsapi.get_ticket()
self.assertEqual(
'bxLdikRXVbTPdHSM05e5u5sUoXNKd8-41ZO3MhKoyN5OfkWITDGgnr2fwJ0m9E8NYzWKVZvdVtaUgWvsdshFKA', # NOQA
result['ticket'])
self.assertEqual(7200, result['expires_in'])
def test_jsapi_get_jsapi_signature(self):
noncestr = 'Wm3WZYTPz0wzccnW'
ticket = 'sM4AOVdWfPE4DxkXGEs8VMCPGGVi4C3VM0P37wVUCFvkVAy_90u5h9nbSlYy3-Sl-HhTdfl2fzFy1AOcHKP7qg' # NOQA
timestamp = 1414587457
url = 'http://mp.weixin.qq.com?params=value'
signature = self.client.jsapi.get_jsapi_signature(
noncestr, ticket, timestamp, url)
self.assertEqual('0f9de62fce790f9a083d5c99e95740ceb90c27ed', signature)
def test_user_convert_to_openid(self):
with HTTMock(wechat_api_mock):
res = self.client.user.convert_to_openid('zhangsan')
self.assertEqual('oDOGms-6yCnGrRovBj2yHij5JL6E', res['openid'])
self.assertEqual('wxf874e15f78cc84a7', res['appid'])
def test_user_convert_to_user_id(self):
with HTTMock(wechat_api_mock):
user_id = self.client.user.convert_to_user_id(
'oDOGms-6yCnGrRovBj2yHij5JL6E')
self.assertEqual('zhangsan', user_id)
def test_upload_media(self):
media_file = six.StringIO('nothing')
with HTTMock(wechat_api_mock):
media = self.client.media.upload('image', media_file)
self.assertEqual('image', media['type'])
self.assertEqual('12345678', media['media_id'])
def test_material_get_count(self):
with HTTMock(wechat_api_mock):
res = self.client.material.get_count(1)
self.assertEqual(37, res['total_count'])
self.assertEqual(3, res['video_count'])
self.assertEqual(10, res['voice_count'])
self.assertEqual(12, res['image_count'])
self.assertEqual(3, res['file_count'])
self.assertEqual(6, res['mpnews_count'])
def test_material_batchget_mpnews(self):
with HTTMock(wechat_api_mock):
res = self.client.material.batchget(1, 'mpnews')
self.assertEqual('mpnews', res['type'])
self.assertEqual(20, res['total_count'])
self.assertEqual(3, res['item_count'])
self.assertEqual('2-G6nrLmr5EC3MMb_-zK1dDdzmd0p7cNliYu',
res['itemlist'][0]['media_id'])
def test_material_delete(self):
media_id = '2-G6nrLmr5EC3MMb_-zK1dDdzmd0p7cNliYu'
with HTTMock(wechat_api_mock):
res = self.client.material.delete(1, media_id)
self.assertEqual('deleted', res['errmsg'])
def test_material_get_mpnews(self):
media_id = '2-G6nrLmr5EC3MMb_-zK1dDdzmd0p7cNliYu'
with HTTMock(wechat_api_mock):
res = self.client.material.get_articles(1, media_id)
self.assertEqual('mpnews', res['type'])
self.assertEqual(
'2-G6nrLmr5EC3MMb_-zK1dDdzmd0' +
'p7cNliYu9V5w7o8K0HuucGBZCzw4HmLa5C',
res['mpnews']['articles'][0]['thumb_media_id'])
self.assertEqual(
'2-G6nrLmr5EC3MMb_-zK1dDdzmd0' +
'p7cNliYu9V5w7oovsUPf3wG4t9N3tE',
res['mpnews']['articles'][1]['thumb_media_id'])
def test_reraise_requests_exception(self):
@urlmatch(netloc=r'(.*\.)?qyapi\.weixin\.qq\.com$')
def _wechat_api_mock(url, request):
return {'status_code': 404, 'content': '404 not found'}
try:
with HTTMock(_wechat_api_mock):
self.client.material.get_count(1)
except WeChatClientException as e:
self.assertEqual(404, e.response.status_code)
def test_shakearound_get_shake_info(self):
with HTTMock(wechat_api_mock):
res = self.client.shakearound.get_shake_info('123456')
self.assertEqual(14000, res['page_id'])
self.assertEqual('zhangsan', res['userid'])
def test_service_get_provider_token(self):
with HTTMock(wechat_api_mock):
res = self.client.service.get_provider_token('provider_secret')
self.assertEqual(7200, res['expires_in'])
self.assertEqual('enLSZ5xxxxxxJRL', res['provider_access_token'])
def test_service_get_login_info(self):
with HTTMock(wechat_api_mock):
res = self.client.service.get_login_info('enLSZ5xxxxxxJRL',
'auth_code')
self.assertTrue(res['is_sys'])
self.assertTrue(res['is_inner'])
def test_chat_create(self):
with HTTMock(wechat_api_mock):
res = self.client.chat.create('1', 'chat', 'zhangsan',
['zhangsan', 'lisi', 'wangwu'])
self.assertEqual(0, res['errcode'])
def test_chat_get(self):
with HTTMock(wechat_api_mock):
chat = self.client.chat.get('235364212115767297')
self.assertEqual('235364212115767297', chat['chatid'])
self.assertEqual('zhangsan', chat['owner'])
def test_chat_update(self):
with HTTMock(wechat_api_mock):
res = self.client.chat.update('235364212115767297', 'lisi',
'企业应用中心', 'zhangsan', ['zhaoli'],
['zhangsan'])
self.assertEqual(0, res['errcode'])
def test_chat_quit(self):
with HTTMock(wechat_api_mock):
res = self.client.chat.quit('235364212115767297', 'lisi')
self.assertEqual(0, res['errcode'])
def test_chat_clear_notify(self):
with HTTMock(wechat_api_mock):
res = self.client.chat.clear_notify('zhangsan', 'single', 'lisi')
self.assertEqual(0, res['errcode'])
def test_chat_set_mute(self):
mute_list = [
{
'userid': 'zhangsan',
'status': 0
},
{
'userid': 'lisi',
'status': 1
},
]
with HTTMock(wechat_api_mock):
res = self.client.chat.set_mute(mute_list)
self.assertEqual(0, res['errcode'])
self.assertEqual(['zhangsan'], res['invaliduser'])
def test_chat_send_text(self):
with HTTMock(wechat_api_mock):
res = self.client.chat.send_text('zhangsan', 'single', 'lisi',
'hello')
self.assertEqual(0, res['errcode'])
def test_chat_send_image(self):
with HTTMock(wechat_api_mock):
res = self.client.chat.send_image('zhangsan', 'single', 'lisi',
'media_id')
self.assertEqual(0, res['errcode'])
def test_chat_send_file(self):
with HTTMock(wechat_api_mock):
res = self.client.chat.send_file('zhangsan', 'single', 'lisi',
'media_id')
self.assertEqual(0, res['errcode'])
class WeChatClientTestCase(unittest.TestCase):
corp_id = "123456"
secret = "123456"
def setUp(self):
self.client = WeChatClient(self.corp_id, self.secret)
def test_fetch_access_token(self):
with HTTMock(wechat_api_mock):
token = self.client.fetch_access_token()
self.assertEqual("1234567890", token["access_token"])
self.assertEqual(7200, token["expires_in"])
self.assertEqual("1234567890", self.client.access_token)
def test_get_wechat_ips(self):
with HTTMock(wechat_api_mock):
res = self.client.misc.get_wechat_ips()
self.assertEqual(["127.0.0.1"], res)
def test_department_create(self):
with HTTMock(wechat_api_mock):
res = self.client.department.create("Test")
self.assertEqual(2, res["id"])
def test_department_update(self):
with HTTMock(wechat_api_mock):
res = self.client.department.update(2, "Test 1")
self.assertEqual(0, res["errcode"])
def test_department_delete(self):
with HTTMock(wechat_api_mock):
res = self.client.department.delete(2)
self.assertEqual(0, res["errcode"])
def test_department_get(self):
with HTTMock(wechat_api_mock):
res = self.client.department.get()
self.assertEqual(2, len(res))
def test_department_get_users(self):
with HTTMock(wechat_api_mock):
res = self.client.department.get_users(2)
self.assertEqual(1, len(res))
def test_tag_create(self):
with HTTMock(wechat_api_mock):
res = self.client.tag.create("test")
self.assertEqual("1", res["tagid"])
def test_tag_update(self):
with HTTMock(wechat_api_mock):
res = self.client.tag.update(1, "test")
self.assertEqual(0, res["errcode"])
def test_tag_delete(self):
with HTTMock(wechat_api_mock):
res = self.client.tag.delete(1)
self.assertEqual(0, res["errcode"])
def test_tag_get_users(self):
with HTTMock(wechat_api_mock):
res = self.client.tag.get_users(1)
self.assertEqual(1, len(res["userlist"]))
self.assertEqual(1, len(res["partylist"]))
def test_tag_add_users(self):
with HTTMock(wechat_api_mock):
res = self.client.tag.add_users(1, [1, 2, 3])
self.assertEqual(0, res["errcode"])
def test_tag_delete_users(self):
with HTTMock(wechat_api_mock):
res = self.client.tag.delete_users(1, [1, 2, 3])
self.assertEqual(0, res["errcode"])
def test_tag_list(self):
with HTTMock(wechat_api_mock):
res = self.client.tag.list()
self.assertEqual(2, len(res))
def test_batch_invite_user(self):
with HTTMock(wechat_api_mock):
res = self.client.batch.invite_user(
"http://example.com", "123456", "123456", "123|456", [123, 456], (12, 34), ""
)
self.assertEqual(0, res["errcode"])
def test_batch_sync_user(self):
with HTTMock(wechat_api_mock):
res = self.client.batch.sync_user("http://example.com", "123456", "123456", "12345678")
self.assertEqual(0, res["errcode"])
def test_batch_replace_user(self):
with HTTMock(wechat_api_mock):
res = self.client.batch.replace_user("http://example.com", "123456", "123456", "12345678")
self.assertEqual(0, res["errcode"])
def test_batch_replace_party(self):
with HTTMock(wechat_api_mock):
res = self.client.batch.replace_party("http://example.com", "123456", "123456", "12345678")
self.assertEqual(0, res["errcode"])
def test_batch_get_result(self):
with HTTMock(wechat_api_mock):
res = self.client.batch.get_result("123456")
self.assertEqual(0, res["errcode"])
self.assertEqual(1, res["status"])
def test_jsapi_get_ticket(self):
with HTTMock(wechat_api_mock):
result = self.client.jsapi.get_ticket()
self.assertEqual(
"bxLdikRXVbTPdHSM05e5u5sUoXNKd8-41ZO3MhKoyN5OfkWITDGgnr2fwJ0m9E8NYzWKVZvdVtaUgWvsdshFKA", # NOQA
result["ticket"],
)
self.assertEqual(7200, result["expires_in"])
def test_jsapi_get_jsapi_signature(self):
noncestr = "Wm3WZYTPz0wzccnW"
ticket = "sM4AOVdWfPE4DxkXGEs8VMCPGGVi4C3VM0P37wVUCFvkVAy_90u5h9nbSlYy3-Sl-HhTdfl2fzFy1AOcHKP7qg" # NOQA
timestamp = 1414587457
url = "http://mp.weixin.qq.com?params=value"
signature = self.client.jsapi.get_jsapi_signature(noncestr, ticket, timestamp, url)
self.assertEqual("0f9de62fce790f9a083d5c99e95740ceb90c27ed", signature)
def test_user_convert_to_openid(self):
with HTTMock(wechat_api_mock):
res = self.client.user.convert_to_openid("zhangsan")
self.assertEqual("oDOGms-6yCnGrRovBj2yHij5JL6E", res["openid"])
self.assertEqual("wxf874e15f78cc84a7", res["appid"])
def test_user_convert_to_user_id(self):
with HTTMock(wechat_api_mock):
user_id = self.client.user.convert_to_user_id("oDOGms-6yCnGrRovBj2yHij5JL6E")
self.assertEqual("zhangsan", user_id)
def test_upload_media(self):
media_file = six.StringIO("nothing")
with HTTMock(wechat_api_mock):
media = self.client.media.upload("image", media_file)
self.assertEqual("image", media["type"])
self.assertEqual("12345678", media["media_id"])
def test_material_get_count(self):
with HTTMock(wechat_api_mock):
res = self.client.material.get_count(1)
self.assertEqual(37, res["total_count"])
self.assertEqual(3, res["video_count"])
self.assertEqual(10, res["voice_count"])
self.assertEqual(12, res["image_count"])
self.assertEqual(3, res["file_count"])
self.assertEqual(6, res["mpnews_count"])
def test_material_batchget_mpnews(self):
with HTTMock(wechat_api_mock):
res = self.client.material.batchget(1, "mpnews")
self.assertEqual("mpnews", res["type"])
self.assertEqual(20, res["total_count"])
self.assertEqual(3, res["item_count"])
self.assertEqual("2-G6nrLmr5EC3MMb_-zK1dDdzmd0p7cNliYu", res["itemlist"][0]["media_id"])
def test_material_delete(self):
media_id = "2-G6nrLmr5EC3MMb_-zK1dDdzmd0p7cNliYu"
with HTTMock(wechat_api_mock):
res = self.client.material.delete(1, media_id)
self.assertEqual("deleted", res["errmsg"])
def test_material_get_mpnews(self):
media_id = "2-G6nrLmr5EC3MMb_-zK1dDdzmd0p7cNliYu"
with HTTMock(wechat_api_mock):
res = self.client.material.get_articles(1, media_id)
self.assertEqual("mpnews", res["type"])
self.assertEqual(
"2-G6nrLmr5EC3MMb_-zK1dDdzmd0" + "p7cNliYu9V5w7o8K0HuucGBZCzw4HmLa5C",
res["mpnews"]["articles"][0]["thumb_media_id"],
)
self.assertEqual(
"2-G6nrLmr5EC3MMb_-zK1dDdzmd0" + "p7cNliYu9V5w7oovsUPf3wG4t9N3tE",
res["mpnews"]["articles"][1]["thumb_media_id"],
)
def test_reraise_requests_exception(self):
@urlmatch(netloc=r"(.*\.)?qyapi\.weixin\.qq\.com$")
def _wechat_api_mock(url, request):
return {"status_code": 404, "content": "404 not found"}
try:
with HTTMock(_wechat_api_mock):
self.client.material.get_count(1)
except WeChatClientException as e:
self.assertEqual(404, e.response.status_code)
def test_shakearound_get_shake_info(self):
with HTTMock(wechat_api_mock):
res = self.client.shakearound.get_shake_info("123456")
self.assertEqual(14000, res["page_id"])
self.assertEqual("zhangsan", res["userid"])
def test_service_get_provider_token(self):
with HTTMock(wechat_api_mock):
res = self.client.service.get_provider_token("provider_secret")
self.assertEqual(7200, res["expires_in"])
self.assertEqual("enLSZ5xxxxxxJRL", res["provider_access_token"])
def test_service_get_login_info(self):
with HTTMock(wechat_api_mock):
res = self.client.service.get_login_info("enLSZ5xxxxxxJRL", "auth_code")
self.assertTrue(res["is_sys"])
self.assertTrue(res["is_inner"])
def test_chat_create(self):
with HTTMock(wechat_api_mock):
res = self.client.chat.create("1", "chat", "zhangsan", ["zhangsan", "lisi", "wangwu"])
self.assertEqual(0, res["errcode"])
def test_chat_get(self):
with HTTMock(wechat_api_mock):
chat = self.client.chat.get("235364212115767297")
self.assertEqual("235364212115767297", chat["chatid"])
self.assertEqual("zhangsan", chat["owner"])
def test_chat_update(self):
with HTTMock(wechat_api_mock):
res = self.client.chat.update("235364212115767297", "lisi", "企业应用中心", "zhangsan", ["zhaoli"], ["zhangsan"])
self.assertEqual(0, res["errcode"])
def test_chat_quit(self):
with HTTMock(wechat_api_mock):
res = self.client.chat.quit("235364212115767297", "lisi")
self.assertEqual(0, res["errcode"])
def test_chat_clear_notify(self):
with HTTMock(wechat_api_mock):
res = self.client.chat.clear_notify("zhangsan", "single", "lisi")
self.assertEqual(0, res["errcode"])
def test_chat_set_mute(self):
mute_list = [{"userid": "zhangsan", "status": 0}, {"userid": "lisi", "status": 1}]
with HTTMock(wechat_api_mock):
res = self.client.chat.set_mute(mute_list)
self.assertEqual(0, res["errcode"])
self.assertEqual(["zhangsan"], res["invaliduser"])
def test_chat_send_text(self):
with HTTMock(wechat_api_mock):
res = self.client.chat.send_text("zhangsan", "single", "lisi", "hello")
self.assertEqual(0, res["errcode"])
def test_chat_send_image(self):
with HTTMock(wechat_api_mock):
res = self.client.chat.send_image("zhangsan", "single", "lisi", "media_id")
self.assertEqual(0, res["errcode"])
def test_chat_send_file(self):
with HTTMock(wechat_api_mock):
res = self.client.chat.send_file("zhangsan", "single", "lisi", "media_id")
self.assertEqual(0, res["errcode"])
class WeChatClientTestCase(unittest.TestCase):
corp_id = '123456'
secret = '123456'
def setUp(self):
self.client = WeChatClient(self.corp_id, self.secret)
def test_fetch_access_token(self):
with HTTMock(wechat_api_mock):
token = self.client.fetch_access_token()
self.assertEqual('1234567890', token['access_token'])
self.assertEqual(7200, token['expires_in'])
self.assertEqual('1234567890', self.client.access_token)
def test_get_wechat_ips(self):
with HTTMock(wechat_api_mock):
res = self.client.misc.get_wechat_ips()
self.assertEqual(['127.0.0.1'], res)
def test_department_create(self):
with HTTMock(wechat_api_mock):
res = self.client.department.create('Test')
self.assertEqual(2, res['id'])
def test_department_update(self):
with HTTMock(wechat_api_mock):
res = self.client.department.update(2, 'Test 1')
self.assertEqual(0, res['errcode'])
def test_department_delete(self):
with HTTMock(wechat_api_mock):
res = self.client.department.delete(2)
self.assertEqual(0, res['errcode'])
def test_department_get(self):
with HTTMock(wechat_api_mock):
res = self.client.department.get()
self.assertEqual(2, len(res))
def test_department_get_users(self):
with HTTMock(wechat_api_mock):
res = self.client.department.get_users(2)
self.assertEqual(1, len(res))
def test_department_get_users_detail(self):
with HTTMock(wechat_api_mock):
res = self.client.department.get_users(2, simple=False)
self.assertEqual(1, len(res))
def test_department_map_users(self):
with HTTMock(wechat_api_mock):
users = self.client.department.get_map_users(2, key='email')
self.assertEqual(users,
{'*****@*****.**': 'zhangthree-userid'})
users = self.client.department.get_map_users(key='mobile')
self.assertEqual(users, {'15723333333': 'zhangthree-userid'})
def test_tag_create(self):
with HTTMock(wechat_api_mock):
res = self.client.tag.create('test')
self.assertEqual('1', res['tagid'])
def test_tag_update(self):
with HTTMock(wechat_api_mock):
res = self.client.tag.update(1, 'test')
self.assertEqual(0, res['errcode'])
def test_tag_delete(self):
with HTTMock(wechat_api_mock):
res = self.client.tag.delete(1)
self.assertEqual(0, res['errcode'])
def test_tag_get_users(self):
with HTTMock(wechat_api_mock):
res = self.client.tag.get_users(1)
self.assertEqual(1, len(res['userlist']))
self.assertEqual(1, len(res['partylist']))
def test_tag_add_users(self):
with HTTMock(wechat_api_mock):
res = self.client.tag.add_users(1, [1, 2, 3])
self.assertEqual(0, res['errcode'])
def test_tag_delete_users(self):
with HTTMock(wechat_api_mock):
res = self.client.tag.delete_users(1, [1, 2, 3])
self.assertEqual(0, res['errcode'])
def test_tag_list(self):
with HTTMock(wechat_api_mock):
res = self.client.tag.list()
self.assertEqual(2, len(res))
def test_batch_invite_user(self):
with HTTMock(wechat_api_mock):
res = self.client.batch.invite_user('http://example.com', '123456',
'123456', '123|456',
[123, 456], (12, 34), '')
self.assertEqual(0, res['errcode'])
def test_batch_sync_user(self):
with HTTMock(wechat_api_mock):
res = self.client.batch.sync_user('http://example.com', '123456',
'123456', '12345678')
self.assertEqual(0, res['errcode'])
def test_batch_replace_user(self):
with HTTMock(wechat_api_mock):
res = self.client.batch.replace_user('http://example.com',
'123456', '123456',
'12345678')
self.assertEqual(0, res['errcode'])
def test_batch_replace_party(self):
with HTTMock(wechat_api_mock):
res = self.client.batch.replace_party('http://example.com',
'123456', '123456',
'12345678')
self.assertEqual(0, res['errcode'])
def test_batch_get_result(self):
with HTTMock(wechat_api_mock):
res = self.client.batch.get_result('123456')
self.assertEqual(0, res['errcode'])
self.assertEqual(1, res['status'])
def test_jsapi_get_ticket(self):
with HTTMock(wechat_api_mock):
result = self.client.jsapi.get_ticket()
self.assertEqual(
'bxLdikRXVbTPdHSM05e5u5sUoXNKd8-41ZO3MhKoyN5OfkWITDGgnr2fwJ0m9E8NYzWKVZvdVtaUgWvsdshFKA', # NOQA
result['ticket'])
self.assertEqual(7200, result['expires_in'])
def test_jsapi_get_jsapi_signature(self):
noncestr = 'Wm3WZYTPz0wzccnW'
ticket = 'sM4AOVdWfPE4DxkXGEs8VMCPGGVi4C3VM0P37wVUCFvkVAy_90u5h9nbSlYy3-Sl-HhTdfl2fzFy1AOcHKP7qg' # NOQA
timestamp = 1414587457
url = 'http://mp.weixin.qq.com?params=value'
signature = self.client.jsapi.get_jsapi_signature(
noncestr, ticket, timestamp, url)
self.assertEqual('0f9de62fce790f9a083d5c99e95740ceb90c27ed', signature)
def test_user_convert_to_openid(self):
with HTTMock(wechat_api_mock):
res = self.client.user.convert_to_openid('zhangsan')
self.assertEqual('oDOGms-6yCnGrRovBj2yHij5JL6E', res['openid'])
self.assertEqual('wxf874e15f78cc84a7', res['appid'])
def test_user_convert_to_user_id(self):
with HTTMock(wechat_api_mock):
user_id = self.client.user.convert_to_user_id(
'oDOGms-6yCnGrRovBj2yHij5JL6E')
self.assertEqual('zhangsan', user_id)
def test_upload_media(self):
media_file = six.StringIO('nothing')
with HTTMock(wechat_api_mock):
media = self.client.media.upload('image', media_file)
self.assertEqual('image', media['type'])
self.assertEqual('12345678', media['media_id'])
def test_material_get_count(self):
with HTTMock(wechat_api_mock):
res = self.client.material.get_count(1)
self.assertEqual(37, res['total_count'])
self.assertEqual(3, res['video_count'])
self.assertEqual(10, res['voice_count'])
self.assertEqual(12, res['image_count'])
self.assertEqual(3, res['file_count'])
self.assertEqual(6, res['mpnews_count'])
def test_material_batchget_mpnews(self):
with HTTMock(wechat_api_mock):
res = self.client.material.batchget(1, 'mpnews')
self.assertEqual('mpnews', res['type'])
self.assertEqual(20, res['total_count'])
self.assertEqual(3, res['item_count'])
self.assertEqual('2-G6nrLmr5EC3MMb_-zK1dDdzmd0p7cNliYu',
res['itemlist'][0]['media_id'])
def test_material_delete(self):
media_id = '2-G6nrLmr5EC3MMb_-zK1dDdzmd0p7cNliYu'
with HTTMock(wechat_api_mock):
res = self.client.material.delete(1, media_id)
self.assertEqual('deleted', res['errmsg'])
def test_material_get_mpnews(self):
media_id = '2-G6nrLmr5EC3MMb_-zK1dDdzmd0p7cNliYu'
with HTTMock(wechat_api_mock):
res = self.client.material.get_articles(1, media_id)
self.assertEqual('mpnews', res['type'])
self.assertEqual(
'2-G6nrLmr5EC3MMb_-zK1dDdzmd0' +
'p7cNliYu9V5w7o8K0HuucGBZCzw4HmLa5C',
res['mpnews']['articles'][0]['thumb_media_id'])
self.assertEqual(
'2-G6nrLmr5EC3MMb_-zK1dDdzmd0' +
'p7cNliYu9V5w7oovsUPf3wG4t9N3tE',
res['mpnews']['articles'][1]['thumb_media_id'])
def test_reraise_requests_exception(self):
@urlmatch(netloc=r'(.*\.)?qyapi\.weixin\.qq\.com$')
def _wechat_api_mock(url, request):
return {'status_code': 404, 'content': '404 not found'}
try:
with HTTMock(_wechat_api_mock):
self.client.material.get_count(1)
except WeChatClientException as e:
self.assertEqual(404, e.response.status_code)
def test_shakearound_get_shake_info(self):
with HTTMock(wechat_api_mock):
res = self.client.shakearound.get_shake_info('123456')
self.assertEqual(14000, res['page_id'])
self.assertEqual('zhangsan', res['userid'])
def test_service_get_provider_token(self):
with HTTMock(wechat_api_mock):
res = self.client.service.get_provider_token('provider_secret')
self.assertEqual(7200, res['expires_in'])
self.assertEqual('enLSZ5xxxxxxJRL', res['provider_access_token'])
def test_service_get_login_info(self):
with HTTMock(wechat_api_mock):
res = self.client.service.get_login_info('enLSZ5xxxxxxJRL',
'auth_code')
self.assertTrue(res['is_sys'])
self.assertTrue(res['is_inner'])
def test_chat_create(self):
with HTTMock(wechat_api_mock):
res = self.client.chat.create('1', 'chat', 'zhangsan',
['zhangsan', 'lisi', 'wangwu'])
self.assertEqual(0, res['errcode'])
def test_chat_get(self):
with HTTMock(wechat_api_mock):
chat = self.client.chat.get('235364212115767297')
self.assertEqual('235364212115767297', chat['chatid'])
self.assertEqual('zhangsan', chat['owner'])
def test_chat_update(self):
with HTTMock(wechat_api_mock):
res = self.client.chat.update('235364212115767297', 'lisi',
'企业应用中心', 'zhangsan', ['zhaoli'],
['zhangsan'])
self.assertEqual(0, res['errcode'])
def test_chat_quit(self):
with HTTMock(wechat_api_mock):
res = self.client.chat.quit('235364212115767297', 'lisi')
self.assertEqual(0, res['errcode'])
def test_chat_clear_notify(self):
with HTTMock(wechat_api_mock):
res = self.client.chat.clear_notify('zhangsan', 'single', 'lisi')
self.assertEqual(0, res['errcode'])
def test_chat_set_mute(self):
mute_list = [
{
'userid': 'zhangsan',
'status': 0
},
{
'userid': 'lisi',
'status': 1
},
]
with HTTMock(wechat_api_mock):
res = self.client.chat.set_mute(mute_list)
self.assertEqual(0, res['errcode'])
self.assertEqual(['zhangsan'], res['invaliduser'])
def test_chat_send_text(self):
with HTTMock(wechat_api_mock):
res = self.client.chat.send_text('zhangsan', 'single', 'lisi',
'hello')
self.assertEqual(0, res['errcode'])
def test_chat_send_image(self):
with HTTMock(wechat_api_mock):
res = self.client.chat.send_image('zhangsan', 'single', 'lisi',
'media_id')
self.assertEqual(0, res['errcode'])
def test_chat_send_file(self):
with HTTMock(wechat_api_mock):
res = self.client.chat.send_file('zhangsan', 'single', 'lisi',
'media_id')
self.assertEqual(0, res['errcode'])
def test_external_contact_get_follow_user_list(self):
with HTTMock(wechat_api_mock):
res = self.client.external_contact.get_follow_user_list()
self.assertEqual(0, res['errcode'])
def test_external_contact_list(self):
with HTTMock(wechat_api_mock):
res = self.client.external_contact.list('userid')
self.assertEqual(0, res['errcode'])
def test_external_contact_get(self):
with HTTMock(wechat_api_mock):
res = self.client.external_contact.get('external_userid')
self.assertEqual(0, res['errcode'])
def test_external_contact_add_contact_way(self):
with HTTMock(wechat_api_mock):
res = self.client.external_contact.add_contact_way(
1, 1, 1, 'remark', True, 'state', ['UserID1', 'UserID2'],
['PartyID1', 'PartyID2'])
self.assertEqual(0, res['errcode'])
def test_external_contact_get_contact_way(self):
with HTTMock(wechat_api_mock):
res = self.client.external_contact.get_contact_way(
'42b34949e138eb6e027c123cba77fad7')
self.assertEqual(0, res['errcode'])
def test_external_contact_update_contact_way(self):
with HTTMock(wechat_api_mock):
res = self.client.external_contact.update_contact_way(
'42b34949e138eb6e027c123cba77fad7', '渠道客户', True, 1,
'teststate', ['UserID1', 'UserID2', 'UserID3'],
['PartyID1', 'PartyID2'])
self.assertEqual(0, res['errcode'])
def test_external_contact_del_contact_way(self):
with HTTMock(wechat_api_mock):
res = self.client.external_contact.del_contact_way(
'42b34949e138eb6e027c123cba77fad7')
self.assertEqual(0, res['errcode'])
def test_external_contact_add_msg_template(self):
with HTTMock(wechat_api_mock):
res = self.client.external_contact.add_msg_template({
"external_userid": [
"woAJ2GCAAAXtWyujaWJHDDGi0mACas1w",
"wmqfasd1e1927831291723123109r712"
],
"sender":
"zhangsan",
"text": {
"content": "文本消息内容"
},
"image": {
"media_id": "MEDIA_ID"
}
})
self.assertEqual(0, res['errcode'])
def test_external_contact_get_group_msg_result(self):
with HTTMock(wechat_api_mock):
res = self.client.external_contact.get_group_msg_result(
'msgGCAAAXtWyujaWJHDDGi0mACas1w')
self.assertEqual(0, res['errcode'])
def test_external_contact_get_user_behavior_data(self):
with HTTMock(wechat_api_mock):
res = self.client.external_contact.get_user_behavior_data(
["zhangsan", "lisi"], 1536508800, 1536940800)
self.assertEqual(0, res['errcode'])
def test_external_contact_send_welcome_msg(self):
with HTTMock(wechat_api_mock):
res = self.client.external_contact.send_welcome_msg({
"welcome_code":
"CALLBACK_CODE",
"text": {
"content": "文本消息内容"
},
"image": {
"media_id": "MEDIA_ID"
},
"link": {
"title": "消息标题",
"picurl": "https://example.pic.com/path",
"desc": "消息描述",
"url": "https://example.link.com/path"
},
"miniprogram": {
"title": "消息标题",
"pic_media_id": "MEDIA_ID",
"appid": "wx8bd80126147df384",
"page": "/path/index"
}
})
self.assertEqual(0, res['errcode'])
def test_external_contact_get_unassigned_list(self):
with HTTMock(wechat_api_mock):
res = self.client.external_contact.get_unassigned_list(0, 100)
self.assertEqual(0, res['errcode'])
def test_external_contact_transfer(self):
with HTTMock(wechat_api_mock):
res = self.client.external_contact.transfer(
"woAJ2GCAAAXtWyujaWJHDDGi0mACH71w", "zhangsan", "lisi")
self.assertEqual(0, res['errcode'])
class WeChatClientTestCase(unittest.TestCase):
corp_id = '123456'
secret = '123456'
def setUp(self):
self.client = WeChatClient(self.corp_id, self.secret)
def test_fetch_access_token(self):
with HTTMock(wechat_api_mock):
token = self.client.fetch_access_token()
self.assertEqual('1234567890', token['access_token'])
self.assertEqual(7200, token['expires_in'])
self.assertEqual('1234567890', self.client.access_token)
def test_get_wechat_ips(self):
with HTTMock(wechat_api_mock):
res = self.client.misc.get_wechat_ips()
self.assertEqual(['127.0.0.1'], res)
def test_department_create(self):
with HTTMock(wechat_api_mock):
res = self.client.department.create('Test')
self.assertEqual(2, res['id'])
def test_department_update(self):
with HTTMock(wechat_api_mock):
res = self.client.department.update(2, 'Test 1')
self.assertEqual(0, res['errcode'])
def test_department_delete(self):
with HTTMock(wechat_api_mock):
res = self.client.department.delete(2)
self.assertEqual(0, res['errcode'])
def test_department_get(self):
with HTTMock(wechat_api_mock):
res = self.client.department.get()
self.assertEqual(2, len(res))
def test_department_get_users(self):
with HTTMock(wechat_api_mock):
res = self.client.department.get_users(2)
self.assertEqual(1, len(res))
def test_tag_create(self):
with HTTMock(wechat_api_mock):
res = self.client.tag.create('test')
self.assertEqual('1', res['tagid'])
def test_tag_update(self):
with HTTMock(wechat_api_mock):
res = self.client.tag.update(1, 'test')
self.assertEqual(0, res['errcode'])
def test_tag_delete(self):
with HTTMock(wechat_api_mock):
res = self.client.tag.delete(1)
self.assertEqual(0, res['errcode'])
def test_tag_get_users(self):
with HTTMock(wechat_api_mock):
res = self.client.tag.get_users(1)
self.assertEqual(1, len(res['userlist']))
self.assertEqual(1, len(res['partylist']))
def test_tag_add_users(self):
with HTTMock(wechat_api_mock):
res = self.client.tag.add_users(1, [1, 2, 3])
self.assertEqual(0, res['errcode'])
def test_tag_delete_users(self):
with HTTMock(wechat_api_mock):
res = self.client.tag.delete_users(1, [1, 2, 3])
self.assertEqual(0, res['errcode'])
def test_tag_list(self):
with HTTMock(wechat_api_mock):
res = self.client.tag.list()
self.assertEqual(2, len(res))
def test_batch_invite_user(self):
with HTTMock(wechat_api_mock):
res = self.client.batch.invite_user('http://example.com', '123456',
'123456', '123|456',
[123, 456], (12, 34), '')
self.assertEqual(0, res['errcode'])
def test_batch_sync_user(self):
with HTTMock(wechat_api_mock):
res = self.client.batch.sync_user('http://example.com', '123456',
'123456', '12345678')
self.assertEqual(0, res['errcode'])
def test_batch_replace_user(self):
with HTTMock(wechat_api_mock):
res = self.client.batch.replace_user('http://example.com',
'123456', '123456',
'12345678')
self.assertEqual(0, res['errcode'])
def test_batch_replace_party(self):
with HTTMock(wechat_api_mock):
res = self.client.batch.replace_party('http://example.com',
'123456', '123456',
'12345678')
self.assertEqual(0, res['errcode'])
def test_batch_get_result(self):
with HTTMock(wechat_api_mock):
res = self.client.batch.get_result('123456')
self.assertEqual(0, res['errcode'])
self.assertEqual(1, res['status'])
class WeChatClientTestCase(unittest.TestCase):
corp_id = '123456'
secret = '123456'
def setUp(self):
self.client = WeChatClient(self.corp_id, self.secret)
def test_fetch_access_token(self):
with HTTMock(wechat_api_mock):
token = self.client.fetch_access_token()
self.assertEqual('1234567890', token['access_token'])
self.assertEqual(7200, token['expires_in'])
self.assertEqual('1234567890', self.client.access_token)
def test_get_wechat_ips(self):
with HTTMock(wechat_api_mock):
res = self.client.misc.get_wechat_ips()
self.assertEqual(['127.0.0.1'], res)
def test_department_create(self):
with HTTMock(wechat_api_mock):
res = self.client.department.create('Test')
self.assertEqual(2, res['id'])
def test_department_update(self):
with HTTMock(wechat_api_mock):
res = self.client.department.update(2, 'Test 1')
self.assertEqual(0, res['errcode'])
def test_department_delete(self):
with HTTMock(wechat_api_mock):
res = self.client.department.delete(2)
self.assertEqual(0, res['errcode'])
def test_department_get(self):
with HTTMock(wechat_api_mock):
res = self.client.department.get()
self.assertEqual(2, len(res))
def test_department_get_users(self):
with HTTMock(wechat_api_mock):
res = self.client.department.get_users(2)
self.assertEqual(1, len(res))
def test_tag_create(self):
with HTTMock(wechat_api_mock):
res = self.client.tag.create('test')
self.assertEqual('1', res['tagid'])
def test_tag_update(self):
with HTTMock(wechat_api_mock):
res = self.client.tag.update(1, 'test')
self.assertEqual(0, res['errcode'])
def test_tag_delete(self):
with HTTMock(wechat_api_mock):
res = self.client.tag.delete(1)
self.assertEqual(0, res['errcode'])
def test_tag_get_users(self):
with HTTMock(wechat_api_mock):
res = self.client.tag.get_users(1)
self.assertEqual(1, len(res['userlist']))
self.assertEqual(1, len(res['partylist']))
def test_tag_add_users(self):
with HTTMock(wechat_api_mock):
res = self.client.tag.add_users(1, [1, 2, 3])
self.assertEqual(0, res['errcode'])
def test_tag_delete_users(self):
with HTTMock(wechat_api_mock):
res = self.client.tag.delete_users(1, [1, 2, 3])
self.assertEqual(0, res['errcode'])
def test_tag_list(self):
with HTTMock(wechat_api_mock):
res = self.client.tag.list()
self.assertEqual(2, len(res))
class WeChatClientTestCase(unittest.TestCase):
corp_id = '123456'
secret = '123456'
def setUp(self):
self.client = WeChatClient(self.corp_id, self.secret)
def test_fetch_access_token(self):
with HTTMock(wechat_api_mock):
token = self.client.fetch_access_token()
self.assertEqual('1234567890', token['access_token'])
self.assertEqual(7200, token['expires_in'])
self.assertEqual('1234567890', self.client.access_token)
def test_get_wechat_ips(self):
with HTTMock(wechat_api_mock):
res = self.client.misc.get_wechat_ips()
self.assertEqual(['127.0.0.1'], res)
def test_department_create(self):
with HTTMock(wechat_api_mock):
res = self.client.department.create('Test')
self.assertEqual(2, res['id'])
def test_department_update(self):
with HTTMock(wechat_api_mock):
res = self.client.department.update(2, 'Test 1')
self.assertEqual(0, res['errcode'])
def test_department_delete(self):
with HTTMock(wechat_api_mock):
res = self.client.department.delete(2)
self.assertEqual(0, res['errcode'])
def test_department_get(self):
with HTTMock(wechat_api_mock):
res = self.client.department.get()
self.assertEqual(2, len(res))
def test_department_get_users(self):
with HTTMock(wechat_api_mock):
res = self.client.department.get_users(2)
self.assertEqual(1, len(res))
def test_tag_create(self):
with HTTMock(wechat_api_mock):
res = self.client.tag.create('test')
self.assertEqual('1', res['tagid'])
def test_tag_update(self):
with HTTMock(wechat_api_mock):
res = self.client.tag.update(1, 'test')
self.assertEqual(0, res['errcode'])
def test_tag_delete(self):
with HTTMock(wechat_api_mock):
res = self.client.tag.delete(1)
self.assertEqual(0, res['errcode'])
def test_tag_get_users(self):
with HTTMock(wechat_api_mock):
res = self.client.tag.get_users(1)
self.assertEqual(1, len(res['userlist']))
self.assertEqual(1, len(res['partylist']))
def test_tag_add_users(self):
with HTTMock(wechat_api_mock):
res = self.client.tag.add_users(1, [1, 2, 3])
self.assertEqual(0, res['errcode'])
def test_tag_delete_users(self):
with HTTMock(wechat_api_mock):
res = self.client.tag.delete_users(1, [1, 2, 3])
self.assertEqual(0, res['errcode'])
def test_tag_list(self):
with HTTMock(wechat_api_mock):
res = self.client.tag.list()
self.assertEqual(2, len(res))
def test_batch_invite_user(self):
with HTTMock(wechat_api_mock):
res = self.client.batch.invite_user(
'http://example.com',
'123456',
'123456',
'123|456',
[123, 456],
(12, 34),
''
)
self.assertEqual(0, res['errcode'])
def test_batch_sync_user(self):
with HTTMock(wechat_api_mock):
res = self.client.batch.sync_user(
'http://example.com',
'123456',
'123456',
'12345678'
)
self.assertEqual(0, res['errcode'])
def test_batch_replace_user(self):
with HTTMock(wechat_api_mock):
res = self.client.batch.replace_user(
'http://example.com',
'123456',
'123456',
'12345678'
)
self.assertEqual(0, res['errcode'])
def test_batch_replace_party(self):
with HTTMock(wechat_api_mock):
res = self.client.batch.replace_party(
'http://example.com',
'123456',
'123456',
'12345678'
)
self.assertEqual(0, res['errcode'])
def test_batch_get_result(self):
with HTTMock(wechat_api_mock):
res = self.client.batch.get_result('123456')
self.assertEqual(0, res['errcode'])
self.assertEqual(1, res['status'])