diff --git a/compute/apiv1/smoke_test.go b/compute/apiv1/smoke_test.go index a61e1a72246..b83157da314 100644 --- 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 {