tests(compute): fix failing test (#4549)

googleapis · Aug 3, 2021 · 6e79385 · 6e79385
1 parent e9ddbfe
commit 6e79385
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/compute/apiv1/smoke_test.go b/compute/apiv1/smoke_test.go
@@ -281,10 +281,11 @@ func TestPaginationWithMaxRes(t *testing.T) {
 	for err == nil {
 		if element.GetName() == "nvidia-tesla-t4" {
 			found = true
+			break
 		}
 		element, err = itr.Next()
 	}
-	if err != iterator.Done {
+	if err != nil && err != iterator.Done {
 		t.Fatal(err)
 	}
 	if !found {
@@ -316,7 +317,7 @@ func TestPaginationDefault(t *testing.T) {
 		}
 		element, err = itr.Next()
 	}
-	if err != iterator.Done {
+	if err != nil && err != iterator.Done {
 		t.Fatal(err)
 	}
 	if !found {