package egress

import (
	"context"
	"fmt"
	"os"
	"testing"

	"github.com/linkerd/linkerd2/testutil"
)

//////////////////////
///   TEST SETUP   ///
//////////////////////

var TestHelper *testutil.TestHelper

func TestMain(m *testing.M) {
	TestHelper = testutil.NewTestHelper()
	// Block test execution until control plane is running
	TestHelper.WaitUntilDeployReady(testutil.LinkerdDeployReplicasEdge)
	os.Exit(m.Run())
}

//////////////////////
/// TEST EXECUTION ///
//////////////////////

func TestEgressHttp(t *testing.T) {
	ctx := context.Background()
	out, err := TestHelper.LinkerdRun("inject", "testdata/proxy.yaml")
	if err != nil {
		testutil.AnnotatedFatal(t, "unexpected error", err)
	}

	TestHelper.WithDataPlaneNamespace(ctx, "egress-test", map[string]string{}, t, func(t *testing.T, ns string) {

		out, err = TestHelper.KubectlApply(out, ns)
		if err != nil {
			testutil.AnnotatedFatalf(t, "unexpected error", "unexpected error: %v output:\n%s", err, out)
		}

		err = TestHelper.CheckPods(ctx, ns, "egress-test", 1)
		if err != nil {
			//nolint:errorlint
			if rce, ok := err.(*testutil.RestartCountError); ok {
				testutil.AnnotatedWarn(t, "CheckPods timed-out", rce)
			} else {
				testutil.AnnotatedError(t, "CheckPods timed-out", err)
			}
		}

		testCase := func(url, methodToUse string) {
			testName := fmt.Sprintf("Can use egress to send %s request to (%s)", methodToUse, url)
			t.Run(testName, func(t *testing.T) {
				cmd := []string{
					"-n", ns, "exec", "deploy/egress-test", "-c", "http-egress",
					"--", "curl", "-sko", "/dev/null", "-w", "%{http_code}", "-X", methodToUse, url,
				}
				out, err := TestHelper.Kubectl("", cmd...)
				if err != nil {
					testutil.AnnotatedFatalf(t, fmt.Sprintf("failed to exec %s", cmd), "failed to exec %s: %s (%s)", cmd, err, out)
				}

				var status int
				_, err = fmt.Sscanf(out, "%d", &status)
				if err != nil {
					testutil.AnnotatedFatalf(t, "failed to parse status code", "failed to parse status code (%s): %s", out, err)
				}

				if status < 100 || status >= 500 {
					testutil.Fatalf(t, "got HTTP error code: %d\n", status)
				}
			})
		}

		supportedProtocols := []string{"http", "https"}
		methods := []string{"GET", "POST"}
		for _, protocolToUse := range supportedProtocols {
			for _, methodToUse := range methods {
				serviceName := fmt.Sprintf("%s://www.linkerd.io", protocolToUse)
				testCase(serviceName, methodToUse)
			}
		}

		// Test egress for a domain with fewer than 3 segments.
		testCase("http://linkerd.io", "GET")

	})
}