mirror of
https://github.com/masterking32/MasterHttpRelayVPN.git
synced 2026-05-17 21:24:37 +03:00
AMMIROSOH
67 lines
2.3 KiB
Python
67 lines
2.3 KiB
Python
import pathlib
|
|
import sys
|
|
import unittest
|
|
|
|
|
|
ROOT = pathlib.Path(__file__).resolve().parents[1]
|
|
SRC = ROOT / "src"
|
|
if str(SRC) not in sys.path:
|
|
sys.path.insert(0, str(SRC))
|
|
|
|
from core.constants import FRONT_SNI_POOL_GOOGLE
|
|
from relay.fronting_support import (
|
|
build_sni_pool,
|
|
parse_content_range,
|
|
validate_range_response,
|
|
)
|
|
|
|
|
|
class BuildSniPoolTests(unittest.TestCase):
|
|
def test_uses_explicit_overrides_with_normalization_and_deduplication(self):
|
|
result = build_sni_pool(
|
|
"www.google.com",
|
|
[" Mail.Google.com ", "mail.google.com", "accounts.google.com."],
|
|
)
|
|
self.assertEqual(result, ["mail.google.com", "accounts.google.com"])
|
|
|
|
def test_google_front_domain_uses_google_pool(self):
|
|
result = build_sni_pool("www.google.com", None)
|
|
self.assertEqual(result, list(FRONT_SNI_POOL_GOOGLE))
|
|
|
|
def test_non_google_front_domain_falls_back_to_single_host(self):
|
|
self.assertEqual(build_sni_pool("cdn.example.com", None), ["cdn.example.com"])
|
|
self.assertEqual(build_sni_pool("", None), ["www.google.com"])
|
|
|
|
|
|
class RangeParsingTests(unittest.TestCase):
|
|
def test_parse_content_range_accepts_valid_values(self):
|
|
self.assertEqual(parse_content_range("bytes 10-19/20"), (10, 19, 20))
|
|
|
|
def test_parse_content_range_rejects_invalid_values(self):
|
|
self.assertIsNone(parse_content_range(""))
|
|
self.assertIsNone(parse_content_range("bytes 10-19/*"))
|
|
self.assertIsNone(parse_content_range("bytes 10-9/20"))
|
|
self.assertIsNone(parse_content_range("bytes 10-19/19"))
|
|
|
|
def test_validate_range_response_success_and_failures(self):
|
|
headers = {"content-range": "bytes 10-13/20"}
|
|
body = b"data"
|
|
|
|
self.assertIsNone(validate_range_response(206, headers, body, 10, 13, 20))
|
|
self.assertEqual(
|
|
validate_range_response(200, headers, body, 10, 13, 20),
|
|
"status 200",
|
|
)
|
|
self.assertEqual(
|
|
validate_range_response(206, {"content-range": "bytes 11-13/20"}, body, 10, 13, 20),
|
|
"Content-Range mismatch 11-13",
|
|
)
|
|
self.assertEqual(
|
|
validate_range_response(206, headers, b"abc", 10, 13, 20),
|
|
"short chunk 3/4 B",
|
|
)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|