Function TestQuitQuitQuitHTTPPost

cmd/root_test.go:1464–1503

(t *testing.T)

}

func TestQuitQuitQuitHTTPPost(t *testing.T) {
	c := NewCommand(WithDialer(&spyDialer{}))
	c.SilenceUsage = true
	c.SilenceErrors = true
	c.SetArgs([]string{"--quitquitquit", "--admin-port", "9192", "my-project:my-region:my-instance"})
	ctx, cancel := context.WithCancel(context.Background())
	defer cancel()

	errCh := make(chan error)
	go func() {
		err := c.ExecuteContext(ctx)
		errCh <- err
	}()
	resp, err := tryDial("HEAD", "http://localhost:9192/quitquitquit")
	if err != nil {
		t.Fatalf("failed to dial endpoint: %v", err)
	}
	if resp.StatusCode != http.StatusBadRequest {
		t.Fatalf("expected a 400 status, got = %v", resp.StatusCode)
	}
	resp, err = tryDial("POST", "http://localhost:9192/quitquitquit")
	if err != nil {
		t.Fatalf("failed to dial endpoint: %v", err)
	}
	if resp.StatusCode != http.StatusOK {
		t.Fatalf("expected a 200 status, got = %v", resp.StatusCode)
	}

	var gotErr error
	select {
	case err := <-errCh:
		gotErr = err
	case <-time.After(30 * time.Second):
		t.Fatal("timeout waiting for error")
	}

	if !errors.Is(gotErr, errQuitQuitQuit) {
		t.Fatalf("want = %v, got = %v", errQuitQuitQuit, gotErr)
	}
}

func TestQuitQuitQuitHTTPGet(t *testing.T) {
	c := NewCommand(WithDialer(&spyDialer{}))