github.com/Racer159/jackal@v0.32.7-0.20240401174413-0bd2339e4f2e/src/test/e2e/21_connect_creds_test.go

// SPDX-License-Identifier: Apache-2.0
// SPDX-FileCopyrightText: 2021-Present The Jackal Authors

// Package test provides e2e tests for Jackal.
package test

import (
	"crypto/tls"
	"fmt"
	"io"
	"net/http"
	"strings"
	"testing"

	"github.com/Racer159/jackal/src/pkg/cluster"
	"github.com/stretchr/testify/require"
)

type RegistryResponse struct {
	Repositories []string `json:"repositories"`
}

func TestConnectAndCreds(t *testing.T) {
	t.Log("E2E: Connect")
	e2e.SetupWithCluster(t)

	prevAgentSecretData, _, err := e2e.Kubectl("get", "secret", "agent-hook-tls", "-n", "jackal", "-o", "jsonpath={.data}")
	require.NoError(t, err)

	connectToJackalServices(t)

	stdOut, stdErr, err := e2e.Jackal("tools", "update-creds", "--confirm")
	require.NoError(t, err, stdOut, stdErr)

	newAgentSecretData, _, err := e2e.Kubectl("get", "secret", "agent-hook-tls", "-n", "jackal", "-o", "jsonpath={.data}")
	require.NoError(t, err)
	require.NotEqual(t, prevAgentSecretData, newAgentSecretData, "agent secrets should not be the same")

	connectToJackalServices(t)

	stdOut, stdErr, err = e2e.Jackal("package", "remove", "init", "--components=logging", "--confirm")
	require.NoError(t, err, stdOut, stdErr)

	// Prune the images from Grafana and ensure that they are gone
	stdOut, stdErr, err = e2e.Jackal("tools", "registry", "prune", "--confirm")
	require.NoError(t, err, stdOut, stdErr)

	stdOut, stdErr, err = e2e.Jackal("tools", "registry", "ls", "127.0.0.1:31337/library/registry")
	require.NoError(t, err, stdOut, stdErr)
	require.Contains(t, stdOut, "2.8.3")
	stdOut, stdErr, err = e2e.Jackal("tools", "registry", "ls", "127.0.0.1:31337/grafana/promtail")
	require.NoError(t, err, stdOut, stdErr)
	require.Equal(t, stdOut, "")
	stdOut, stdErr, err = e2e.Jackal("tools", "registry", "ls", "127.0.0.1:31337/grafana/grafana")
	require.NoError(t, err, stdOut, stdErr)
	require.Equal(t, stdOut, "")
	stdOut, stdErr, err = e2e.Jackal("tools", "registry", "ls", "127.0.0.1:31337/grafana/loki")
	require.NoError(t, err, stdOut, stdErr)
	require.Equal(t, stdOut, "")
}

func TestMetrics(t *testing.T) {
	t.Log("E2E: Emits metrics")
	e2e.SetupWithCluster(t)

	c, err := cluster.NewCluster()
	require.NoError(t, err)

	tunnel, err := c.NewTunnel("jackal", "svc", "agent-hook", "", 8888, 8443)
	require.NoError(t, err)
	_, err = tunnel.Connect()
	require.NoError(t, err)
	defer tunnel.Close()

	// Skip certificate verification
	// this is an https endpoint being accessed through port-forwarding
	tr := &http.Transport{
		TLSClientConfig: &tls.Config{InsecureSkipVerify: true},
	}

	client := &http.Client{Transport: tr}
	httpsEndpoint := strings.ReplaceAll(tunnel.HTTPEndpoint(), "http", "https")
	resp, err := client.Get(httpsEndpoint + "/metrics")
	if err != nil {
		t.Fatal(err)
	}
	defer resp.Body.Close()

	// Read the response body
	body, err := io.ReadAll(resp.Body)
	if err != nil {
		t.Fatal(err)
	}

	desiredString := "go_gc_duration_seconds_count"
	require.Equal(t, true, strings.Contains(string(body), desiredString))
	require.NoError(t, err, resp)
	require.Equal(t, 200, resp.StatusCode)
}

func connectToJackalServices(t *testing.T) {
	// Make the Registry contains the images we expect
	stdOut, stdErr, err := e2e.Jackal("tools", "registry", "catalog")
	require.NoError(t, err, stdOut, stdErr)
	registryList := strings.Split(strings.Trim(stdOut, "\n "), "\n")

	// We assert greater than or equal to since the base init has 12 images
	// HOWEVER during an upgrade we could have mismatched versions/names resulting in more images
	require.GreaterOrEqual(t, len(registryList), 7)
	require.Contains(t, stdOut, "Racer159/jackal/agent")
	require.Contains(t, stdOut, "gitea/gitea")
	require.Contains(t, stdOut, "grafana/grafana")
	require.Contains(t, stdOut, "grafana/loki")
	require.Contains(t, stdOut, "grafana/promtail")
	require.Contains(t, stdOut, "kiwigrid/k8s-sidecar")
	require.Contains(t, stdOut, "library/registry")

	// Get the git credentials
	stdOut, stdErr, err = e2e.Jackal("tools", "get-creds", "git")
	require.NoError(t, err, stdOut, stdErr)
	gitPushPassword := strings.TrimSpace(stdOut)
	stdOut, stdErr, err = e2e.Jackal("tools", "get-creds", "git-readonly")
	require.NoError(t, err, stdOut, stdErr)
	gitPullPassword := strings.TrimSpace(stdOut)
	stdOut, stdErr, err = e2e.Jackal("tools", "get-creds", "artifact")
	require.NoError(t, err, stdOut, stdErr)
	gitArtifactToken := strings.TrimSpace(stdOut)

	// Connect to Gitea
	c, err := cluster.NewCluster()
	require.NoError(t, err)
	tunnelGit, err := c.Connect(cluster.JackalGit)
	require.NoError(t, err)
	defer tunnelGit.Close()

	// Make sure Gitea comes up cleanly
	gitPushURL := fmt.Sprintf("http://jackal-git-user:%s@%s/api/v1/user", gitPushPassword, tunnelGit.Endpoint())
	respGit, err := http.Get(gitPushURL)
	require.NoError(t, err)
	require.Equal(t, 200, respGit.StatusCode)
	gitPullURL := fmt.Sprintf("http://jackal-git-read-user:%s@%s/api/v1/user", gitPullPassword, tunnelGit.Endpoint())
	respGit, err = http.Get(gitPullURL)
	require.NoError(t, err)
	require.Equal(t, 200, respGit.StatusCode)
	gitArtifactURL := fmt.Sprintf("http://jackal-git-user:%s@%s/api/v1/user", gitArtifactToken, tunnelGit.Endpoint())
	respGit, err = http.Get(gitArtifactURL)
	require.NoError(t, err)
	require.Equal(t, 200, respGit.StatusCode)

	// Connect to the Logging Stack
	c, err = cluster.NewCluster()
	require.NoError(t, err)
	tunnelLog, err := c.Connect(cluster.JackalLogging)
	require.NoError(t, err)
	defer tunnelLog.Close()

	// Make sure Grafana comes up cleanly
	respLog, err := http.Get(tunnelLog.HTTPEndpoint())
	require.NoError(t, err)
	require.Equal(t, 200, respLog.StatusCode)
}