Enhance TestCLI integration test so it can catch mistakes with env vars
- Also remove a log statement from a test which caused a lot of extra output when the tests are run with `go test -v`
This commit is contained in:
parent
dba951fe89
commit
2ecb43154b
@ -27,6 +27,30 @@ func TestCLI(t *testing.T) {
|
|||||||
strings.ReplaceAll(library.GetEnv(t, "PINNIPED_TEST_USER_GROUPS"), " ", ""), ",",
|
strings.ReplaceAll(library.GetEnv(t, "PINNIPED_TEST_USER_GROUPS"), " ", ""), ",",
|
||||||
)
|
)
|
||||||
|
|
||||||
|
// Remove all Pinniped environment variables for the remainder of this test
|
||||||
|
// because some of their names clash with the env vars expected by our
|
||||||
|
// kubectl exec plugin. We would like this test to prove that the exec
|
||||||
|
// plugin receives all of the necessary env vars via the auto-generated
|
||||||
|
// kubeconfig from the Pinniped CLI.
|
||||||
|
initialEnvVars := make(map[string]string)
|
||||||
|
for _, e := range os.Environ() {
|
||||||
|
pair := strings.SplitN(e, "=", 2)
|
||||||
|
name := pair[0]
|
||||||
|
value := pair[1]
|
||||||
|
if strings.HasPrefix(name, "PINNIPED_") {
|
||||||
|
initialEnvVars[name] = value
|
||||||
|
err := os.Unsetenv(name)
|
||||||
|
require.NoError(t, err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
// Put them back for other tests to use after this one
|
||||||
|
t.Cleanup(func() {
|
||||||
|
for k, v := range initialEnvVars {
|
||||||
|
err := os.Setenv(k, v)
|
||||||
|
require.NoError(t, err)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
|
||||||
// Build pinniped CLI.
|
// Build pinniped CLI.
|
||||||
pinnipedExe, cleanupFunc := buildPinnipedCLI(t)
|
pinnipedExe, cleanupFunc := buildPinnipedCLI(t)
|
||||||
defer cleanupFunc()
|
defer cleanupFunc()
|
||||||
@ -34,6 +58,10 @@ func TestCLI(t *testing.T) {
|
|||||||
// Run pinniped CLI to get kubeconfig.
|
// Run pinniped CLI to get kubeconfig.
|
||||||
kubeConfig := runPinnipedCLI(t, pinnipedExe, token, namespaceName)
|
kubeConfig := runPinnipedCLI(t, pinnipedExe, token, namespaceName)
|
||||||
|
|
||||||
|
// In addition to the client-go based testing below, also try the kubeconfig
|
||||||
|
// with kubectl once just in case it is somehow different.
|
||||||
|
runKubectlCLI(t, kubeConfig, namespaceName, testUsername)
|
||||||
|
|
||||||
// Create Kubernetes client with kubeconfig from pinniped CLI.
|
// Create Kubernetes client with kubeconfig from pinniped CLI.
|
||||||
kubeClient := library.NewClientsetForKubeConfig(t, kubeConfig)
|
kubeClient := library.NewClientsetForKubeConfig(t, kubeConfig)
|
||||||
|
|
||||||
@ -87,3 +115,38 @@ func runPinnipedCLI(t *testing.T, pinnipedExe, token, namespaceName string) stri
|
|||||||
|
|
||||||
return string(output)
|
return string(output)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func runKubectlCLI(t *testing.T, kubeConfig, namespaceName, username string) string {
|
||||||
|
t.Helper()
|
||||||
|
|
||||||
|
f, err := ioutil.TempFile("", "pinniped-generated-kubeconfig-*")
|
||||||
|
require.NoError(t, err)
|
||||||
|
defer func() {
|
||||||
|
err := os.Remove(f.Name())
|
||||||
|
require.NoError(t, err)
|
||||||
|
}()
|
||||||
|
_, err = f.WriteString(kubeConfig)
|
||||||
|
require.NoError(t, err)
|
||||||
|
err = f.Close()
|
||||||
|
require.NoError(t, err)
|
||||||
|
|
||||||
|
output, err := exec.Command(
|
||||||
|
"kubectl",
|
||||||
|
"get",
|
||||||
|
"pods",
|
||||||
|
"--kubeconfig", f.Name(),
|
||||||
|
"--namespace", namespaceName,
|
||||||
|
).CombinedOutput()
|
||||||
|
|
||||||
|
// Expect an error because this user has no RBAC permission. However, the
|
||||||
|
// error message should state that we had already authenticated as the test user.
|
||||||
|
expectedErrorMessage := `Error from server (Forbidden): pods is forbidden: User "` +
|
||||||
|
username +
|
||||||
|
`" cannot list resource "pods" in API group "" in the namespace "` +
|
||||||
|
namespaceName +
|
||||||
|
`"` + "\n"
|
||||||
|
require.EqualError(t, err, "exit status 1")
|
||||||
|
require.Equal(t, expectedErrorMessage, string(output))
|
||||||
|
|
||||||
|
return string(output)
|
||||||
|
}
|
||||||
|
@ -34,9 +34,6 @@ func TestCredentialIssuerConfig(t *testing.T) {
|
|||||||
CredentialIssuerConfigs(namespaceName).
|
CredentialIssuerConfigs(namespaceName).
|
||||||
List(ctx, metav1.ListOptions{})
|
List(ctx, metav1.ListOptions{})
|
||||||
require.NoError(t, err)
|
require.NoError(t, err)
|
||||||
for _, config := range actualConfigList.Items {
|
|
||||||
t.Logf("found CredentialIssuerConfig: %+v", config)
|
|
||||||
}
|
|
||||||
|
|
||||||
require.Len(t, actualConfigList.Items, 1)
|
require.Len(t, actualConfigList.Items, 1)
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user