89 lines
2.1 KiB
Go
89 lines
2.1 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"io"
|
|
"net"
|
|
"net/http"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func TestInvalidAuthConfigFailsBeforeDataCreation(t *testing.T) {
|
|
for _, key := range []string{"PANEL_PUBLIC_URL", "TRUSTED_PROXY_CIDRS"} {
|
|
t.Run(key, func(t *testing.T) {
|
|
t.Setenv("PANEL_PUBLIC_URL", "")
|
|
t.Setenv("TRUSTED_PROXY_CIDRS", "")
|
|
t.Setenv(key, "invalid")
|
|
root := filepath.Join(t.TempDir(), "not-created")
|
|
t.Setenv("CONTAINER_DATA_ROOT", root)
|
|
if err := run(); err == nil || !strings.Contains(err.Error(), key) {
|
|
t.Fatalf("unexpected error: %v", err)
|
|
}
|
|
if _, err := os.Stat(root); !os.IsNotExist(err) {
|
|
t.Fatalf("data directory created: %v", err)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestServeWaitsForActiveHandler(t *testing.T) {
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
defer cancel()
|
|
address := make(chan string, 1)
|
|
entered := make(chan struct{})
|
|
release := make(chan struct{})
|
|
srv := &http.Server{Addr: "127.0.0.1:0", BaseContext: func(l net.Listener) context.Context {
|
|
address <- l.Addr().String()
|
|
return context.Background()
|
|
}, Handler: http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
close(entered)
|
|
<-release
|
|
_, _ = w.Write([]byte("finished"))
|
|
})}
|
|
done := make(chan error, 1)
|
|
go func() { done <- serve(ctx, srv) }()
|
|
var addr string
|
|
select {
|
|
case addr = <-address:
|
|
case <-time.After(5 * time.Second):
|
|
t.Fatal("server did not listen")
|
|
}
|
|
requestDone := make(chan error, 1)
|
|
go func() {
|
|
resp, err := http.Get("http://" + addr)
|
|
if err == nil {
|
|
_, err = io.Copy(io.Discard, resp.Body)
|
|
resp.Body.Close()
|
|
}
|
|
requestDone <- err
|
|
}()
|
|
select {
|
|
case <-entered:
|
|
case <-time.After(5 * time.Second):
|
|
t.Fatal("handler did not start")
|
|
}
|
|
cancel()
|
|
select {
|
|
case err := <-done:
|
|
close(release)
|
|
t.Fatalf("returned before handler completed: %v", err)
|
|
case <-time.After(100 * time.Millisecond):
|
|
}
|
|
close(release)
|
|
select {
|
|
case err := <-done:
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
case <-time.After(5 * time.Second):
|
|
t.Fatal("shutdown stalled")
|
|
}
|
|
if err := <-requestDone; err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|