Function TestWaitCommandFlags

cmd/wait_test.go:24–57

(t *testing.T)

)

func TestWaitCommandFlags(t *testing.T) {
	ln, err := net.Listen("tcp", "localhost:0")
	if err != nil {
		t.Fatal(err)
	}
	defer ln.Close()
	host, port, err := net.SplitHostPort(ln.Addr().String())
	if err != nil {
		t.Fatal(err)
	}
	go func() {
		conn, err := ln.Accept()
		if err != nil {
			return
		}
		defer conn.Close()
		// Use a read deadline to produce read error
		conn.SetReadDeadline(time.Now().Add(100 * time.Millisecond))
		// Read client request first.
		io.ReadAll(conn)
		// Write a generic 200 response back.
		conn.Write([]byte("HTTP/1.1 200 OK\r\n\r\n"))
	}()

	_, err = invokeProxyCommand([]string{
		"wait",
		"--http-address", host,
		"--http-port", port,
		"--max=1s",
	})
	if err != nil {
		t.Fatal(err)
	}
}

func TestWaitCommandFails(t *testing.T) {
	_, err := invokeProxyCommand([]string{