124 lines
6.9 KiB
Python
124 lines
6.9 KiB
Python
#!/usr/bin/env python
|
||
# create time: 16/01/2018 14:37
|
||
import unittest
|
||
|
||
from dingtalkchatbot.chatbot import (ActionCard, CardItem, DingtalkChatbot,
|
||
FeedLink, is_not_null_and_blank_str)
|
||
|
||
__author__ = 'Devin -- http://zhangchuzhao.site'
|
||
|
||
|
||
class TestDingtalkChatbot(unittest.TestCase):
|
||
"""DingtalkChatbot 测试用例"""
|
||
|
||
@classmethod
|
||
def setUpClass(cls):
|
||
cls.webhook = 'https://oapi.dingtalk.com/robot/send?access_token=52d9034cc78680bc0d4ba6a65748e77fa7b96ee43d57b96116910606f7863d59'
|
||
cls.xiaoding = DingtalkChatbot(cls.webhook)
|
||
|
||
def test_is_not_null_and_blank_str(self):
|
||
"""测试字符串不为空函数"""
|
||
self.assertFalse(is_not_null_and_blank_str(''), 'pass')
|
||
self.assertFalse(is_not_null_and_blank_str(' '), 'pass')
|
||
self.assertFalse(is_not_null_and_blank_str(' '), 'pass')
|
||
self.assertTrue(is_not_null_and_blank_str('abc'), 'pass')
|
||
self.assertTrue(is_not_null_and_blank_str('123'), 'pass')
|
||
|
||
def test_send_text(self):
|
||
"""测试发送文本消息函数"""
|
||
result = self.xiaoding.send_text(msg='我就是小丁，小丁就是我！', is_at_all=True)
|
||
self.assertEqual(result['errcode'], 0)
|
||
|
||
def test_send_image(self):
|
||
"""测试发送表情图片消息函数"""
|
||
result = self.xiaoding.send_image(
|
||
pic_url='http://uc-test-manage-00.umlife.net/jenkins/pic/flake8.png')
|
||
self.assertEqual(result['errcode'], 0)
|
||
|
||
def test_send_link(self):
|
||
"""测试发送链接消息函数"""
|
||
result = self.xiaoding.send_link(title='万万没想到，某小璐竟然...', text='故事是这样子的...',
|
||
message_url='http://www.kwongwah.com.my/?p=454748", pic_url="https://pbs.twimg.com/media/CEwj7EDWgAE5eIF.jpg')
|
||
self.assertEqual(result['errcode'], 0)
|
||
|
||
def test_send_markdown(self):
|
||
"""测试发送Markdown格式消息函数"""
|
||
result = self.xiaoding.send_markdown(title='氧气文字', text='#### 广州天气\n'
|
||
'> 9度，西北风1级，空气良89，相对温度73%\n\n'
|
||
'> \n'
|
||
'> ###### 10点20分发布 [天气](http://www.thinkpage.cn/) \n',
|
||
is_at_all=True)
|
||
self.assertEqual(result['errcode'], 0)
|
||
|
||
def test_send_actioncard(self):
|
||
"""测试发送整体跳转ActionCard消息功能（CardItem新API)"""
|
||
btns1 = [CardItem(title='查看详情', url='https://www.dingtalk.com/')]
|
||
actioncard1 = ActionCard(title='万万没想到，竟然...',
|
||
text=' \n### 故事是这样子的...',
|
||
btns=btns1,
|
||
btn_orientation=1,
|
||
hide_avatar=1)
|
||
result = self.xiaoding.send_action_card(actioncard1)
|
||
self.assertEqual(result['errcode'], 0)
|
||
|
||
"""测试发送单独跳转ActionCard消息功能"""
|
||
btns2 = [CardItem(title='支持', url='https://www.dingtalk.com/'), CardItem(
|
||
title='反对', url='http://www.back china.com/news/2018/01/11/537468.html')]
|
||
actioncard2 = ActionCard(title='万万没想到，竟然...',
|
||
text=' \n### 故事是这样子的...',
|
||
btns=btns2,
|
||
btn_orientation=1,
|
||
hide_avatar=1)
|
||
result = self.xiaoding.send_action_card(actioncard2)
|
||
self.assertEqual(result['errcode'], 0)
|
||
|
||
def test_send_actioncard_old_api(self):
|
||
"""测试发送整体跳转ActionCard消息功能（数据列表btns旧API)"""
|
||
btns1 = [{'title': '查看详情', 'actionURL': 'https://www.dingtalk.com/'}]
|
||
actioncard1 = ActionCard(title='万万没想到，竟然...',
|
||
text=' \n### 故事是这样子的...',
|
||
btns=btns1,
|
||
btn_orientation=1,
|
||
hide_avatar=1)
|
||
result = self.xiaoding.send_action_card(actioncard1)
|
||
self.assertEqual(result['errcode'], 0)
|
||
|
||
"""测试发送单独跳转ActionCard消息功能"""
|
||
btns2 = [{'title': '支持', 'actionURL': 'https://www.dingtalk.com/'},
|
||
{'title': '反对', 'actionURL': 'http://www.back china.com/news/2018/01/11/537468.html'}]
|
||
actioncard2 = ActionCard(title='万万没想到，竟然...',
|
||
text=' \n### 故事是这样子的...',
|
||
btns=btns2,
|
||
btn_orientation=1,
|
||
hide_avatar=1)
|
||
result = self.xiaoding.send_action_card(actioncard2)
|
||
self.assertEqual(result['errcode'], 0)
|
||
|
||
def test_send_feedcard(self):
|
||
"""测试发送FeedCard类型消息功能（CardItem新API)"""
|
||
carditem1 = CardItem(title='氧气美女', url='https://www.dingtalk.com/',
|
||
pic_url='https://unzippedtv.com/wp-content/uploads/sites/28/2016/02/asian.jpg')
|
||
carditem2 = CardItem(title='氧眼美女', url='https://www.dingtalk.com/',
|
||
pic_url='https://unzippedtv.com/wp-content/uploads/sites/28/2016/02/asian.jpg')
|
||
carditem3 = CardItem(title='氧神美女', url='https://www.dingtalk.com/',
|
||
pic_url='https://unzippedtv.com/wp-content/uploads/sites/28/2016/02/asian.jpg')
|
||
cards = [carditem1, carditem2, carditem3]
|
||
result = self.xiaoding.send_feed_card(cards)
|
||
self.assertEqual(result['errcode'], 0)
|
||
|
||
def test_send_feedcard_old_api(self):
|
||
"""测试发送FeedCard类型消息功能(FeedLink旧API)"""
|
||
feedlink1 = FeedLink(title='氧气美女', message_url='https://www.dingtalk.com/',
|
||
pic_url='https://unzippedtv.com/wp-content/uploads/sites/28/2016/02/asian.jpg')
|
||
feedlink2 = FeedLink(title='氧眼美女', message_url='https://www.dingtalk.com/',
|
||
pic_url='https://unzippedtv.com/wp-content/uploads/sites/28/2016/02/asian.jpg')
|
||
feedlink3 = FeedLink(title='氧神美女', message_url='https://www.dingtalk.com/',
|
||
pic_url='https://unzippedtv.com/wp-content/uploads/sites/28/2016/02/asian.jpg')
|
||
links = [feedlink1, feedlink2, feedlink3]
|
||
result = self.xiaoding.send_feed_card(links)
|
||
self.assertEqual(result['errcode'], 0)
|
||
|
||
|
||
if __name__ == '__main__':
|
||
unittest.main()
|