go-zero/rest/handlers_test.go

package rest

import (
	"net/http"
	"net/http/httptest"
	"testing"

	"github.com/stretchr/testify/assert"
)

func TestCorsHandlerWithOrigins(t *testing.T) {
	tests := []struct {
		name    string
		origins []string
		expect  string
	}{
		{
			name:   "allow all origins",
			expect: allOrigins,
		},
		{
			name:    "allow one origin",
			origins: []string{"local"},
			expect:  "local",
		},
		{
			name:    "allow many origins",
			origins: []string{"local", "remote"},
			expect:  "local",
		},
	}

	for _, test := range tests {
		t.Run(test.name, func(t *testing.T) {
			w := httptest.NewRecorder()
			handler := CorsHandler(test.origins...)
			handler.ServeHTTP(w, nil)
			assert.Equal(t, http.StatusNoContent, w.Result().StatusCode)
			assert.Equal(t, test.expect, w.Header().Get(allowOrigin))
		})
	}
}