diff options
Diffstat (limited to 'test/test_plugins.py')
| -rw-r--r-- | test/test_plugins.py | 22 |
1 files changed, 22 insertions, 0 deletions
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() |