fix test failure

Author: reuvenlax

sdks/java/io/google-cloud-platform/src/main/java/org/apache/beam/sdk/io/gcp/bigquery/BigQueryIOTranslation.java

@@ -438,6 +438,7 @@ static class BigQueryIOWriteTranslator implements TransformPayloadTranslator<Wri
             .addNullableByteArrayField("row_mutation_information_fn")
             .addNullableByteArrayField("bad_record_error_handler")
             .addNullableByteArrayField("bad_record_router")
+            .addNullableInt64Field("auto_schema_update_strict_timeout_ms")
             .build();
 
     public static final String BIGQUERY_WRITE_TRANSFORM_URN =
@@ -582,6 +583,9 @@ public Row toConfigRow(Write<?> transform) {
       fieldValues.put("use_beam_schema", transform.getUseBeamSchema());
       fieldValues.put("auto_sharding", transform.getAutoSharding());
       fieldValues.put("auto_schema_update", transform.getAutoSchemaUpdate());
+      fieldValues.put(
+          "auto_schema_update_strict_timeout_ms",
+          transform.getAutoSchemaUpdateStrictTimeout().getMillis());
       if (transform.getWriteProtosClass() != null) {
         fieldValues.put("write_protos_class", toByteArray(transform.getWriteProtosClass()));
       }
@@ -870,6 +874,15 @@ public Write<?> fromConfigRow(Row configRow, PipelineOptions options) {
         if (autoSchemaUpdate != null) {
           builder = builder.setAutoSchemaUpdate(autoSchemaUpdate);
         }
+        Long autoSchemaUpdateStrictTimeoutMillis =
+            configRow.getInt64("auto_schema_update_strict_timeout_ms");
+        if (autoSchemaUpdateStrictTimeoutMillis != null) {
+          builder =
+              builder
+                  .setAutoSchemaUpdate(true)
+                  .setAutoSchemaUpdateStrictTimeout(
+                      org.joda.time.Duration.millis(autoSchemaUpdateStrictTimeoutMillis));
+        }
         byte[] writeProtosClasses = configRow.getBytes("write_protos_class");
         if (writeProtosClasses != null) {
           builder =

sdks/java/io/google-cloud-platform/src/test/java/org/apache/beam/sdk/io/gcp/bigquery/BigQueryIOTranslationTest.java

@@ -132,6 +132,8 @@ public class BigQueryIOTranslationTest {
     WRITE_TRANSFORM_SCHEMA_MAPPING.put("getAutoSharding", "auto_sharding");
     WRITE_TRANSFORM_SCHEMA_MAPPING.put("getPropagateSuccessful", "propagate_successful");
     WRITE_TRANSFORM_SCHEMA_MAPPING.put("getAutoSchemaUpdate", "auto_schema_update");
+    WRITE_TRANSFORM_SCHEMA_MAPPING.put(
+        "getAutoSchemaUpdateStrictTimeout", "auto_schema_update_strict_timeout_ms");
     WRITE_TRANSFORM_SCHEMA_MAPPING.put("getWriteProtosClass", "write_protos_class");
     WRITE_TRANSFORM_SCHEMA_MAPPING.put("getDirectWriteProtos", "direct_write_protos");
     WRITE_TRANSFORM_SCHEMA_MAPPING.put("getDeterministicRecordIdFn", "deterministic_record_id_fn");