safeweb: allow passing http.Server in safeweb.Config (#13688)
Extend safeweb.Config with the ability to pass a http.Server that safeweb will use to server traffic. Updates corp#8207 Signed-off-by: Patrick O'Doherty <patrick@tailscale.com>
This commit is contained in:
committed by
GitHub
parent
8fdffb8da0
commit
4ad3f01225
@@ -10,6 +10,7 @@ import (
|
||||
"strconv"
|
||||
"strings"
|
||||
"testing"
|
||||
"time"
|
||||
|
||||
"github.com/google/go-cmp/cmp"
|
||||
"github.com/gorilla/csrf"
|
||||
@@ -609,3 +610,26 @@ func TestStrictTransportSecurityOptions(t *testing.T) {
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
func TestOverrideHTTPServer(t *testing.T) {
|
||||
s, err := NewServer(Config{})
|
||||
if err != nil {
|
||||
t.Fatalf("NewServer: %v", err)
|
||||
}
|
||||
if s.h.IdleTimeout != 0 {
|
||||
t.Fatalf("got %v; want 0", s.h.IdleTimeout)
|
||||
}
|
||||
|
||||
c := http.Server{
|
||||
IdleTimeout: 10 * time.Second,
|
||||
}
|
||||
|
||||
s, err = NewServer(Config{HTTPServer: &c})
|
||||
if err != nil {
|
||||
t.Fatalf("NewServer: %v", err)
|
||||
}
|
||||
|
||||
if s.h.IdleTimeout != c.IdleTimeout {
|
||||
t.Fatalf("got %v; want %v", s.h.IdleTimeout, c.IdleTimeout)
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user