tests: move utility function into the only test using it, and convert the veneer output to proto format

Author: feywind

testproxy/services/sample-row-keys.ts

@@ -14,10 +14,10 @@
 
 import * as grpc from '@grpc/grpc-js';
 import {GoogleError} from 'google-gax';
-import {getSRKRequest} from './utils/request/sampleRowKeys';
 import {ClientImplMaker, normalizeCallback} from './utils';
 
 import {google} from '../protos/protos';
+import {log} from './utils/log';
 type ISampleRowKeysRequest = google.bigtable.testproxy.ISampleRowKeysRequest;
 type ISampleRowKeysResult = google.bigtable.testproxy.ISampleRowKeysResult;
 
@@ -32,12 +32,20 @@ export const sampleRowKeys: ClientImplMaker<
 
     const bigtable = clientMap.get(clientId!);
 
+    const [, , , instanceId, , tableId] = tableName!.split('/');
+    const instance = bigtable.instance(instanceId);
+    const table = instance.table(tableId);
+
     try {
-      const response = await getSRKRequest(bigtable, {tableName});
+      const response = await table.sampleRowKeys();
 
+      log.info('sampleRowKeys response %o', response);
       return {
         status: {code: grpc.status.OK, details: []},
-        response,
+        samples: response[0].map(sample => ({
+          rowKey: sample.key,
+          offsetBytes: sample.offset,
+        })),
       };
     } catch (e) {
       const error = e as GoogleError;