Add more unit tests

This commit is contained in:
Jamie Curnow
2023-08-02 15:32:16 +10:00
parent 155e09407f
commit 9ac0e8c943
7 changed files with 363 additions and 2 deletions

View File

@ -0,0 +1,25 @@
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"))
}