Function TestLoginCustomKeyPath

test/integration/login_test.go:126–221

(t *testing.T)

}

func TestLoginCustomKeyPath(t *testing.T) {
	tests := []struct {
		name    string
		keyType commands.KeyType
	}{
		{
			name:    "ECDSA Key",
			keyType: commands.ECDSA,
		},
		{
			name:    "ED25519 Key",
			keyType: commands.ED25519,
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			// Check that user can login and that valid openpubkey keys are written to
			// the correct places on disk

			// Setup fake OIDC server on localhost
			t.Log("------- setup OIDC server on localhost ------")
			opServer, err := NewFakeOpServer()
			require.NoError(t, err, "failed to create fake OIDC server")
			defer opServer.Close()
			t.Logf("OP server running at %s", opServer.URL)

			// Call login
			t.Log("------- call login cmd ------")
			errCh := make(chan error)
			opkProvider, loginURL, err := opServer.OpkProvider()
			require.NoError(t, err, "failed to create OPK provider")

			homePath, err := os.UserHomeDir()

			require.NoError(t, err)

			sshPath := filepath.Join(homePath, ".ssh")

			// Make ~/.ssh if folder does not exist
			err = os.MkdirAll(sshPath, os.ModePerm)
			require.NoError(t, err)

			seckeyPath := filepath.Join(sshPath, "opkssh-key")
			defer removeKey(seckeyPath)

			go func() {
				loginCmd := commands.LoginCmd{Fs: afero.NewOsFs(), KeyTypeArg: tt.keyType}
				err := loginCmd.Login(TestCtx, opkProvider, false, seckeyPath)
				errCh <- err
			}()

			// Wait for auth callback server on localhost to come up. It should come up
			// when login command is called
			timeoutErr := WaitForServer(TestCtx, fmt.Sprintf("%s://%s", loginURL.Scheme, loginURL.Host), LoginCallbackServerTimeout)
			require.NoError(t, timeoutErr, "login callback server took too long to startup")

			// Do OIDC login