forked from mirrors/homebox
98 lines
1.8 KiB
Go
98 lines
1.8 KiB
Go
package server
|
|
|
|
import (
|
|
"net/http"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func testServer(t *testing.T, r http.Handler) *Server {
|
|
svr := NewServer("127.0.0.1", "19245")
|
|
|
|
go func() {
|
|
err := svr.Start(r)
|
|
assert.NoError(t, err)
|
|
}()
|
|
|
|
ping := func() error {
|
|
_, err := http.Get("http://127.0.0.1:19245")
|
|
return err
|
|
}
|
|
|
|
for {
|
|
if err := ping(); err == nil {
|
|
break
|
|
}
|
|
time.Sleep(time.Millisecond * 100)
|
|
}
|
|
|
|
return svr
|
|
}
|
|
|
|
func Test_ServerShutdown_Error(t *testing.T) {
|
|
svr := NewServer("127.0.0.1", "19245")
|
|
|
|
err := svr.Shutdown("test")
|
|
assert.ErrorIs(t, err, ErrServerNotStarted)
|
|
}
|
|
|
|
func Test_ServerStarts_Error(t *testing.T) {
|
|
svr := testServer(t, nil)
|
|
|
|
err := svr.Start(nil)
|
|
assert.ErrorIs(t, err, ErrServerAlreadyStarted)
|
|
|
|
err = svr.Shutdown("test")
|
|
assert.NoError(t, err)
|
|
}
|
|
|
|
func Test_ServerStarts(t *testing.T) {
|
|
svr := testServer(t, nil)
|
|
err := svr.Shutdown("test")
|
|
assert.NoError(t, err)
|
|
}
|
|
|
|
func Test_GracefulServerShutdownWithWorkers(t *testing.T) {
|
|
isFinished := false
|
|
|
|
svr := testServer(t, nil)
|
|
|
|
svr.Background(func() {
|
|
time.Sleep(time.Second * 4)
|
|
isFinished = true
|
|
})
|
|
|
|
err := svr.Shutdown("test")
|
|
|
|
assert.NoError(t, err)
|
|
assert.True(t, isFinished)
|
|
|
|
}
|
|
|
|
func Test_GracefulServerShutdownWithRequests(t *testing.T) {
|
|
isFinished := false
|
|
|
|
router := http.NewServeMux()
|
|
|
|
// add long running handler func
|
|
router.HandleFunc("/test", func(rw http.ResponseWriter, r *http.Request) {
|
|
time.Sleep(time.Second * 3)
|
|
isFinished = true
|
|
})
|
|
|
|
svr := testServer(t, router)
|
|
|
|
// Make request to "/test"
|
|
go func() {
|
|
_, _ = http.Get("http://127.0.0.1:19245/test") // This is probably bad?
|
|
}()
|
|
|
|
time.Sleep(time.Second) // Hack to wait for the request to be made
|
|
|
|
err := svr.Shutdown("test")
|
|
assert.NoError(t, err)
|
|
|
|
assert.True(t, isFinished)
|
|
}
|