Fix unit test and added call_retry check for WebSocketConnectionClosedException

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()

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'])