package devtunnel_test

import (
	"context"
	"io"
	"net"
	"net/http"
	"testing"
	"time"

	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"

	"cdr.dev/slog/sloggers/slogtest"
	"github.com/coder/coder/coderd/devtunnel"
)

// The tunnel leaks a few goroutines that aren't impactful to production scenarios.
// func TestMain(m *testing.M) {
// 	goleak.VerifyTestMain(m)
// }

func TestTunnel(t *testing.T) {
	t.Parallel()

	// It's not super useful for us to test this constantly, it'll only cause
	// flakes is the tunnel becomes unavailable for some reason.
	t.Skip()
	// if testing.Short() {
	// 	t.Skip()
	// 	return
	// }

	ctx, cancelTun := context.WithCancel(context.Background())
	defer cancelTun()

	server := http.Server{
		Handler: http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
			w.WriteHeader(http.StatusOK)
		}),
		BaseContext: func(_ net.Listener) context.Context {
			return ctx
		},
	}

	cfg, err := devtunnel.GenerateConfig()
	require.NoError(t, err)

	tun, errCh, err := devtunnel.NewWithConfig(ctx, slogtest.Make(t, nil), cfg)
	require.NoError(t, err)
	t.Log(tun.URL)

	go server.Serve(tun.Listener)
	defer tun.Listener.Close()

	httpClient := &http.Client{
		Timeout: 10 * time.Second,
	}

	require.Eventually(t, func() bool {
		req, err := http.NewRequestWithContext(ctx, "GET", tun.URL, nil)
		require.NoError(t, err)

		res, err := httpClient.Do(req)
		require.NoError(t, err)
		defer res.Body.Close()
		_, _ = io.Copy(io.Discard, res.Body)

		return res.StatusCode == http.StatusOK
	}, time.Minute, time.Second)

	httpClient.CloseIdleConnections()
	assert.NoError(t, server.Close())
	cancelTun()

	select {
	case <-errCh:
	case <-time.After(10 * time.Second):
		t.Error("tunnel did not close after 10 seconds")
	}
}