contentfulproxy/proxy/proxy_test.go

package proxy_test

import (
	"context"
	"io"
	"net/http"
	"net/http/httptest"
	"sync"
	"testing"
	"time"

	"github.com/foomo/contentfulproxy/proxy"
	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
	"go.uber.org/zap/zaptest"
)

const (
	responseFoo    = `i am a foo response`
	responseBar    = `i am bar response`
	responseUpdate = `update`
	responseFlush  = `update`
)

type getStats func(path string) int

func GetBackend(t *testing.T) (getStats, http.HandlerFunc) {
	t.Helper()
	stats := map[string]int{}
	statLock := sync.RWMutex{}
	return func(path string) int {
			statLock.RLock()
			defer statLock.RUnlock()
			count, ok := stats[path]
			if !ok {
				return -1
			}
			return count
		}, func(w http.ResponseWriter, r *http.Request) {
			statLock.Lock()
			stats[r.URL.Path]++
			statLock.Unlock()

			t.Log("backend: url called", r.URL.Path)

			switch r.URL.Path {
			case "/foo":
				_, _ = w.Write([]byte(responseFoo))
				return
			case "/bar":
				_, _ = w.Write([]byte(responseBar))
				return
			}
			http.Error(w, "not found", http.StatusNotFound)
		}
}

func GetWebHook(t *testing.T) (getStats, http.HandlerFunc) {
	t.Helper()
	stats := map[string]int{}
	statLock := sync.RWMutex{}
	return func(path string) int {
			statLock.RLock()
			defer statLock.RUnlock()
			count, ok := stats[path]
			if !ok {
				return -1
			}
			return count
		}, func(w http.ResponseWriter, r *http.Request) {
			statLock.Lock()
			stats[r.URL.Path]++
			statLock.Unlock()

			t.Log("webhook: url called", r.URL.Path)

			switch r.URL.Path {
			case "/test1":
				_, _ = w.Write([]byte(responseUpdate))
				return
			case "/test2":
				_, _ = w.Write([]byte(responseFlush))
				return
			}
			http.Error(w, "not found", http.StatusNotFound)
		}
}

func getTestServer(t *testing.T) (func(path string) int, func(path string) int, *httptest.Server) {
	t.Helper()
	l := zaptest.NewLogger(t)

	gs, backendHandler := GetBackend(t)
	ws, webHookHandler := GetWebHook(t)
	backendServer := httptest.NewServer(backendHandler)
	webHookServer := httptest.NewServer(webHookHandler)

	p, _ := proxy.NewProxy(
		context.Background(),
		l,
		func() string { return backendServer.URL },
		func() string { return "" },
		func() []string {
			return []string{
				webHookServer.URL + "/test1",
				webHookServer.URL + "/test2",
			}
		},
	)
	s := httptest.NewServer(p)
	t.Log("we have a proxy in front of it running on", s.URL)
	return gs, ws, s
}

func TestProxy(t *testing.T) {
	gs, ws, server := getTestServer(t)

	get := func(path string) string {
		resp, err := http.Get(server.URL + path)
		require.NoError(t, err)
		defer resp.Body.Close()
		responseBytes, err := io.ReadAll(resp.Body)
		require.NoError(t, err)
		return string(responseBytes)
	}
	for j := 0; j < 10; j++ {
		wg := sync.WaitGroup{}
		for i := 0; i < 10; i++ {
			wg.Add(1)
			go func() {
				get("/foo")
				wg.Done()
			}()
		}
		wg.Wait()
	}
	assert.Equal(t, 1, gs("/foo"))

	// check the current status
	// response, err := http.Get(server.URL + "/info")
	// require.NoError(t, err)

	//
	resp, err := http.Get(server.URL + "/update")
	require.NoError(t, err)
	defer resp.Body.Close()

	time.Sleep(time.Second * 1)

	//
	assert.Equal(t, 1, ws("/test1"))
	assert.Equal(t, 1, ws("/test2"))
}