rm httpbin from rest tests

Author: jahvon

internal/services/rest/rest_test.go

@@ -1,7 +1,10 @@
 package rest_test
 
 import (
+	"fmt"
 	"net/http"
+	"net/http/httptest"
+	"strconv"
 	"testing"
 	"time"
 
@@ -17,6 +20,37 @@ func TestRest(t *testing.T) {
 }
 
 var _ = Describe("Rest", func() {
+	var testServer *httptest.Server
+
+	BeforeEach(func() {
+		testServer = httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+			query := r.URL.Query()
+
+			if code := query.Get("code"); code != "" {
+				if statusCode, err := strconv.Atoi(code); err == nil {
+					w.WriteHeader(statusCode)
+					return
+				}
+			}
+			if sleep := query.Get("sleep"); sleep != "" {
+				if duration, err := time.ParseDuration(sleep); err == nil {
+					time.Sleep(duration)
+				}
+			}
+			if query.Get("print-headers") == "true" {
+				w.Header().Set("Content-Type", "application/json")
+				w.WriteHeader(http.StatusOK)
+				_, _ = fmt.Fprintf(w, `{"headers": {"%s": "%s"}}`,
+					"Test-Header", r.Header.Get("Test-Header"))
+				return
+			}
+
+			w.Header().Set("Content-Type", "application/json")
+			w.WriteHeader(http.StatusOK)
+			_, _ = w.Write([]byte(`success`))
+		}))
+	})
+
 	Context("SendRequest", func() {
 		It("should return error when invalid URL is provided", func() {
 			req := &rest.Request{
@@ -30,7 +64,7 @@ var _ = Describe("Rest", func() {
 
 		It("should return error when unexpected status code is received", func() {
 			req := &rest.Request{
-				URL:     "https://httpbin.org/status/500",
+				URL:     testServer.URL + "?code=500",
 				Method:  "GET",
 				Timeout: 30 * time.Second,
 			}
@@ -40,18 +74,18 @@ var _ = Describe("Rest", func() {
 
 		It("should return the correct body when a valid request is made", func() {
 			req := &rest.Request{
-				URL:     "https://httpbin.org/get",
+				URL:     testServer.URL,
 				Method:  "GET",
 				Timeout: 30 * time.Second,
 			}
 			resp, err := rest.SendRequest(req, []int{http.StatusOK})
 			Expect(err).NotTo(HaveOccurred())
-			Expect(resp.Body).To(ContainSubstring("\"url\": \"https://httpbin.org/get\""))
+			Expect(resp.Body).To(Equal("success"))
 		})
 
 		It("should timeout when the request takes longer than the specified timeout", func() {
 			req := &rest.Request{
-				URL:     "https://httpbin.org/delay/3",
+				URL:     testServer.URL + "?sleep=2s",
 				Method:  "GET",
 				Timeout: 1 * time.Second,
 			}
@@ -62,7 +96,7 @@ var _ = Describe("Rest", func() {
 
 		It("should return the correct headers when a valid request is made", func() {
 			req := &rest.Request{
-				URL:     "https://httpbin.org/headers",
+				URL:     testServer.URL + "?print-headers=true",
 				Method:  "GET",
 				Headers: map[string]string{"Test-Header": "Test-Value"},
 				Timeout: 30 * time.Second,