Adds node_type check and renames node-specific class

dmuhs · Oct 1, 2020 · 49487ff · 49487ff
1 parent 9d6ad1b
commit 49487ff
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/teatime/plugins/eth1/account_import.py b/teatime/plugins/eth1/account_import.py
@@ -3,7 +3,7 @@
 from teatime.reporting import Issue, Severity
 
 
-class AccountImport(Plugin):
+class GethAccountImport(Plugin):
     """Detect whether it's possible to import an account on the node.
 
     Severity: Medium

diff --git a/teatime/plugins/eth1/rpc.py b/teatime/plugins/eth1/rpc.py
@@ -1,5 +1,5 @@
 """This module contains plugins for controlling the HTTP RPC server status."""
-from teatime.plugins import Context, Plugin
+from teatime.plugins import Context, NodeType, Plugin
 from teatime.reporting import Issue, Severity
 
 
@@ -15,6 +15,9 @@ class GethStartRPC(Plugin):
     INTRUSIVE = True
 
     def _check(self, context: Context) -> None:
+        if context.node_type != NodeType.GETH:
+            return
+
         payload = self.get_rpc_json(context.target, method="admin_startRPC")
         context.report.add_issue(
             Issue(
@@ -40,6 +43,9 @@ class GethStopRPC(Plugin):
     INTRUSIVE = True
 
     def _check(self, context: Context) -> None:
+        if context.node_type != NodeType.GETH:
+            return
+
         payload = self.get_rpc_json(context.target, method="admin_stopRPC")
         context.report.add_issue(
             Issue(