chenchenghao/sglang
1
0
Fork 0
Code Issues Pull Requests Actions 11 Packages Projects Releases Wiki Activity

[Utils] Move network/socket utilities from common.py to network.py (#20646)

Browse Source
This commit is contained in:
Liangsheng Yin
2026-03-15 20:35:24 -07:00
committed by GitHub
parent afc71bae3a
commit f0458e0b49
28 changed files with 418 additions and 422 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
test/registered/utils/test_socket_utils.py
View File

@@ -3,7 +3,7 @@ import socket
import unittest
from unittest.mock import patch
from sglang.srt.utils.common import (
from sglang.srt.utils.network import (
_get_addrinfos_for_bind,
bind_port,
get_free_port,
@@ -78,7 +78,7 @@ class TestTryBindSocket(CustomTestCase):
def test_gaierror_fallback(self):
"""_get_addrinfos_for_bind should fall back to AF_INET on gaierror."""
with patch(
"sglang.srt.utils.common.socket.getaddrinfo",
"sglang.srt.utils.network.socket.getaddrinfo",
side_effect=socket.gaierror("mocked"),
):
infos = _get_addrinfos_for_bind()
@@ -90,7 +90,7 @@ class TestTryBindSocket(CustomTestCase):
def test_gaierror_fallback_preserves_host(self):
"""Fallback should use the provided host, not default to 0.0.0.0."""
with patch(
"sglang.srt.utils.common.socket.getaddrinfo",
"sglang.srt.utils.network.socket.getaddrinfo",
side_effect=socket.gaierror("mocked"),
):
infos = _get_addrinfos_for_bind(host="10.0.0.1", port=8080)
@@ -187,7 +187,7 @@ class TestReservePort(CustomTestCase):
def test_reserve_port_no_free_port_raises(self):
"""reserve_port should raise RuntimeError if no port is available."""
with patch(
"sglang.srt.utils.common.try_bind_socket",
"sglang.srt.utils.network.try_bind_socket",
side_effect=OSError("mocked"),
):
with self.assertRaises(RuntimeError):