Skip to content

Commit 148b9be

Browse files
committed
fix(milvus): milvus testing
extended tests for milvus_standalone.py
1 parent 25a187f commit 148b9be

File tree

1 file changed

+47
-0
lines changed

1 file changed

+47
-0
lines changed

tests/kind/milvus_standalone_test.ipy

Lines changed: 47 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -24,6 +24,9 @@ import nuvolaris.etcd as etcd
2424
import nuvolaris.milvus_standalone as milvus
2525
import nuvolaris.minio_deploy as minio
2626
import nuvolaris.testutil as tu
27+
from nuvolaris.milvus_simple_client import MilvusSimpleClient as MilvusClient
28+
from nuvolaris.user_config import UserConfig
29+
from nuvolaris.user_metadata import UserMetadata
2730

2831
# tu.enable_debug_logging()
2932
# test
@@ -37,6 +40,50 @@ os.environ['MILVUS_API_HOST']='localhost'
3740
assert(etcd.create())
3841
assert(minio.create())
3942
assert(milvus.create())
43+
44+
# generate some user config
45+
ucfg = UserConfig({
46+
"namespace": "demouser",
47+
"password": "s0meComplicat3d",
48+
"email": "[email protected]",
49+
"milvus": {
50+
"password": "s0meComplicat3d",
51+
"database": "demouser"
52+
}
53+
})
54+
# generate some user metadata
55+
metadata = UserMetadata(ucfg)
56+
57+
# test user creation
58+
assert(milvus.create_ow_milvus(ucfg, metadata))
59+
60+
# load user metadata
61+
user_metadata = metadata.get_metadata()
62+
body = {}
63+
envs = list(user_metadata['metadata'])
64+
65+
for env in envs:
66+
body[env['key']]=env['value']
67+
68+
# test user metadata
69+
host = body["MILVUS_HOST"]
70+
token = body["MILVUS_TOKEN"]
71+
database = body["MILVUS_DB_NAME"]
72+
port = body["MILVUS_PORT"]
73+
assert host
74+
assert token
75+
assert database
76+
assert port
77+
78+
# test client
79+
client = MilvusClient(uri=f'http://localhost:{port}',token=token, db_name=database)
80+
client.create_collection("testcollection", dimension=128)
81+
assert("testcollection" in client.list_collections())
82+
client.drop_collection("testcollection")
83+
84+
# test user deletion
85+
assert(milvus.delete_ow_milvus(ucfg))
86+
4087
assert(milvus.delete())
4188
assert(etcd.delete())
4289
assert(minio.delete())

0 commit comments

Comments
 (0)