From ea219576df902b261b699191a83b7952be132840 Mon Sep 17 00:00:00 2001
From: Bernd Verst <4535280+berndverst@users.noreply.github.com>
Date: Fri, 23 Sep 2022 13:42:45 -0700
Subject: [PATCH] Apply suggestions from code review

Signed-off-by: Bernd Verst <4535280+berndverst@users.noreply.github.com>
---
 state/azure/cosmosdb/cosmosdb_query.go | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/state/azure/cosmosdb/cosmosdb_query.go b/state/azure/cosmosdb/cosmosdb_query.go
index 7d6ed61d7..7506f0659 100644
--- a/state/azure/cosmosdb/cosmosdb_query.go
+++ b/state/azure/cosmosdb/cosmosdb_query.go
@@ -174,12 +174,12 @@ func (q *Query) execute(client *azcosmos.ContainerClient) ([]state.QueryItem, st
 			if err != nil {
 				return nil, "", err
 			}
-			if len(items) >= resultLimit {
+			if (resultLimit != 0) && (len(items) >= resultLimit) {
 				break
 			}
 			items = append(items, tempItem)
 		}
-		if len(items) >= resultLimit {
+		if (resultLimit != 0) && (len(items) >= resultLimit) {
 			break
 		}
 	}