Initial commit: DSH Web Manager

cmd/server/main_test.go

@@ -0,0 +1,88 @@
+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)
+	}
+}