INNER CODE UNIT · Go

ChannelServer

rancher/system-upgrade-controller · e2e/framework/channel.go:10

func ChannelServer(location string, statusCode int) *httptest.Server {
	hostname, err := os.Hostname()
	if err != nil {
		Failf("cannot read hostname: %v", err)
	}
	server := &httptest.Server{
		Config: &http.Server{Handler: http.HandlerFunc(func(w http.ResponseWriter, _ *http.Request) {
			w.Header().Set("Location", location)
			w.WriteHeader(statusCode)
		})},
	}
	server.Listener, err = net.Listen("tcp", net.JoinHostPort(hostname, "0"))
	if err != nil {
		Failf("cannot create tcp listener: %v", err)
	}
	server.Start()
	return server
}

View source record →