update tests/unit/test_parsers.py

tests/unit/test_parsers.py

@@ -0,0 +1,73 @@
+import pytest
+from app.grabber.parsers import VMessParser, VLESSParser, TrojanParser, SSParser
+from app.models.proxy import Proxy
+
+
+class TestVMessParser:
+    def test_parse_valid_vmess(self):
+        vmess_json = '{"add": "127.0.0.1", "port": 443, "id": "test-uuid", "aid": 0}'
+        import base64
+
+        encoded = base64.b64encode(vmess_json.encode()).decode()
+        url = f"vmess://{encoded}"
+
+        proxy = VMessParser.parse(url)
+
+        assert proxy is not None
+        assert proxy.ip == "127.0.0.1"
+        assert proxy.port == 443
+        assert proxy.protocol == "vmess"
+
+    def test_parse_invalid_vmess(self):
+        url = "vmess://invalid-base64!@#"
+
+        with pytest.raises(ValueError):
+            VMessParser.parse(url)
+
+
+class TestVLESSParser:
+    def test_parse_valid_vless(self):
+        url = "vless://uuid-here@192.168.1.1:443?type=tcp&encryption=none"
+
+        proxy = VLESSParser.parse(url)
+
+        assert proxy is not None
+        assert proxy.ip == "192.168.1.1"
+        assert proxy.port == 443
+        assert proxy.protocol == "vless"
+
+    def test_parse_vless_with_domain(self):
+        url = "vless://uuid@example.com:443?type=tcp"
+
+        proxy = VLESSParser.parse(url)
+
+        assert proxy is not None
+        assert "example.com" in proxy.ip or proxy.ip != ""
+
+
+class TestTrojanParser:
+    def test_parse_valid_trojan(self):
+        url = "trojan://password123@10.0.0.1:443?sni=example.com"
+
+        proxy = TrojanParser.parse(url)
+
+        assert proxy is not None
+        assert proxy.ip == "10.0.0.1"
+        assert proxy.port == 443
+        assert proxy.protocol == "trojan"
+
+
+class TestSSParser:
+    def test_parse_valid_ss(self):
+        method_pass = "aes-256-gcm:password123"
+        import base64
+
+        encoded = base64.b64encode(method_pass.encode()).decode()
+        url = f"ss://{encoded}@192.168.1.1:8388"
+
+        proxy = SSParser.parse(url)
+
+        assert proxy is not None
+        assert proxy.ip == "192.168.1.1"
+        assert proxy.port == 8388
+        assert proxy.protocol == "shadowsocks"