From df3ba0decf0bce739a2362c7718623ef05268643 Mon Sep 17 00:00:00 2001
From: soltysss <y.soltys@scalr.com>
Date: Tue, 13 Jun 2023 13:12:26 +0300
Subject: [PATCH] SCALRCORE-26173 - Terraform provider > clients > handle rate
 limit, fix tests

---
 scalr_test.go | 21 +++++++++++++++++----
 1 file changed, 17 insertions(+), 4 deletions(-)

diff --git a/scalr_test.go b/scalr_test.go
index fee156e..02c0715 100644
--- a/scalr_test.go
+++ b/scalr_test.go
@@ -8,6 +8,7 @@ import (
 	"io/ioutil"
 	"net/http"
 	"net/http/httptest"
+	"net/url"
 	"os"
 	"testing"
 )
@@ -190,26 +191,38 @@ func TestClient_retryHTTPCheck(t *testing.T) {
 		checkErr          error
 	}{
 		"429-no-server-errors": {
-			resp:     &http.Response{StatusCode: 429},
+			resp: &http.Response{
+				StatusCode: 429,
+				Request:    &http.Request{URL: &url.URL{Host: "scalr.test", Path: "/test/thing"}},
+			},
 			err:      nil,
 			checkOK:  true,
 			checkErr: nil,
 		},
 		"429-with-server-errors": {
-			resp:              &http.Response{StatusCode: 429},
+			resp: &http.Response{
+				StatusCode: 429,
+				Request:    &http.Request{URL: &url.URL{Host: "scalr.test", Path: "/test/thing"}},
+			},
 			err:               nil,
 			retryServerErrors: true,
 			checkOK:           true,
 			checkErr:          nil,
 		},
 		"500-no-server-errors": {
-			resp:     &http.Response{StatusCode: 500},
+			resp: &http.Response{
+				StatusCode: 500,
+				Request:    &http.Request{URL: &url.URL{Host: "scalr.test", Path: "/test/thing"}},
+			},
 			err:      nil,
 			checkOK:  false,
 			checkErr: nil,
 		},
 		"500-with-server-errors": {
-			resp:              &http.Response{StatusCode: 500},
+			resp: &http.Response{
+				StatusCode: 500,
+				Request:    &http.Request{URL: &url.URL{Host: "scalr.test", Path: "/test/thing"}},
+			},
 			err:               nil,
 			retryServerErrors: true,
 			checkOK:           true,