package auth import ( "net/http" "net/http/httptest" "strconv" "testing" "github.com/stretchr/testify/require" ) func TestParseAuthTokenFromRequest(t *testing.T) { cases := []struct { header string cookie string query string expectedToken string expectedLocation TokenLocation }{ {"", "", "", "", TokenLocationNotFound}, {"token mytoken", "", "", "mytoken", TokenLocationHeader}, {"BEARER mytoken", "", "", "mytoken", TokenLocationHeader}, {"", "mytoken", "", "mytoken", TokenLocationCookie}, {"", "", "mytoken", "mytoken", TokenLocationQueryString}, } for testnum, tc := range cases { pathname := "/test/here" if tc.query != "" { pathname += "?access_token=" + tc.query } req := httptest.NewRequest("GET", pathname, nil) if tc.header != "" { req.Header.Add(HEADER_AUTH, tc.header) } if tc.cookie != "" { req.AddCookie(&http.Cookie{ Name: SESSION_COOKIE_TOKEN, Value: tc.cookie, }) } token, location := ParseAuthTokenFromRequest(req) require.Equal(t, tc.expectedToken, token, "Wrong token on test "+strconv.Itoa(testnum)) require.Equal(t, tc.expectedLocation, location, "Wrong location on test "+strconv.Itoa(testnum)) } }