From bb25ec29642873af8d2c8bad2622a7746afecce6 Mon Sep 17 00:00:00 2001
From: Dmitrii Gridnev <dgridnev@qase.io>
Date: Mon, 10 Jun 2024 11:58:13 +0200
Subject: [PATCH] create: fix creation test run with environment

Implemented pagination in `get environment` method
---
 internal/client/clientv1.go | 49 ++++++++++++++++++++++---------------
 1 file changed, 29 insertions(+), 20 deletions(-)

diff --git a/internal/client/clientv1.go b/internal/client/clientv1.go
index 7e51c48..c288e0e 100644
--- a/internal/client/clientv1.go
+++ b/internal/client/clientv1.go
@@ -11,7 +11,7 @@ import (
 )
 
 const (
-	defaultLimit = 20
+	defaultLimit = 50
 )
 
 // ClientV1 is a client for Qase API v1
@@ -36,23 +36,34 @@ func (c *ClientV1) GetEnvironments(ctx context.Context, projectCode string) ([]r
 
 	ctx, client := c.getApiV1Client(ctx)
 
-	resp, r, err := client.EnvironmentsAPI.
-		GetEnvironments(ctx, projectCode).
-		Limit(defaultLimit).
-		Execute()
+	environments := make([]run.Environment, 0)
 
-	if err != nil {
-		logger.Debug("failed to get environments", "response", r)
-		return nil, fmt.Errorf("failed to get environments: %w", err)
-	}
+	var offset int32 = 0
+	for {
+		resp, r, err := client.EnvironmentsAPI.
+			GetEnvironments(ctx, projectCode).
+			Limit(defaultLimit).
+			Offset(offset).
+			Execute()
 
-	environments := make([]run.Environment, 0, len(resp.Result.Entities))
-	for _, env := range resp.Result.Entities {
-		environments = append(environments, run.Environment{
-			Title: env.GetTitle(),
-			ID:    env.GetId(),
-			Slug:  env.GetSlug(),
-		})
+		if err != nil {
+			logger.Debug("failed to get environments", "response", r)
+			return nil, fmt.Errorf("failed to get environments: %w", err)
+		}
+
+		for _, env := range resp.Result.Entities {
+			environments = append(environments, run.Environment{
+				Title: env.GetTitle(),
+				ID:    env.GetId(),
+				Slug:  env.GetSlug(),
+			})
+		}
+
+		if resp.Result.GetTotal() <= offset {
+			break
+		} else {
+			offset += defaultLimit
+		}
 	}
 
 	logger.Debug("got environments", "environments", environments)
@@ -103,12 +114,11 @@ func (c *ClientV1) GetPlans(ctx context.Context, projectCode string) ([]run.Plan
 
 	plans := make([]run.Plan, 0)
 
-	var limit int32 = defaultLimit
 	var offset int32 = 0
 	for {
 		resp, r, err := client.PlansAPI.
 			GetPlans(ctx, projectCode).
-			Limit(limit).
+			Limit(defaultLimit).
 			Offset(offset).
 			Execute()
 
@@ -124,10 +134,9 @@ func (c *ClientV1) GetPlans(ctx context.Context, projectCode string) ([]run.Plan
 			})
 		}
 
-		if resp.Result.GetTotal() <= limit {
+		if resp.Result.GetTotal() <= offset {
 			break
 		} else {
-			limit += defaultLimit
 			offset += defaultLimit
 		}
 	}