package edges

import (
	"io/ioutil"
	"os"
	"strings"
	"testing"

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

var TestHelper *testutil.TestHelper

func TestMain(m *testing.M) {
	TestHelper = testutil.NewTestHelper()
	os.Exit(m.Run())
}

// TestEdges requires that there has been traffic recently between linkerd-web
// and linkerd-controller for edges to have been registered, which is the
// case when running this test in the context of the other integration tests.

// This test has been disabled because it can fail due to
// https://github.com/linkerd/linkerd2/issues/3706
// This test should be updated and re-enabled when that issue is addressed.
/*
func TestEdges(t *testing.T) {
	ns := TestHelper.GetLinkerdNamespace()
	cmd := []string{
		"edges",
		"-n", ns,
		"deploy",
		"-ojson",
	}
	out, stderr, err := TestHelper.LinkerdRun(cmd...)
	if err != nil {
		t.Fatalf("Unexpected error: %v\nError output: %s", err, stderr)
	}

	tpl := template.Must(template.ParseFiles("testdata/linkerd_edges.golden"))
	vars := struct{ Ns string }{ns}
	var b bytes.Buffer
	if err := tpl.Execute(&b, vars); err != nil {
		t.Fatalf("failed to parse linkerd_edges.golden template: %s", err)
	}

	r := regexp.MustCompile(b.String())
	if !r.MatchString(out) {
		t.Errorf("Expected output:\n%s\nactual:\n%s", b.String(), out)
	}
}
*/

// TestDirectEdges deploys a terminus and then generates a load generator which
// sends traffic directly to the pod ip of the terminus pod. Traffic which is
// addressed this way (as opposed to using the service name) does not show up
// in the `linkerd edges` command. This test should be updated once
// `linkerd edges` is updated to support this kind of traffic.
func TestDirectEdges(t *testing.T) {

	// setup

	testNamespace := TestHelper.GetTestNamespace("direct-edges-test")
	err := TestHelper.CreateDataPlaneNamespaceIfNotExists(testNamespace, nil)
	if err != nil {
		t.Fatalf("failed to create %s namespace: %s", testNamespace, err)
	}

	// inject terminus

	out, stderr, err := TestHelper.LinkerdRun("inject", "--manual", "testdata/terminus.yaml")
	if err != nil {
		t.Fatalf("'linkerd %s' command failed with %s: %s\n", "inject", err.Error(), stderr)
	}

	// deploy terminus

	out, err = TestHelper.KubectlApply(out, testNamespace)
	if err != nil {
		t.Fatalf("kubectl apply command failed\n%s", out)
	}

	if err := TestHelper.CheckPods(testNamespace, "terminus", 1); err != nil {
		t.Error(err)
	}

	if err := TestHelper.CheckDeployment(testNamespace, "terminus", 1); err != nil {
		t.Errorf("Error validating deployment [%s]:\n%s", "terminus", err)
	}

	// get terminus pod ip

	ip, err := TestHelper.Kubectl("", "-n", testNamespace, "get", "pod", "-ojsonpath=\"{.items[*].status.podIP}\"")
	if err != nil {
		t.Error(err)
	}
	ip = strings.Trim(ip, "\"") // strip quotes

	bytes, err := ioutil.ReadFile("testdata/slow-cooker.yaml")
	if err != nil {
		t.Error(err)
	}

	slowcooker := string(bytes)
	slowcooker = strings.ReplaceAll(slowcooker, "___TERMINUS_POD_IP___", ip)

	// inject slow cooker

	out, stderr, err = TestHelper.PipeToLinkerdRun(slowcooker, "inject", "--manual", "-")
	if err != nil {
		t.Fatalf("'linkerd %s' command failed with %s: %s\n", "inject", err.Error(), stderr)
	}

	// deploy slow cooker

	out, err = TestHelper.KubectlApply(out, testNamespace)
	if err != nil {
		t.Fatalf("kubectl apply command failed\n%s", out)
	}

	if err := TestHelper.CheckPods(testNamespace, "slow-cooker", 1); err != nil {
		t.Error(err)
	}

	if err := TestHelper.CheckDeployment(testNamespace, "slow-cooker", 1); err != nil {
		t.Errorf("Error validating deployment [%s]:\n%s", "terminus", err)
	}

	// check edges

	_, stderr, err = TestHelper.LinkerdRun("-n", testNamespace, "edges", "pods")
	if err != nil {
		t.Fatalf("'linkerd %s' command failed with %s: %s\n", "edges", err.Error(), stderr)
	}
	stderr = strings.TrimSpace(stderr)
	if stderr != "No edges found." {
		t.Fatalf("Expected: \"%s\" Got: \"%s\"", "No edges found.", stderr)
	}
}