@@ -24,6 +24,9 @@ import nuvolaris.etcd as etcd
24
24
import nuvolaris.milvus_standalone as milvus
25
25
import nuvolaris.minio_deploy as minio
26
26
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
27
30
28
31
# tu.enable_debug_logging()
29
32
# test
@@ -37,6 +40,50 @@ os.environ['MILVUS_API_HOST']='localhost'
37
40
assert(etcd.create())
38
41
assert(minio.create())
39
42
assert(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
+
40
87
assert(milvus.delete())
41
88
assert(etcd.delete())
42
89
assert(minio.delete())
0 commit comments