Test that one-shot mode exits

Signed-off-by: Richard Wall <[email protected]>

Author: wallrj

pkg/agent/run_test.go

@@ -0,0 +1,71 @@
+package agent
+
+import (
+	"bytes"
+	"context"
+	"os"
+	"os/exec"
+	"testing"
+	"time"
+
+	"github.com/jetstack/preflight/pkg/logs"
+	"github.com/spf13/cobra"
+	"github.com/stretchr/testify/require"
+	"k8s.io/klog/v2"
+)
+
+// TestRunOneShot runs the agent in `--one-shot` mode and verifies that it exits
+// after the first data gathering iteration.
+func TestRunOneShot(t *testing.T) {
+	if _, found := os.LookupEnv("GO_CHILD"); found {
+		// Silence the warning about missing pod name for event generation
+		// TODO(wallrj): This should not be required when an `--input-file` has been supplied.
+		t.Setenv("POD_NAME", "venafi-kubernetes-e2e")
+		// Silence the error about missing kubeconfig.
+		// TODO(wallrj): This should not be required when an `--input-file` has been supplied.
+		t.Setenv("KUBECONFIG", "testdata/success/kubeconfig.yaml")
+
+		c := &cobra.Command{}
+		InitAgentCmdFlags(c, &Flags)
+		logs.AddFlags(c.Flags())
+
+		err := c.ParseFlags([]string{
+			"--one-shot",
+			// TODO(wallrj): This should not be required when an `--input-file` has been supplied.
+			"--api-token=should-not-be-required",
+			// TODO(wallrj): This should not be required when an `--input-file` has been supplied.
+			"--install-namespace=default",
+			"--agent-config-file=testdata/success/config.yaml",
+			"--input-path=testdata/success/input.json",
+			"--output-path=/dev/null",
+			"-v=1",
+		})
+		require.NoError(t, err)
+
+		logs.Initialize()
+		Run(c, nil)
+		klog.Flush()
+		return
+	}
+	t.Log("Running child process")
+	ctx, cancel := context.WithTimeout(context.Background(), time.Second*3)
+	defer cancel()
+	cmd := exec.CommandContext(ctx, os.Args[0], "-test.run=^TestRunOneShot$")
+	var (
+		stdout bytes.Buffer
+		stderr bytes.Buffer
+	)
+	cmd.Stdout = &stdout
+	cmd.Stderr = &stderr
+	cmd.Env = append(
+		os.Environ(),
+		"GO_CHILD=true",
+	)
+	err := cmd.Run()
+
+	stdoutStr := stdout.String()
+	stderrStr := stderr.String()
+	t.Logf("STDOUT\n%s\n", stdoutStr)
+	t.Logf("STDERR\n%s\n", stderrStr)
+	require.NoError(t, err, context.Cause(ctx))
+}

pkg/agent/testdata/success/config.yaml

@@ -0,0 +1,2 @@
+cluster_id: "venafi-kubernetes-agent-e2e"
+organization_id: "venafi-kubernetes-agent-e2e"

pkg/agent/testdata/success/input.json

@@ -0,0 +1 @@
+[]

pkg/agent/testdata/success/kubeconfig.yaml

@@ -0,0 +1,15 @@
+# Just enough kubeconfig to satisfy client-go
+apiVersion: v1
+kind: Config
+current-context: cluster-1
+contexts:
+- name: cluster-1
+  context:
+    cluster: cluster-1
+    user: user-1
+clusters:
+- name: cluster-1
+  cluster:
+    server: https://192.0.2.1:8443
+preferences: {}
+users: []