Skip to content

Commit ff9a9de

Browse files
committed
Update unit test with grpc.Future interface
1 parent 4312f8a commit ff9a9de

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

tests/unit_grpc/test_grpc_index_upsert.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -13,9 +13,10 @@
1313
SparseValues,
1414
)
1515
from pinecone.grpc.utils import dict_to_proto_struct
16+
from grpc import Future as GrpcFuture
1617

1718

18-
class MockUpsertDelegate:
19+
class MockUpsertDelegate(GrpcFuture):
1920
def __init__(self, upsert_response: UpsertResponse):
2021
self.response = upsert_response
2122

@@ -40,6 +41,9 @@ def running(self):
4041
def add_done_callback(self, callback):
4142
pass
4243

44+
def traceback(self, timeout=None):
45+
pass
46+
4347

4448
@pytest.fixture
4549
def expected_vec1(vals1):

0 commit comments

Comments
 (0)