From 8c20927f68a7ce945d72dfc22e4b32a0ee2c93a6 Mon Sep 17 00:00:00 2001
From: Tamas Kocsis <tamas@zeronet.io>
Date: Tue, 8 Sep 2020 19:35:58 +0200
Subject: [PATCH] Allow test port checker functions from CLI

---
 plugins/Benchmark/BenchmarkPlugin.py | 31 ++++++++++++++++++++++++++++
 1 file changed, 31 insertions(+)

diff --git a/plugins/Benchmark/BenchmarkPlugin.py b/plugins/Benchmark/BenchmarkPlugin.py
index 39590622..b47da642 100644
--- a/plugins/Benchmark/BenchmarkPlugin.py
+++ b/plugins/Benchmark/BenchmarkPlugin.py
@@ -340,6 +340,32 @@ class ActionsPlugin:
             yield "."
             assert ok, "does not verify from %s" % address
 
+    def testPortCheckers(self):
+        """
+        Test all active open port checker
+        """
+        from Peer import PeerPortchecker
+        for ip_type, func_names in PeerPortchecker.PeerPortchecker.checker_functions.items():
+            yield "\n- %s:" % ip_type
+            for func_name in func_names:
+                yield "\n - Tracker %s: " % func_name
+                try:
+                    for res in self.testPortChecker(func_name):
+                        yield res
+                except Exception as err:
+                    yield Debug.formatException(err)
+
+    def testPortChecker(self, func_name):
+        """
+        Test single open port checker
+        """
+        from Peer import PeerPortchecker
+        peer_portchecker = PeerPortchecker.PeerPortchecker(None)
+        s = time.time()
+        announce_func = getattr(peer_portchecker, func_name)
+        res = announce_func(3894)
+        yield res
+
     def testAll(self):
         """
         Run all tests to check system compatibility with ZeroNet functions
@@ -361,4 +387,9 @@ class ConfigPlugin(object):
                 '--filter', help='Filter running benchmark',
                 default=None, metavar='test name'
             )
+        elif self.getCmdlineValue("test") == "portChecker":
+            self.test_parser.add_argument(
+                '--func_name', help='Name of open port checker function',
+                default=None, metavar='func_name'
+            )
         return back