From a5960a4b4694d741715705ccee2d281a9b6f4039 Mon Sep 17 00:00:00 2001
From: Karl Weinmeister <11586922+kweinmeister@users.noreply.github.com>
Date: Wed, 22 Dec 2021 15:54:41 -0600
Subject: [PATCH] test: Add null check in teardown (#752)
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

* test: Add null check in teardown

* test: Remove extra exception type

* 🦉 Updates from OwlBot

See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md

* chore: Updated shared-configuration in pom.xml

* chore: revert changes to auto-generated pom.xml files

* chore: update changes to auto-generated pom.xml files

Co-authored-by: Owl Bot <gcf-owl-bot[bot]@users.noreply.github.com>
---
 ...portDataVideoObjectTrackingSampleTest.java | 22 ++++++++++---------
 1 file changed, 12 insertions(+), 10 deletions(-)

diff --git a/aiplatform/snippets/src/test/java/aiplatform/ImportDataVideoObjectTrackingSampleTest.java b/aiplatform/snippets/src/test/java/aiplatform/ImportDataVideoObjectTrackingSampleTest.java
index 6d8b5e7a7ed..4052c91941c 100644
--- a/aiplatform/snippets/src/test/java/aiplatform/ImportDataVideoObjectTrackingSampleTest.java
+++ b/aiplatform/snippets/src/test/java/aiplatform/ImportDataVideoObjectTrackingSampleTest.java
@@ -98,17 +98,19 @@ public void setUp()
   @After
   public void tearDown() throws InterruptedException, ExecutionException, IOException {
     // delete the temp dataset
-    DatasetServiceSettings datasetServiceSettings =
-        DatasetServiceSettings.newBuilder()
-            .setEndpoint("us-central1-aiplatform.googleapis.com:443")
-            .build();
-    try (DatasetServiceClient datasetServiceClient =
-        DatasetServiceClient.create(datasetServiceSettings)) {
-      DatasetName datasetName = DatasetName.of(PROJECT, LOCATION, datasetId);
+    if (datasetId != null) {
+      DatasetServiceSettings datasetServiceSettings =
+          DatasetServiceSettings.newBuilder()
+              .setEndpoint("us-central1-aiplatform.googleapis.com:443")
+              .build();
+      try (DatasetServiceClient datasetServiceClient =
+          DatasetServiceClient.create(datasetServiceSettings)) {
+        DatasetName datasetName = DatasetName.of(PROJECT, LOCATION, datasetId);
 
-      OperationFuture<Empty, DeleteOperationMetadata> operationFuture =
-          datasetServiceClient.deleteDatasetAsync(datasetName);
-      operationFuture.get();
+        OperationFuture<Empty, DeleteOperationMetadata> operationFuture =
+            datasetServiceClient.deleteDatasetAsync(datasetName);
+        operationFuture.get();
+      }
     }
     System.out.flush();
     System.setOut(originalPrintStream);