mirror of
https://gitee.com/milvus-io/milvus.git
synced 2025-12-29 06:55:27 +08:00
ea58edc128
* Add python tests about param timeout Signed-off-by: yangxuan <jumpthepig@gmail.com> * add faiss configure args msg print Signed-off-by: yangxuan <xuan.yang@zilliz.com> * change assertion conditions Signed-off-by: yangxuan <xuan.yang@zilliz.com> Co-authored-by: yangxuan <xuan.yang@zilliz.com>
142 lines
4.7 KiB
Python
142 lines
4.7 KiB
Python
import logging
|
|
import pytest
|
|
|
|
__version__ = '0.10.1'
|
|
|
|
|
|
class TestPing:
|
|
def test_server_version(self, connect):
|
|
'''
|
|
target: test get the server version
|
|
method: call the server_version method after connected
|
|
expected: version should be the milvus version
|
|
'''
|
|
status, res = connect.server_version()
|
|
assert res == __version__
|
|
|
|
def test_server_status(self, connect):
|
|
'''
|
|
target: test get the server status
|
|
method: call the server_status method after connected
|
|
expected: status returned should be ok
|
|
'''
|
|
status, msg = connect.server_status()
|
|
assert status.OK()
|
|
|
|
def test_server_cmd_with_params_version(self, connect):
|
|
'''
|
|
target: test cmd: version
|
|
method: cmd = "version" ...
|
|
expected: when cmd = 'version', return version of server;
|
|
'''
|
|
cmd = "version"
|
|
status, msg = connect._cmd(cmd)
|
|
logging.getLogger().info(status)
|
|
logging.getLogger().info(msg)
|
|
assert status.OK()
|
|
assert msg == __version__
|
|
|
|
def test_server_cmd_with_params_others(self, connect):
|
|
'''
|
|
target: test cmd: lalala
|
|
method: cmd = "lalala" ...
|
|
expected: when cmd = 'version', return version of server;
|
|
'''
|
|
cmd = "rm -rf test"
|
|
status, msg = connect._cmd(cmd)
|
|
logging.getLogger().info(status)
|
|
logging.getLogger().info(msg)
|
|
assert status.OK()
|
|
# assert msg == __version__
|
|
|
|
def test_connected(self, connect):
|
|
# assert connect.connected()
|
|
assert connect
|
|
|
|
|
|
class TestPingWithTimeout:
|
|
def test_server_version_legal_timeout(self, connect):
|
|
'''
|
|
target: test get the server version with legal timeout
|
|
method: call the server_version method after connected with altering timeout
|
|
expected: version should be the milvus version
|
|
'''
|
|
status, res = connect.server_version(20)
|
|
assert res == __version__
|
|
|
|
def test_server_version_negative_timeout(self, connect):
|
|
'''
|
|
target: test get the server version with negative timeout
|
|
method: call the server_version method after connected with altering timeout
|
|
expected: when timeout is illegal raises an error;
|
|
'''
|
|
status, res = connect.server_version(-1)
|
|
assert not status.OK()
|
|
|
|
def test_server_cmd_with_params_version_with_legal_timeout(self, connect):
|
|
'''
|
|
target: test cmd: version and timeout
|
|
method: cmd = "version" , timeout=10
|
|
expected: when cmd = 'version', return version of server;
|
|
'''
|
|
cmd = "version"
|
|
status, msg = connect._cmd(cmd, 10)
|
|
logging.getLogger().info(status)
|
|
logging.getLogger().info(msg)
|
|
assert status.OK()
|
|
assert msg == __version__
|
|
|
|
def test_server_cmd_with_params_version_with_illegal_timeout(self, connect):
|
|
'''
|
|
target: test cmd: version and timeout
|
|
method: cmd = "version" , timeout=-1
|
|
expected: when timeout is illegal raises an error;
|
|
'''
|
|
status, res = connect.server_version(-1)
|
|
assert not status.OK()
|
|
|
|
def test_server_cmd_with_params_others_with_illegal_timeout(self, connect):
|
|
'''
|
|
target: test cmd: lalala, timeout = -1
|
|
method: cmd = "lalala", timeout = -1
|
|
expected: when timeout is illegal raises an error;
|
|
'''
|
|
cmd = "rm -rf test"
|
|
status, res = connect.server_version(-1)
|
|
assert not status.OK()
|
|
|
|
|
|
class TestPingDisconnect:
|
|
def test_server_version(self, dis_connect):
|
|
'''
|
|
target: test get the server version, after disconnect
|
|
method: call the server_version method after connected
|
|
expected: version should not be the pymilvus version
|
|
'''
|
|
res = None
|
|
with pytest.raises(Exception) as e:
|
|
status, res = connect.server_version()
|
|
assert res is None
|
|
|
|
def test_server_status(self, dis_connect):
|
|
'''
|
|
target: test get the server status, after disconnect
|
|
method: call the server_status method after connected
|
|
expected: status returned should be not ok
|
|
'''
|
|
status = None
|
|
with pytest.raises(Exception) as e:
|
|
status, msg = connect.server_status()
|
|
assert status is None
|
|
|
|
def test_server_version_with_timeout(self, dis_connect):
|
|
'''
|
|
target: test get the server status with timeout settings after disconnect
|
|
method: call the server_status method after connected
|
|
expected: status returned should be not ok
|
|
'''
|
|
status = None
|
|
with pytest.raises(Exception) as e:
|
|
status, msg = connect.server_status(100)
|
|
assert status is None
|