stage1_testServiceAccountsList refactoring

rafaelMurata · May 9, 2024 · 336fa98 · 336fa98
1 parent 1daece5
commit 336fa98
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/iam/snippets/src/test/java/ServiceAccountTests.java b/iam/snippets/src/test/java/ServiceAccountTests.java
@@ -23,14 +23,14 @@
 
 import com.google.cloud.iam.admin.v1.IAMClient;
 import com.google.cloud.testing.junit4.MultipleAttemptsRule;
-import com.google.common.collect.Lists;
 import com.google.iam.admin.v1.ServiceAccount;
 import com.google.iam.admin.v1.ServiceAccountKey;
 import java.io.ByteArrayOutputStream;
 import java.io.IOException;
 import java.io.PrintStream;
 import java.util.List;
 import java.util.UUID;
+import java.util.concurrent.atomic.AtomicBoolean;
 import org.junit.After;
 import org.junit.Before;
 import org.junit.BeforeClass;
@@ -94,7 +94,14 @@ public void stage1_testServiceAccountsList() throws IOException {
     IAMClient.ListServiceAccountsPagedResponse response =
             ListServiceAccounts.listServiceAccounts(PROJECT_ID);
 
-    assertTrue(response.iterateAll().iterator().hasNext());
+    AtomicBoolean containsAccount = new AtomicBoolean(false);
+    response.iterateAll().forEach(account -> {
+      if (account.getName().contains(SERVICE_ACCOUNT)) {
+        containsAccount.set(true);
+      }
+    });
+
+    assertTrue(containsAccount.get());
   }
 
   @Test