26 lines
482 B
Go
26 lines
482 B
Go
|
package middleware
|
||
|
|
|
||
|
|
import (
|
||
|
|
"net/http"
|
||
|
|
"net/http/httptest"
|
||
|
|
"testing"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"git.vinchent.xyz/vinchent/go-web/framework"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestTimeout(t *testing.T) {
|
||
|
|
t.Run("Test timeout handler", func(t *testing.T) {
|
||
|
|
timeoutHandler := Timeout(1 * time.Millisecond)
|
||
|
|
|
||
|
|
request := httptest.NewRequest(http.MethodGet, "/", nil)
|
||
|
|
response := httptest.NewRecorder()
|
||
|
|
c := framework.NewContext(response, request)
|
||
|
|
|
||
|
|
err := timeoutHandler(c)
|
||
|
|
if err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
})
|
||
|
|
}
|