Loading tests/milvus_python_test/conftest.py +11 −5 Original line number Diff line number Diff line Loading @@ -42,16 +42,22 @@ def connect(request): port = request.config.getoption("--port") milvus = Milvus() try: milvus.connect(host=ip, port=port) except: status = milvus.connect(host=ip, port=port) logging.getLogger().info(status) if not status.OK(): # try again logging.getLogger().info("------------------------------------") logging.getLogger().info("Try to connect again") logging.getLogger().info("------------------------------------") res = milvus.connect(host=ip, port=port) except Exception as e: logging.getLogger().error(str(e)) pytest.exit("Milvus server can not connected, exit pytest ...") def fin(): try: milvus.disconnect() except: pass request.addfinalizer(fin) return milvus Loading tests/milvus_python_test/test_connect.py +2 −2 Original line number Diff line number Diff line Loading @@ -249,8 +249,8 @@ class TestConnect: ''' milvus = Milvus() uri_value = "tcp://%s:19540" % args["ip"] with pytest.raises(Exception) as e: milvus.connect(host=args["ip"], port="", uri=uri_value) assert milvus.connected() def test_connect_param_priority_uri(self, args): ''' Loading Loading
tests/milvus_python_test/conftest.py +11 −5 Original line number Diff line number Diff line Loading @@ -42,16 +42,22 @@ def connect(request): port = request.config.getoption("--port") milvus = Milvus() try: milvus.connect(host=ip, port=port) except: status = milvus.connect(host=ip, port=port) logging.getLogger().info(status) if not status.OK(): # try again logging.getLogger().info("------------------------------------") logging.getLogger().info("Try to connect again") logging.getLogger().info("------------------------------------") res = milvus.connect(host=ip, port=port) except Exception as e: logging.getLogger().error(str(e)) pytest.exit("Milvus server can not connected, exit pytest ...") def fin(): try: milvus.disconnect() except: pass request.addfinalizer(fin) return milvus Loading
tests/milvus_python_test/test_connect.py +2 −2 Original line number Diff line number Diff line Loading @@ -249,8 +249,8 @@ class TestConnect: ''' milvus = Milvus() uri_value = "tcp://%s:19540" % args["ip"] with pytest.raises(Exception) as e: milvus.connect(host=args["ip"], port="", uri=uri_value) assert milvus.connected() def test_connect_param_priority_uri(self, args): ''' Loading