@@ -24,6 +24,9 @@ import nuvolaris.etcd as etcd
2424import nuvolaris.milvus_standalone as milvus
2525import nuvolaris.minio_deploy as minio
2626import 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'
3740assert(etcd.create())
3841assert(minio.create())
3942assert(milvus.create())
43+
44+ # generate some user config
45+ ucfg = UserConfig({
46+ "namespace": "demouser",
47+ "password": "s0meComplicat3d",
48+ 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+
4087assert(milvus.delete())
4188assert(etcd.delete())
4289assert(minio.delete())
0 commit comments