ADD 单元测试代码

test_bullet_comments_crawler.py

@@ -0,0 +1,66 @@
+import unittest
+from unittest.mock import patch # 使用 @patch 装饰器来模拟 requests.get 方法
+import requests
+from bullet_comments_crawler import get_bvid, get_cid, get_comment
+
+class TestBulletCommentsCrawler(unittest.TestCase):
+    """
+    测试 bullet_comments_crawler 模块中的函数。
+    """
+
+    @patch('requests.get')
+    def test_get_bvid_valid(self, mock_get):
+        """
+        测试 get_bvid 函数在接收到有效响应时的行为。
+        """
+        mock_get.return_value.text = 'aid:1,bvid:"BV12XYZ",'
+        result = get_bvid('http://valid.url', {})
+        self.assertIn('BV12XYZ', result)
+
+    @patch('requests.get')
+    def test_get_bvid_invalid(self, mock_get):
+        """
+        测试 get_bvid 函数在接收到无效响应时的行为。
+        """
+        mock_get.side_effect = requests.exceptions.RequestException("Invalid URL")
+        result = get_bvid('http://invalid.url', {})
+        self.assertEqual(result, set())
+
+    @patch('requests.get')
+    def test_get_cid_valid(self, mock_get):
+        """
+        测试 get_cid 函数在接收到有效 JSON 响应时的行为。
+        """
+        mock_get.return_value.json.return_value = {'data': [{'cid': 111}]}
+        cid = get_cid('http://valid.api.url', {})
+        self.assertEqual(cid, 111)
+
+    @patch('requests.get')
+    def test_get_cid_invalid(self, mock_get):
+        """
+        测试 get_cid 函数在接收到无效 JSON 响应时的行为。
+        """
+        mock_get.return_value.json.return_value = {'data': []}
+        cid = get_cid('http://invalid.api.url', {})
+        self.assertIsNone(cid)
+
+    @patch('requests.get')
+    def test_get_comment_valid(self, mock_get):
+        """
+        测试 get_comment 函数在接收到有效 XML 响应时的行为。
+        """
+        mock_get.return_value.text = '<xml><d>弹幕1</d><d>弹幕2</d></xml>'
+        comments = get_comment('http://valid.comment.url', {})
+        self.assertEqual(comments, ['弹幕1', '弹幕2'])
+
+    @patch('requests.get')
+    def test_get_comment_invalid(self, mock_get):
+        """
+        测试 get_comment 函数在接收到无效响应时的行为。
+        """
+        mock_get.side_effect = requests.exceptions.RequestException("Invalid URL")
+        comments = get_comment('http://invalid.comment.url', {})
+        self.assertEqual(comments, [])
+
+if __name__ == '__main__':
+    unittest.main()