fixing test

googleapis · Nov 18, 2023 · 38725f0 · 38725f0
1 parent 9827fb9
commit 38725f0
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/google-cloud-datastore/src/test/java/com/google/cloud/datastore/DatastoreTest.java b/google-cloud-datastore/src/test/java/com/google/cloud/datastore/DatastoreTest.java
@@ -193,7 +193,7 @@ public void setUp() {
 
   @After
   public void tearDown() throws Exception {
-    // TODO(gapic_upgrade): Constant ping: temporarily addressing the connection refused error
+    // TODO(gapic_upgrade): Constant ping: temporarily addressing the flaky connection refused error
     helper.checkHealth();
   }
 

diff --git a/samples/snippets/src/test/java/com/google/datastore/snippets/ConceptsTest.java b/samples/snippets/src/test/java/com/google/datastore/snippets/ConceptsTest.java
@@ -137,6 +137,9 @@ public void tearDown() throws Exception {
     KeyQuery taskQuery = Query.newKeyQueryBuilder().setKind("Task").build();
     Key[] taskKeysToDelete = Iterators.toArray(datastoreRealBackend.run(taskQuery), Key.class);
     datastoreRealBackend.delete(taskKeysToDelete);
+
+    // TODO(gapic_upgrade): Constant ping: temporarily addressing the flaky connection refused error
+    HELPER.checkHealth();
   }
 
   /**