返回提交历史
Modified
etc/unittest/backend.py
+3
-3
XFEstudio/gpt4free
fix: Update socket import path in safety URL tests
7bee6870
代码差异
1 个文件
+3
-3
@@ -44,19 +44,19 @@ class TestBackendApi(unittest.TestCase):
44
44
self.assertIsInstance(response, list)
45
45
self.assertTrue(len(response) > 0)
46
46
47
@patch('g4f.gui.server.backend_api.socket.getaddrinfo')
47
@patch('g4f.image.socket.getaddrinfo')
48
48
def test_is_safe_url_with_backslash_confusion(self, mock_getaddrinfo):
49
49
mock_getaddrinfo.return_value = [(socket.AF_INET, socket.SOCK_STREAM, 6, '', ('127.0.0.1', 0))]
50
50
from g4f.gui.server.backend_api import _is_safe_url
51
51
self.assertFalse(_is_safe_url('http://127.0.0.1:6666\\@www.baidu.com'))
52
52
53
@patch('g4f.gui.server.backend_api.socket.getaddrinfo')
53
@patch('g4f.image.socket.getaddrinfo')
54
54
def test_is_safe_url_blocks_private(self, mock_getaddrinfo):
55
55
mock_getaddrinfo.return_value = [(socket.AF_INET, socket.SOCK_STREAM, 6, '', ('127.0.0.1', 0))]
56
56
from g4f.gui.server.backend_api import _is_safe_url
57
57
self.assertFalse(_is_safe_url('http://127.0.0.1'))
58
58
59
@patch('g4f.gui.server.backend_api.socket.getaddrinfo')
59
@patch('g4f.image.socket.getaddrinfo')
60
60
def test_is_safe_url_allows_public(self, mock_getaddrinfo):
61
61
mock_getaddrinfo.return_value = [(socket.AF_INET, socket.SOCK_STREAM, 6, '', ('8.8.8.8', 0))]
62
62
from g4f.gui.server.backend_api import _is_safe_url