上传post的demo：发送弹幕

testCase/live/test_enter_live.py → testCase/live/test_live_enter.py

@@ -20,7 +20,7 @@ class LiveEnter(LiveCase):
         游客进入直播间
         '''
         rep = requests.get(self.url, params=self.params).json()
-        assert rep['error'] == 0, rep
+        self.assertEqual(rep['error'], 0, rep)
         print('游客进入直播间成功！')
 
     @require_login(settings.LIVE_WATCHER)
@@ -29,7 +29,7 @@ class LiveEnter(LiveCase):
         app用户进入直播间
         '''
         rep = requests.get(self.url, params=self.params).json()
-        assert rep['error'] == 0, rep
+        self.assertEqual(rep['error'], 0, rep)
         print('app用户进入直播间成功！')
 
     @require_login(settings.LIVE_USER)
@@ -38,5 +38,5 @@ class LiveEnter(LiveCase):
         主播进入直播间
         '''
         rep = requests.get(self.url, params=self.params).json()
-        assert rep['error'] == 0, rep
+        self.assertEqual(rep['error'], 0, rep)
         print('主播进入直播间成功！')

testCase/live/test_live_sendmsg.py

+import requests
+
+from utils.gmhttp import require_login
+from .livecase import LiveCase
+from conf import settings
+
+
+class LiveSendMsg(LiveCase):
+    '''发送弹幕'''
+
+    def setUp(self):
+        uri = '/api/live/sendmsg'
+        self.url = self.host + uri
+        self.data = {'channel_id': self.channel_id,
+                     'msg': '这是一条羞涩的弹幕'
+                     }
+        self.data_warn = {'channel_id': self.channel_id,
+                          'msg': '习近平，该条弹幕应被拦截'
+                          }
+
+    def test_live_sendmsg_visitor(self):
+        '''
+        游客进入直播间
+        '''
+        rep = requests.post(self.url, data=self.data).json()
+        self.assertEqual(rep['error_code'], 70000, rep)
+        print('提示游客登录。')
+
+    @require_login(settings.LIVE_WATCHER)
+    def test_live_sendmsg_user(self):
+        '''
+        app用户进入直播间
+        '''
+        rep = requests.post(self.url, data=self.data).json()
+        self.assertEqual(rep['error'], 1, rep)
+        print('app用户发送弹幕成功！')
+
+    @require_login(settings.LIVE_USER)
+    def test_live_sendmsg_author(self):
+        '''
+        主播进入直播间
+        '''
+        rep = requests.post(self.url, self.data).json()
+        self.assertEqual(rep['error'], 0, rep)
+        print('主播发送弹幕成功！')
+
+    @require_login(settings.LIVE_WATCHER)
+    def test_live_sendmsg_warn(self):
+        '''
+        发送敏感词被成功拦截
+        '''
+        rep = requests.post(self.url, data=self.data_warn).json()
+
+        self.assertEqual(rep['error_code'], 96604, rep)
+        self.assertEqual(rep['message'], '您输入的内容包含敏感信息，请重新输入', rep)
+        print('敏感词拦截成功！')

utils/gmhttp.py

@@ -43,8 +43,10 @@ def require_login(arg=None, **user_dict):
             require_login.uid = newhttp.uid
             try:
                 return func(*args, **kwargs)
-            except Exception:
-                pass
+            except AssertionError as e:
+                raise e
+            except Exception as e:
+                raise e
             finally:
                 setattr(func_module, 'requests', source_requests)
         return _inner