fix: update test code

Signed-off-by: hlts2 <[email protected]>
vdaas · Jul 25, 2024 · c662359 · c662359
1 parent e80a93f
commit c662359
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/Makefile b/Makefile
@@ -49,6 +49,7 @@ PROTOS = \
 	v1/vald/filter.proto \
 	v1/vald/insert.proto \
 	v1/vald/object.proto \
+	v1/vald/index.proto \
 	v1/vald/remove.proto \
 	v1/vald/search.proto \
 	v1/vald/update.proto \

diff --git a/src/tests/test_e2e.py b/src/tests/test_e2e.py
@@ -10,6 +10,7 @@
 from vald.v1.vald import upsert_pb2_grpc
 from vald.v1.vald import remove_pb2_grpc
 from vald.v1.vald import object_pb2_grpc
+from vald.v1.vald import index_pb2_grpc
 from vald.v1.payload import payload_pb2
 
 
@@ -79,7 +80,7 @@ def test_save_index(self):
         self.assertIsInstance(result, payload_pb2.Empty)
 
     def test_index_info(self):
-        stub = agent_pb2_grpc.AgentStub(self.channel)
+        stub = index_pb2_grpc.IndexStub(self.channel)
         result = stub.IndexInfo(payload_pb2.Empty())
         self.assertIsInstance(result, payload_pb2.Info.Index.Count)
         self.assertEqual(result.stored, 99)