Added unittests for pro module

Kotsias, Panagiotis-Christos · Kotsias, Panagiotis-Christos · commit 2dfab55cac91 · 2020-10-28T10:04:00.000Z
diff --git a/test/test_pro_modules.py b/test/test_pro_modules.py
@@ -0,0 +1,47 @@
+import json
+from datetime import datetime
+import os
+from unittest import TestCase
+
+from etherscan.etherscan import Etherscan
+
+CONFIG_PATH = "configs/stable.json"
+API_PRO_KEY = os.environ["API_PRO_KEY"]  # Encrypted env var by Travis
+
+
+def load(fname):
+    with open(fname, "r") as f:
+        return json.load(f)
+
+
+def dump(data, fname):
+    with open(fname, "w") as f:
+        json.dump(data, f, indent=2)
+
+
+class Case(TestCase):
+    _MODULE = ""
+
+    def test_methods(self):
+        print(f"\nMODULE: {self._MODULE}")
+        config = load(CONFIG_PATH)
+        etherscan = Etherscan.from_config(CONFIG_PATH, API_PRO_KEY)
+        for fun, v in config.items():
+            if not fun.startswith("_"):  # disabled if _
+                if v["module"] == self._MODULE:
+                    res = getattr(etherscan, fun)(**v["kwargs"])
+                    print(f"METHOD: {fun}, RTYPE: {type(res)}")
+                    # Create log files (will update existing ones)
+                    fname = f"logs/{fun}.json"
+                    log = {
+                        "method": fun,
+                        "module": v["module"],
+                        "kwargs": v["kwargs"],
+                        "log_timestamp": datetime.now().strftime("%Y-%m-%d-%H:%M:%S"),
+                        "res": res,
+                    }
+                    dump(log, fname)
+
+
+class TestProModules(Case):
+    _MODULE = "pro"
