[iotdb] branch test_pyt1 updated: update

[haonan]

This is an automated email from the ASF dual-hosted git repository.

haonan pushed a commit to branch test_pyt1
in repository https://gitbox.apache.org/repos/asf/iotdb.git


The following commit(s) were added to refs/heads/test_pyt1 by this push:
     new 481213b8dac update
481213b8dac is described below

commit 481213b8dace09a56daceb0b58daa9ef2f4432e0
Author: HTHou <hhao...@outlook.com>
AuthorDate: Tue Aug 1 11:31:32 2023 +0800

    update
---
 iotdb-client/client-py/iotdb/IoTDBContainer.py | 19 ++++++++++++++++---
 1 file changed, 16 insertions(+), 3 deletions(-)

diff --git a/iotdb-client/client-py/iotdb/IoTDBContainer.py 
b/iotdb-client/client-py/iotdb/IoTDBContainer.py
index 5072c5cdb30..70f49dfa406 100644
--- a/iotdb-client/client-py/iotdb/IoTDBContainer.py
+++ b/iotdb-client/client-py/iotdb/IoTDBContainer.py
@@ -19,8 +19,11 @@
 from os import environ
 
 from testcontainers.core.container import DockerContainer
+from testcontainers.core.exceptions import ContainerStartException
 from testcontainers.core.utils import setup_logger
-from testcontainers.core.waiting_utils import wait_container_is_ready, 
wait_for_logs
+from testcontainers.core.waiting_utils import wait_container_is_ready
+
+from iotdb.Session import Session
 
 logger = setup_logger(__name__)
 
@@ -33,8 +36,18 @@ class IoTDBContainer(DockerContainer):
         pass
 
     @wait_container_is_ready()
-    def __connect(self):
-        wait_for_logs(self, "Now, enjoy yourself!", 60)
+    def _connect(self):
+        session = Session(
+            self.get_container_host_ip(), self.get_exposed_port(6667), "root", 
"root"
+        )
+        session.open(False)
+        with session.execute_statement(
+                "SHOW CLUSTER"
+        ) as session_data_set:
+            while session_data_set.has_next():
+                if session_data_set.next().get_fields()[2].get_string_value() 
!= "Running":
+                    raise ContainerStartException("IoTDB is not started")
+        session.close()
 
     def __init__(self, image="apache/iotdb:latest", **kwargs):
         super(IoTDBContainer, self).__init__(image)