mirror of
https://github.com/xiaoxinpro/nginx-proxy-manager-zh.git
synced 2025-01-24 05:38:15 -05:00
26 lines
598 B
Go
26 lines
598 B
Go
|
package middleware_test
|
||
|
|
|
||
|
|
import (
|
||
|
|
"net/http"
|
||
|
|
"net/http/httptest"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"npm/internal/api/middleware"
|
||
|
|
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestAccessControl(t *testing.T) {
|
||
|
|
handler := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||
|
|
w.WriteHeader(http.StatusOK)
|
||
|
|
})
|
||
|
|
|
||
|
|
rr := httptest.NewRecorder()
|
||
|
|
req, err := http.NewRequest("GET", "/", nil)
|
||
|
|
assert.Nil(t, err)
|
||
|
|
accessControl := middleware.AccessControl(handler)
|
||
|
|
accessControl.ServeHTTP(rr, req)
|
||
|
|
assert.Equal(t, http.StatusOK, rr.Code)
|
||
|
|
assert.Equal(t, "*", rr.Header().Get("Access-Control-Allow-Origin"))
|
||
|
|
}
|