From 6e79385b1436cc0bfc4adb9eefe980878b89afcc Mon Sep 17 00:00:00 2001
From: georgiyekkert <gekkert@griddynamics.com>
Date: Tue, 3 Aug 2021 16:25:50 -0700
Subject: [PATCH] tests(compute): fix failing test (#4549)

---
 compute/apiv1/smoke_test.go | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

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 {