179 lines · 5.3 KB · Go
RawDownload
// Copyright 2021 The Gitea Authors. All rights reserved.// SPDX-License-Identifier: MIT package web import (	"bytes"	"net/http"	"net/http/httptest"	"strconv"	"testing" 	chi "github.com/go-chi/chi/v5"	"github.com/stretchr/testify/assert"	"github.com/stretchr/testify/require") func TestRoute1(t *testing.T) {	buff := bytes.NewBufferString("")	recorder := httptest.NewRecorder()	recorder.Body = buff 	r := NewRoute()	r.Get("/{username}/{reponame}/{type:issues|pulls}", func(resp http.ResponseWriter, req *http.Request) {		username := chi.URLParam(req, "username")		assert.Equal(t, "gitea", username)		reponame := chi.URLParam(req, "reponame")		assert.Equal(t, "gitea", reponame)		tp := chi.URLParam(req, "type")		assert.Equal(t, "issues", tp)	}) 	req, err := http.NewRequest("GET", "http://localhost:8000/gitea/gitea/issues", nil)	require.NoError(t, err)	r.ServeHTTP(recorder, req)	assert.Equal(t, http.StatusOK, recorder.Code)} func TestRoute2(t *testing.T) {	buff := bytes.NewBufferString("")	recorder := httptest.NewRecorder()	recorder.Body = buff 	hit := -1 	r := NewRoute()	r.Group("/{username}/{reponame}", func() {		r.Group("", func() {			r.Get("/{type:issues|pulls}", func(resp http.ResponseWriter, req *http.Request) {				username := chi.URLParam(req, "username")				assert.Equal(t, "gitea", username)				reponame := chi.URLParam(req, "reponame")				assert.Equal(t, "gitea", reponame)				tp := chi.URLParam(req, "type")				assert.Equal(t, "issues", tp)				hit = 0			}) 			r.Get("/{type:issues|pulls}/{index}", func(resp http.ResponseWriter, req *http.Request) {				username := chi.URLParam(req, "username")				assert.Equal(t, "gitea", username)				reponame := chi.URLParam(req, "reponame")				assert.Equal(t, "gitea", reponame)				tp := chi.URLParam(req, "type")				assert.Equal(t, "issues", tp)				index := chi.URLParam(req, "index")				assert.Equal(t, "1", index)				hit = 1			})		}, func(resp http.ResponseWriter, req *http.Request) {			if stop, err := strconv.Atoi(req.FormValue("stop")); err == nil {				hit = stop				resp.WriteHeader(http.StatusOK)			}		}) 		r.Group("/issues/{index}", func() {			r.Get("/view", func(resp http.ResponseWriter, req *http.Request) {				username := chi.URLParam(req, "username")				assert.Equal(t, "gitea", username)				reponame := chi.URLParam(req, "reponame")				assert.Equal(t, "gitea", reponame)				index := chi.URLParam(req, "index")				assert.Equal(t, "1", index)				hit = 2			})		})	}) 	req, err := http.NewRequest("GET", "http://localhost:8000/gitea/gitea/issues", nil)	require.NoError(t, err)	r.ServeHTTP(recorder, req)	assert.Equal(t, http.StatusOK, recorder.Code)	assert.Equal(t, 0, hit) 	req, err = http.NewRequest("GET", "http://localhost:8000/gitea/gitea/issues/1", nil)	require.NoError(t, err)	r.ServeHTTP(recorder, req)	assert.Equal(t, http.StatusOK, recorder.Code)	assert.Equal(t, 1, hit) 	req, err = http.NewRequest("GET", "http://localhost:8000/gitea/gitea/issues/1?stop=100", nil)	require.NoError(t, err)	r.ServeHTTP(recorder, req)	assert.Equal(t, http.StatusOK, recorder.Code)	assert.Equal(t, 100, hit) 	req, err = http.NewRequest("GET", "http://localhost:8000/gitea/gitea/issues/1/view", nil)	require.NoError(t, err)	r.ServeHTTP(recorder, req)	assert.Equal(t, http.StatusOK, recorder.Code)	assert.Equal(t, 2, hit)} func TestRoute3(t *testing.T) {	buff := bytes.NewBufferString("")	recorder := httptest.NewRecorder()	recorder.Body = buff 	hit := -1 	m := NewRoute()	r := NewRoute()	r.Mount("/api/v1", m) 	m.Group("/repos", func() {		m.Group("/{username}/{reponame}", func() {			m.Group("/branch_protections", func() {				m.Get("", func(resp http.ResponseWriter, req *http.Request) {					hit = 0				})				m.Post("", func(resp http.ResponseWriter, req *http.Request) {					hit = 1				})				m.Group("/{name}", func() {					m.Get("", func(resp http.ResponseWriter, req *http.Request) {						hit = 2					})					m.Patch("", func(resp http.ResponseWriter, req *http.Request) {						hit = 3					})					m.Delete("", func(resp http.ResponseWriter, req *http.Request) {						hit = 4					})				})			})		})	}) 	req, err := http.NewRequest("GET", "http://localhost:8000/api/v1/repos/gitea/gitea/branch_protections", nil)	require.NoError(t, err)	r.ServeHTTP(recorder, req)	assert.Equal(t, http.StatusOK, recorder.Code)	assert.Equal(t, 0, hit) 	req, err = http.NewRequest("POST", "http://localhost:8000/api/v1/repos/gitea/gitea/branch_protections", nil)	require.NoError(t, err)	r.ServeHTTP(recorder, req)	assert.Equal(t, http.StatusOK, recorder.Code, http.StatusOK)	assert.Equal(t, 1, hit) 	req, err = http.NewRequest("GET", "http://localhost:8000/api/v1/repos/gitea/gitea/branch_protections/master", nil)	require.NoError(t, err)	r.ServeHTTP(recorder, req)	assert.Equal(t, http.StatusOK, recorder.Code)	assert.Equal(t, 2, hit) 	req, err = http.NewRequest("PATCH", "http://localhost:8000/api/v1/repos/gitea/gitea/branch_protections/master", nil)	require.NoError(t, err)	r.ServeHTTP(recorder, req)	assert.Equal(t, http.StatusOK, recorder.Code)	assert.Equal(t, 3, hit) 	req, err = http.NewRequest("DELETE", "http://localhost:8000/api/v1/repos/gitea/gitea/branch_protections/master", nil)	require.NoError(t, err)	r.ServeHTTP(recorder, req)	assert.Equal(t, http.StatusOK, recorder.Code)	assert.Equal(t, 4, hit)}