mirror of
https://github.com/alerta/alerta-contrib.git
synced 2025-03-17 05:52:38 +00:00
114 lines
6.7 KiB
Python
114 lines
6.7 KiB
Python
#!/usr/bin/env python
|
||
# _*_ coding:utf-8 _*_
|
||
# create time: 16/01/2018 14:37
|
||
import unittest
|
||
|
||
from dingtalkchatbot.chatbot import DingtalkChatbot, is_not_null_and_blank_str, ActionCard, FeedLink, CardItem
|
||
|
||
__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()
|