From 64847d1e5137e811e7f462fc058908fa9fcc55fb Mon Sep 17 00:00:00 2001
From: Irvi Aini <7439590+irvifa@users.noreply.github.com>
Date: Fri, 29 May 2020 01:19:15 +0700
Subject: [PATCH] docs(samples): Fix flaky test (#397)

We found that UpdateTableExpirationIT is flaky since during table creation the table is not created.
---
 .../example/bigquery/UpdateTableExpirationIT.java   | 13 +++++++++++--
 1 file changed, 11 insertions(+), 2 deletions(-)

diff --git a/samples/snippets/src/test/java/com/example/bigquery/UpdateTableExpirationIT.java b/samples/snippets/src/test/java/com/example/bigquery/UpdateTableExpirationIT.java
index 428317029..73ca93c06 100644
--- a/samples/snippets/src/test/java/com/example/bigquery/UpdateTableExpirationIT.java
+++ b/samples/snippets/src/test/java/com/example/bigquery/UpdateTableExpirationIT.java
@@ -19,8 +19,12 @@
 import static com.google.common.truth.Truth.assertThat;
 import static junit.framework.TestCase.assertNotNull;
 
+import com.google.cloud.bigquery.Field;
+import com.google.cloud.bigquery.Schema;
+import com.google.cloud.bigquery.StandardSQLTypeName;
 import java.io.ByteArrayOutputStream;
 import java.io.PrintStream;
+import java.util.UUID;
 import java.util.concurrent.TimeUnit;
 import org.junit.After;
 import org.junit.Before;
@@ -58,8 +62,13 @@ public void tearDown() {
 
   @Test
   public void updateTableExpiration() {
-    String tableName = "update_expiration_table";
-    CreateTable.createTable(BIGQUERY_DATASET_NAME, tableName, null);
+    String suffix = UUID.randomUUID().toString().replace('-', '_');
+    String tableName = "update_expiration_table_" + suffix;
+    Schema schema =
+            Schema.of(
+                    Field.of("stringField", StandardSQLTypeName.STRING),
+                    Field.of("booleanField", StandardSQLTypeName.BOOL));
+    CreateTable.createTable(BIGQUERY_DATASET_NAME, tableName, schema);
     Long newExpiration = TimeUnit.MILLISECONDS.convert(1, TimeUnit.DAYS);
     UpdateTableExpiration.updateTableExpiration(BIGQUERY_DATASET_NAME, tableName, newExpiration);
     assertThat(bout.toString()).contains("Table expiration updated successfully");