integration_test.go

package main

import (
	"io"
	"net/http"
	"net/http/httptest"
	"sync"
	"testing"
)

type recordedReq struct {
	Method   string
	Path     string
	RawQuery string
	Header   http.Header
	Body     string
}

func TestIntegration(t *testing.T) {
	var mu sync.Mutex
	var lastReq recordedReq

	server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		mu.Lock()
		defer mu.Unlock()

		bodyBytes, _ := io.ReadAll(r.Body)
		lastReq = recordedReq{
			Method:   r.Method,
			Path:     r.URL.Path,
			RawQuery: r.URL.RawQuery,
			Header:   r.Header,
			Body:     string(bodyBytes),
		}

		w.Header().Set("Content-Type", "application/json")
		w.WriteHeader(http.StatusOK)
		w.Write([]byte(`{"status": "ok"}`))
	}))
	defer server.Close()

	baseAddr = server.URL
	gLabel = "/api/v1/users"

	// Clear headers
	gHeaders = make(map[string]string)

	// 1. GET requests
	t.Run("GET request with no subpath", func(t *testing.T) {
		makeRequest("GET", []string{"get"})
		
		mu.Lock()
		req := lastReq
		mu.Unlock()

		if req.Method != "GET" {
			t.Errorf("expected GET, got %s", req.Method)
		}
		if req.Path != "/api/v1/users" {
			t.Errorf("expected /api/v1/users, got %s", req.Path)
		}
	})

	t.Run("GET request with subpath", func(t *testing.T) {
		makeRequest("GET", []string{"get", "info"})

		mu.Lock()
		req := lastReq
		mu.Unlock()

		if req.Path != "/api/v1/users/info" {
			t.Errorf("expected /api/v1/users/info, got %s", req.Path)
		}
	})

	t.Run("GET request with query params only", func(t *testing.T) {
		makeRequest("GET", []string{"get", "?limit=5"})

		mu.Lock()
		req := lastReq
		mu.Unlock()

		if req.Path != "/api/v1/users" {
			t.Errorf("expected /api/v1/users, got %s", req.Path)
		}
		if req.RawQuery != "limit=5" {
			t.Errorf("expected limit=5, got %s", req.RawQuery)
		}
	})

	t.Run("GET request with subpath and query params", func(t *testing.T) {
		makeRequest("GET", []string{"get", "info?limit=10"})

		mu.Lock()
		req := lastReq
		mu.Unlock()

		if req.Path != "/api/v1/users/info" {
			t.Errorf("expected /api/v1/users/info, got %s", req.Path)
		}
		if req.RawQuery != "limit=10" {
			t.Errorf("expected limit=10, got %s", req.RawQuery)
		}
	})

	// 2. Custom headers
	t.Run("Header commands", func(t *testing.T) {
		handleHeaderCommand([]string{"set", "header", "X-Test-Header", "HelloHeader"})
		if gHeaders["X-Test-Header"] != "HelloHeader" {
			t.Errorf("expected header to be set")
		}

		makeRequest("GET", []string{"get"})
		
		mu.Lock()
		req := lastReq
		mu.Unlock()

		if req.Header.Get("X-Test-Header") != "HelloHeader" {
			t.Errorf("expected header X-Test-Header to be HelloHeader, got %s", req.Header.Get("X-Test-Header"))
		}

		handleClearCommand([]string{"clear", "header", "X-Test-Header"})
		if _, ok := gHeaders["X-Test-Header"]; ok {
			t.Errorf("expected header to be cleared")
		}
	})

	// 3. POST request with content
	t.Run("POST request with inline content", func(t *testing.T) {
		makeRequest("POST", []string{"post", "-c", `{"name":"test"}`})

		mu.Lock()
		req := lastReq
		mu.Unlock()

		if req.Method != "POST" {
			t.Errorf("expected POST, got %s", req.Method)
		}
		if req.Body != `{"name":"test"}` {
			t.Errorf("expected request body to match, got %s", req.Body)
		}
		if req.Header.Get("Content-Type") != "application/json" {
			t.Errorf("expected Content-Type application/json, got %s", req.Header.Get("Content-Type"))
		}
	})

	// 4. PUT/PATCH/DELETE/HEAD/OPTIONS requests
	t.Run("PUT request", func(t *testing.T) {
		makeRequest("PUT", []string{"put", "-c", "put-body"})
		mu.Lock()
		req := lastReq
		mu.Unlock()
		if req.Method != "PUT" {
			t.Errorf("expected PUT, got %s", req.Method)
		}
	})

	t.Run("DELETE request", func(t *testing.T) {
		makeRequest("DELETE", []string{"delete"})
		mu.Lock()
		req := lastReq
		mu.Unlock()
		if req.Method != "DELETE" {
			t.Errorf("expected DELETE, got %s", req.Method)
		}
	})

	t.Run("OPTIONS request", func(t *testing.T) {
		makeRequest("OPTIONS", []string{"options"})
		mu.Lock()
		req := lastReq
		mu.Unlock()
		if req.Method != "OPTIONS" {
			t.Errorf("expected OPTIONS, got %s", req.Method)
		}
	})
}