update tests/unit/test_patterns.py

tests/unit/test_patterns.py

@@ -0,0 +1,71 @@
+import pytest
+from app.grabber.patterns import ProxyPatterns
+
+
+class TestHTTPPatterns:
+    def test_http_ip_port_pattern(self):
+        text = "192.168.1.1:8080\n10.0.0.1:3128"
+        matches = ProxyPatterns.extract_http_proxies(text)
+
+        assert len(matches) == 2
+        assert matches[0] == ("192.168.1.1", "8080")
+        assert matches[1] == ("10.0.0.1", "3128")
+
+    def test_http_url_pattern(self):
+        text = "192.168.1.1:8080\n10.0.0.1:3128"
+        matches = ProxyPatterns.extract_http_proxies(text)
+
+        assert len(matches) >= 1
+
+
+class TestVMessPatterns:
+    def test_vmess_url_detection(self):
+        url = "vmess://eyJhZGQiOiIxMjcuMC4wLjEiLCJwb3J0Ijo0NDMsImlkIjoidXVpZCJ9"
+        assert ProxyPatterns.is_vmess(url) is True
+
+    def test_vmess_extraction(self):
+        text = """
+        Some text
+        vmess://eyJhZGQiOiIxMjcuMC4wLjEiLCJwb3J0Ijo0NDN9
+        More text
+        """
+        matches = ProxyPatterns.extract_vmess_urls(text)
+
+        assert len(matches) == 1
+        assert matches[0].startswith("vmess://")
+
+
+class TestVLESSPatterns:
+    def test_vless_url_detection(self):
+        url = "vless://uuid@server:443?type=tcp"
+        assert ProxyPatterns.is_vless(url) is True
+
+    def test_vless_extraction(self):
+        text = "vless://some-uuid@192.168.1.1:443?encryption=none&type=tcp"
+        matches = ProxyPatterns.extract_vless_urls(text)
+
+        assert len(matches) == 1
+
+
+class TestTrojanPatterns:
+    def test_trojan_url_detection(self):
+        url = "trojan://password@server:443"
+        assert ProxyPatterns.is_trojan(url) is True
+
+    def test_trojan_extraction(self):
+        text = "trojan://pass123@example.com:443?sni=example.com"
+        matches = ProxyPatterns.extract_trojan_urls(text)
+
+        assert len(matches) == 1
+
+
+class TestShadowsocksPatterns:
+    def test_ss_url_detection(self):
+        url = "ss://base64encoded@server:8388"
+        assert ProxyPatterns.is_shadowsocks(url) is True
+
+    def test_ss_extraction(self):
+        text = "ss://YWVzLTI1Ni1nY206cGFzc3dvcmQ@192.168.1.1:8388"
+        matches = ProxyPatterns.extract_ss_urls(text)
+
+        assert len(matches) == 1