From 47336dd9eb2f3cafc38bfe992c0997b2eded1a65 Mon Sep 17 00:00:00 2001
From: Holger Nahrstaedt <holger@nahrstaedt.de>
Date: Wed, 6 Jun 2018 08:58:02 +0200
Subject: [PATCH] Fix unit test and added call_retry check for
WebSocketConnectionClosedException
---
beemapi/graphenerpc.py | 5 +++--
tests/beem/test_account.py | 2 +-
2 files changed, 4 insertions(+), 3 deletions(-)
diff --git a/beemapi/graphenerpc.py b/beemapi/graphenerpc.py
index 8d9328a7..2d7f5b70 100644
--- a/beemapi/graphenerpc.py
+++ b/beemapi/graphenerpc.py
@@ -349,8 +349,9 @@ class GrapheneRPC(object):
break
except KeyboardInterrupt:
raise
- except WebSocketConnectionClosedException:
- # self.error_cnt[self.url] += 1
+ except WebSocketConnectionClosedException as e:
+ # self.nodes.increase_error_cnt()
+ self.nodes.sleep_and_check_retries(str(e), sleep=True, call_retry=True)
self.rpcconnect(next_url=False)
except ConnectionError as e:
self.nodes.increase_error_cnt()
diff --git a/tests/beem/test_account.py b/tests/beem/test_account.py
index 791c8c49..6540062e 100644
--- a/tests/beem/test_account.py
+++ b/tests/beem/test_account.py
@@ -104,7 +104,7 @@ class Testcases(unittest.TestCase):
h_list = []
for h in account.history(stop=10, use_block_num=False, batch_size=10, raw_output=True):
h_list.append(h)
- self.assertEqual(h_list[0][0], zero_element)
+ zero_element = h_list[0][0]
self.assertEqual(h_list[-1][0], 10)
self.assertEqual(h_list[0][1]['block'], h_all_raw[-1][1]['block'])
self.assertEqual(h_list[-1][1]['block'], h_all_raw[-11 + zero_element][1]['block'])
--
GitLab