Set timeout conformance test

package.json

@@ -42,7 +42,7 @@
     "pretest": "npm run compile",
     "test": "c8 mocha build/test",
     "test:snap": "SNAPSHOT_UPDATE=1 npm test",
-    "testproxy": "npm run compile && node testproxy/index.js",
+    "testproxy": "npm run compile && GRPC_TRACE=all GRPC_VERBOSITY=DEBUG node testproxy/index.js",
     "clean": "gts clean",
     "precompile": "gts clean"
   },

src/index.ts

@@ -866,9 +866,18 @@
 
         gaxStream = requestFn!();
         gaxStream
-          .on('error', stream.destroy.bind(stream))
+          .on('error', (err: ServiceError) => {
+            console.log('getting error');
+            stream.destroy.bind(stream);
+          })
           .on('metadata', stream.emit.bind(stream, 'metadata'))
           .on('request', stream.emit.bind(stream, 'request'))
+          .on('end', () => {
+            console.log('gax stream ended');
+          })
+          .on('data', (err: ServiceError) => {
+            console.log('received data from gax stream');
+          })
           .pipe(stream);
       });
     }
@@ -882,6 +891,7 @@
         gaxStream = requestFn!();
         gaxStream
           .on('error', (err: Error) => {
+            console.log('getting error 2');
             stream.destroy(err);
           })
           .on('metadata', metadata => {

src/table.ts

@@ -810,7 +810,7 @@
         // Handling retries in this client. Specify the retry options to
         // make sure nothing is retried in retry-request.
         noResponseRetries: 0,
         shouldRetryFn: (_: any) => {
           return false;
         },
       };
@@ -969,7 +969,7 @@
             userStream.emit('error', error);
           }
         })
         .on('data', _ => {
           // Reset error count after a successful read so the backoff
           // time won't keep increasing when as stream had multiple errors
           numConsecutiveErrors = 0;
@@ -1575,6 +1575,7 @@
     };
 
     const makeNextBatchRequest = () => {
+      console.log('making mutate rows request');
       const entryBatch = entries.filter((entry: Entry, index: number) => {
         return pendingEntryIndices.has(index);
       });
@@ -1597,7 +1598,7 @@
         },
       };
 
-      options.gaxOptions = populateAttemptHeader(
+      const gaxOpts = populateAttemptHeader(
         numRequestsMade,
         options.gaxOptions
       );
@@ -1607,13 +1608,15 @@
           client: 'BigtableClient',
           method: 'mutateRows',
           reqOpts,
-          gaxOpts: options.gaxOptions,
+          gaxOpts,
           retryOpts,
         })
         .on('error', (err: ServiceError) => {
+          console.log('received an error');
           onBatchResponse(err);
         })
         .on('data', (obj: google.bigtable.v2.IMutateRowsResponse) => {
+          console.log('received data');
           obj.entries!.forEach(entry => {
             const originalEntry = entryBatch[entry.index as number];
             const originalEntriesIndex = entryToIndex.get(originalEntry)!;
@@ -1633,7 +1636,10 @@
             mutationErrorsByEntryIndex.set(originalEntriesIndex, errorDetails);
           });
         })
-        .on('end', onBatchResponse);
+        .on('end', () => {
+          console.log('closed the stream');
+          onBatchResponse(null);
+        });
       numRequestsMade++;
     };
 

testproxy/services/create-client.js

@@ -17,6 +17,7 @@ const normalizeCallback = require('./utils/normalize-callback.js');
 
 const grpc = require('@grpc/grpc-js');
 const {Bigtable} = require('../../build/src/index.js');
+const {PreciseDate} = require('@google-cloud/precise-date');
 const {BigtableClient} = require('../../build/src/index.js').v2;
 
 const v2 = Symbol.for('v2');
@@ -54,12 +55,19 @@ const createClient = ({clientMap}) =>
     if (callCredential && callCredential.jsonServiceAccount) {
       authClient = JSON.parse(request.callCredential.jsonServiceAccount);
     }
+    const clientConfig = require('../../src/v2/bigtable_client_config.json');
+    clientConfig.interfaces['google.bigtable.v2.Bigtable'].methods[
+      'MutateRows'
+    ].timeout_millis = parseInt(
+      parseInt(new PreciseDate(request.perOperationTimeout).getFullTime()) /
+        1000000
+    );
     const bigtable = new Bigtable({
       projectId,
       apiEndpoint,
       authClient,
       appProfileId,
-      clientConfig: require('../../src/v2/bigtable_client_config.json'),
+      clientConfig,
     });
     bigtable[v2] = new BigtableClient(bigtable.options.BigtableClient);
     clientMap.set(clientId, bigtable);