在python中模拟单元测试中的url

import unittest
from unittest import mock

def get_data_from_url(url):
    # 假设这个函数会发送真实的URL请求并返回数据
    # 在单元测试中，我们希望模拟这个函数的行为
    # 而不是真实地发送请求
    response = requests.get(url)
    return response.json()

class TestURL(unittest.TestCase):
    @mock.patch('requests.get')
    def test_get_data_from_url(self, mock_get):
        # 模拟requests.get函数的返回值
        mock_get.return_value.json.return_value = {'key': 'value'}

        # 调用被测试函数
        result = get_data_from_url('http://example.com/api')

        # 断言函数的返回值是否符合预期
        self.assertEqual(result, {'key': 'value'})

        # 断言模拟函数的调用情况
        mock_get.assert_called_once_with('http://example.com/api')

if __name__ == '__main__':
    unittest.main()