Merge pull request stakwork#2431 from MirzaHanan/httpget-unit-test

[Unit Tests]: `httpget`

feeds/generic_test.go

@@ -0,0 +1,165 @@
+package feeds
+
+import (
+	"fmt"
+	"net/http"
+	"net/http/httptest"
+	"strings"
+	"testing"
+	"time"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestHttpGet(t *testing.T) {
+	tests := []struct {
+		name           string
+		serverResponse func(w http.ResponseWriter, r *http.Request)
+		url            string
+		expectedBody   string
+		expectError    bool
+		errorContains  string
+		setup          func(*httptest.Server) string
+	}{
+		{
+			name: "Successful GET request with JSON response",
+			serverResponse: func(w http.ResponseWriter, r *http.Request) {
+				assert.Equal(t, "GET", r.Method)
+				assert.Equal(t, "application/json", r.Header.Get("Content-Type"))
+				w.Header().Set("Content-Type", "application/json")
+				fmt.Fprint(w, `{"status": "success"}`)
+			},
+			expectedBody: `{"status": "success"}`,
+			expectError:  false,
+		},
+		{
+			name: "Server returns 404",
+			serverResponse: func(w http.ResponseWriter, r *http.Request) {
+				w.WriteHeader(http.StatusNotFound)
+			},
+			expectedBody: "",
+			expectError:  false,
+		},
+		{
+			name: "Server returns 500",
+			serverResponse: func(w http.ResponseWriter, r *http.Request) {
+				w.WriteHeader(http.StatusInternalServerError)
+				fmt.Fprint(w, "Internal Server Error")
+			},
+			expectedBody: "Internal Server Error",
+			expectError:  false,
+		},
+		{
+			name: "Large response body",
+			serverResponse: func(w http.ResponseWriter, r *http.Request) {
+				largeBody := strings.Repeat("a", 1024*1024)
+				fmt.Fprint(w, largeBody)
+			},
+			expectedBody: strings.Repeat("a", 1024*1024),
+			expectError:  false,
+		},
+		{
+			name: "Slow response",
+			serverResponse: func(w http.ResponseWriter, r *http.Request) {
+				time.Sleep(100 * time.Millisecond)
+				fmt.Fprint(w, "Delayed response")
+			},
+			expectedBody: "Delayed response",
+			expectError:  false,
+		},
+		{
+			name: "Empty response body",
+			serverResponse: func(w http.ResponseWriter, r *http.Request) {
+			},
+			expectedBody: "",
+			expectError:  false,
+		},
+		{
+			name: "Response with special characters",
+			serverResponse: func(w http.ResponseWriter, r *http.Request) {
+				fmt.Fprint(w, "Special chars: ñ, é, 漢字, 🌟")
+			},
+			expectedBody: "Special chars: ñ, é, 漢字, 🌟",
+			expectError:  false,
+		},
+		{
+			name:          "Invalid URL with invalid port",
+			url:           "http://invalid.localhost:99999",
+			expectedBody:  "",
+			expectError:   true,
+			errorContains: "dial tcp",
+		},
+		{
+			name:          "Invalid URL with non-existent domain",
+			url:           "http://this-domain-does-not-exist-123456789.com",
+			expectedBody:  "",
+			expectError:   true,
+			errorContains: "no such host",
+		},
+		{
+			name:          "Invalid URL with invalid protocol",
+			url:           "invalid://example.com",
+			expectedBody:  "",
+			expectError:   true,
+			errorContains: "unsupported protocol",
+		},
+		{
+			name: "Connection closed by server",
+			serverResponse: func(w http.ResponseWriter, r *http.Request) {
+				hj, ok := w.(http.Hijacker)
+				if !ok {
+					t.Fatal("webserver doesn't support hijacking")
+				}
+				conn, _, err := hj.Hijack()
+				if err != nil {
+					t.Fatal(err)
+				}
+				conn.Close()
+			},
+			expectedBody:  "",
+			expectError:   true,
+			errorContains: "EOF",
+		},
+		{
+			name: "Chunked response",
+			serverResponse: func(w http.ResponseWriter, r *http.Request) {
+				flusher, ok := w.(http.Flusher)
+				if !ok {
+					t.Fatal("webserver doesn't support flushing")
+				}
+				fmt.Fprint(w, "chunk1")
+				flusher.Flush()
+				fmt.Fprint(w, "chunk2")
+			},
+			expectedBody: "chunk1chunk2",
+			expectError:  false,
+		},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			var server *httptest.Server
+			if tt.serverResponse != nil {
+				server = httptest.NewServer(http.HandlerFunc(tt.serverResponse))
+				defer server.Close()
+			}
+
+			url := tt.url
+			if server != nil {
+				url = server.URL
+			}
+
+			body, err := httpget(url)
+
+			if tt.expectError {
+				assert.Error(t, err)
+				if tt.errorContains != "" {
+					assert.Contains(t, err.Error(), tt.errorContains)
+				}
+			} else {
+				assert.NoError(t, err)
+				assert.Equal(t, tt.expectedBody, string(body))
+			}
+		})
+	}
+}