From 3f20e0402e5cf070c10eb4c95318eed8617b6d3c Mon Sep 17 00:00:00 2001 From: Kenichi Maehashi Date: Thu, 22 Feb 2018 13:11:14 +0900 Subject: [PATCH 1/2] implement is_open API to WebSocketClient (fix #1) --- pysensorbee/api.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/pysensorbee/api.py b/pysensorbee/api.py index d6cc667..3e35e51 100644 --- a/pysensorbee/api.py +++ b/pysensorbee/api.py @@ -225,6 +225,12 @@ def close(self, **kwargs): """ self._app.close(**kwargs) + def is_open(self): + """ + Returns if the connection is established. + """ + return self._open + def get_error(self): """ Returns the last error occurred. From 3c12e1911f839d58bae03dd0b0ea54f287b59653 Mon Sep 17 00:00:00 2001 From: Kenichi Maehashi Date: Thu, 22 Feb 2018 13:34:21 +0900 Subject: [PATCH 2/2] add test --- pysensorbee/test/test_api.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/pysensorbee/test/test_api.py b/pysensorbee/test/test_api.py index e3b9afc..4ece58f 100644 --- a/pysensorbee/test/test_api.py +++ b/pysensorbee/test/test_api.py @@ -115,6 +115,9 @@ def test_wsquery(self): wsc = api.wsquery(self.TOPOLOGY) wsc.start() try: + # Ensure that the connection is open. + self.assertTrue(wsc.is_open()) + # One-shot wsc._test_oneshot = False def callback_1(wsc2, rid, type, payload):