test: Refactor teardown (#424)

* chore: disabling ITs to reduce cost & resolve quota issue

* revert getTrainingsample changes

* chore: refactored teardown part

* lint

* 🦉 Updates from OwlBot

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

* test: fixed teardown part

Co-authored-by: Owl Bot <gcf-owl-bot[bot]@users.noreply.github.com>

Author: munkhuushmgl

aiplatform/snippets/src/test/java/aiplatform/CreateHyperparameterTuningJobPythonPackageSampleTest.java

@@ -31,6 +31,7 @@
 import org.junit.After;
 import org.junit.Before;
 import org.junit.BeforeClass;
+import org.junit.Ignore;
 import org.junit.Test;
 
 public class CreateHyperparameterTuningJobPythonPackageSampleTest {
@@ -93,6 +94,7 @@ public void tearDown()
   }
 
   @Test
+  @Ignore("https://github.com/googleapis/java-aiplatform/issues/420")
   public void testCreateHyperparameterTuningJobPythonPackageSample() throws IOException {
     String hyperparameterTuningJobDisplayName =
         String.format(

aiplatform/snippets/src/test/java/aiplatform/CreateTrainingPipelineTabularClassificationSampleTest.java

@@ -19,6 +19,10 @@
 import static com.google.common.truth.Truth.assertThat;
 import static junit.framework.TestCase.assertNotNull;
 
+import com.google.cloud.aiplatform.v1.PipelineServiceClient;
+import com.google.cloud.aiplatform.v1.PipelineServiceSettings;
+import com.google.cloud.aiplatform.v1.TrainingPipeline;
+import com.google.cloud.aiplatform.v1.TrainingPipelineName;
 import java.io.ByteArrayOutputStream;
 import java.io.IOException;
 import java.io.PrintStream;
@@ -72,7 +76,25 @@ public void tearDown()
     // Assert
     String cancelResponse = bout.toString();
     assertThat(cancelResponse).contains("Cancelled the Training Pipeline");
-    TimeUnit.MINUTES.sleep(2);
+
+    PipelineServiceSettings pipelineServiceSettings =
+        PipelineServiceSettings.newBuilder()
+            .setEndpoint("us-central1-aiplatform.googleapis.com:443")
+            .build();
+
+    try (PipelineServiceClient pipelineServiceClient =
+        PipelineServiceClient.create(pipelineServiceSettings)) {
+      String location = "us-central1";
+      TrainingPipelineName trainingPipelineName =
+          TrainingPipelineName.of(PROJECT, location, trainingPipelineId);
+
+      TrainingPipeline trainingPipelineResponse =
+          pipelineServiceClient.getTrainingPipeline(trainingPipelineName);
+      while (!trainingPipelineResponse.getState().name().contains("STATE_CANCELLED")) {
+        TimeUnit.SECONDS.sleep(30);
+        trainingPipelineResponse = pipelineServiceClient.getTrainingPipeline(trainingPipelineName);
+      }
+    }
 
     // Delete the Training Pipeline
     DeleteTrainingPipelineSample.deleteTrainingPipelineSample(PROJECT, trainingPipelineId);