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()