update grpc_handler to match proto updates

jackdingilian · jackdingilian · commit 21d7771d439b · 2025-05-28T11:20:22.000-04:00
diff --git a/test_proxy/handlers/grpc_handler.py b/test_proxy/handlers/grpc_handler.py
@@ -78,7 +78,7 @@ def ReadRows(self, request, context, client_response=None):
             status = Status(code=5, message=client_response["error"])
         else:
             rows = [data_pb2.Row(**d) for d in client_response]
-        result = test_proxy_pb2.RowsResult(row=rows, status=status)
+        result = test_proxy_pb2.RowsResult(rows=rows, status=status)
         return result
 
     @delegate_to_client_handler
@@ -122,8 +122,7 @@ def BulkMutateRows(self, request, context, client_response=None):
                     code=client_response.get("code", 5),
                     message=client_response["error"],
                 )
-        # TODO: protos were updated. entry is now entries: https://github.com/googleapis/cndb-client-testing-protos/commit/e6205a2bba04acc10d12421a1402870b4a525fb3
-        response = test_proxy_pb2.MutateRowsResult(status=status, entry=entries)
+        response = test_proxy_pb2.MutateRowsResult(status=status, entries=entries)
         return response
 
     @delegate_to_client_handler
@@ -172,8 +171,7 @@ def SampleRowKeys(self, request, context, client_response=None):
                         offset_bytes=sample[1], row_key=sample[0]
                     )
                 )
-        # TODO: protos were updated. sample is now samples: https://github.com/googleapis/cndb-client-testing-protos/commit/e6205a2bba04acc10d12421a1402870b4a525fb3
-        return test_proxy_pb2.SampleRowKeysResult(status=status, sample=sample_list)
+        return test_proxy_pb2.SampleRowKeysResult(status=status, samples=sample_list)
 
     @delegate_to_client_handler
     def ExecuteQuery(self, request, context, client_response=None):
