Add: protonvpn serverip plugin

3 files changed, 25 insertions, 3 deletions

diff --git a/test/__pycache__/test_vpn_detector.cpython-39.pyc b/test/__pycache__/test_vpn_detector.cpython-39.pyc
index 65bdddc..152359b 100644
--- a/test/__pycache__/test_vpn_detector.cpython-39.pyc
+++ b/test/__pycache__/test_vpn_detector.cpython-39.pyc
diff --git a/test/test_plugins.py b/test/test_plugins.py
index 835491a..88358b0 100644
--- a/test/test_plugins.py
+++ b/test/test_plugins.py
@@ -18,6 +18,7 @@ from ivacyvpn_serverip import IvacyvpnServerip
 from ivacyvpn_servername import IvacyvpnServername
 
 from detection.vpnservices.turbovpn_serverip import TurbovpnServerip
+from protonvpn_serverip import ProtonvpnServerip
 from vpnunlimited_serverip import VpnunlimitedServerip
 from windscribevpn_servername import WindscribevpnServername
 from windscribevpn_serverip import WindscribevpnServerip
@@ -251,6 +252,27 @@ class TestWindscribevpnServerip(unittest.TestCase):
                                                         confidence=self.windscribevpn_detector.confidence)
 
 
+class TestProtonvpnServerip(unittest.TestCase):
+
+        def setUp(self):
+            self.protonvpn_detector = ProtonvpnServerip(start_time='2023-10-16 00:00:00',
+                                                    end_time='2023-10-30 00:00:00')
+
+        def test_get_protonvpn_serverip(self):
+            self.result = self.protonvpn_detector.find_server()
+            print(self.result)
+
+        def test_save_to_knowledgebase(self):
+            self.protonvpn_detector.save_to_knowledgebase(object_type='ip',
+                                                        object_list=self.protonvpn_detector.find_server(),
+                                                        vpn_service_name=self.protonvpn_detector.vpn_service_name,
+                                                        plugin_id=self.protonvpn_detector.plugin_id,
+                                                        plugin_name=self.protonvpn_detector.plugin_name,
+                                                        output_filename=self.protonvpn_detector.output_file_name,
+                                                        confidence=self.protonvpn_detector.confidence)
+
+
+
 if __name__ == '__main__':
     # # 运行指定测试用例
     suite = unittest.TestSuite()
diff --git a/test/test_vpn_detector.py b/test/test_vpn_detector.py
index 1b9578c..c8a6a50 100644
--- a/test/test_vpn_detector.py
+++ b/test/test_vpn_detector.py
@@ -29,11 +29,11 @@ class TestVpnDetector(unittest.TestCase):
 
     def test_file_to_knowledgebase(self):
         object_list = []
-        with open('data/vpnunlimited_serverip/vpnunlimited_serverip_builtin.txt', 'r') as f:
+        with open('data/psiphon3_serverip/as12876_psiphon_builtin.txt', 'r') as f:
             for line in f.readlines():
                 object_list.append(line.strip())
 
-        self.vpn_detector.save_to_knowledgebase(object_type='ip', object_list=object_list, vpn_service_name='vpnunlimited',
-                                                plugin_name='vpnunlimited_serverip', plugin_id=8, output_filename='vpnunlimited_serverip_builtin.csv', confidence='confirmed')
+        self.vpn_detector.save_to_knowledgebase(object_type='ip', object_list=object_list, vpn_service_name='psiphon3vpn',
+                                                plugin_name='psiphon3vpn_serverip', plugin_id=4, output_filename='psiphon3vpn_serverip_builtin.csv', confidence='confirmed')