Added endpoint suffix to service user for pkce token saved in token c…

…ache (flyteorg#264) * Added endpoint suffix to service user for pkce token saved in token cache Signed-off-by: Prafulla Mahindrakar <[email protected]> * nits Signed-off-by: Prafulla Mahindrakar <[email protected]>
eapolinario · Jan 21, 2022 · 85a80af · 85a80af
1 parent 2827633
commit 85a80af
Show file tree

Hide file tree

Showing 2 changed files with 29 additions and 1 deletion.
diff --git a/cmd/core/cmd.go b/cmd/core/cmd.go
@@ -58,9 +58,10 @@ func generateCommandFunc(cmdEntry CommandEntry) func(cmd *cobra.Command, args []
 			return err
 		}
 
+		adminCfg := admin.GetConfig(ctx)
 		clientSet, err := admin.ClientSetBuilder().WithConfig(admin.GetConfig(ctx)).
 			WithTokenCache(pkce.TokenCacheKeyringProvider{
-				ServiceUser: pkce.KeyRingServiceUser,
+				ServiceUser: fmt.Sprintf("%s:%s", adminCfg.Endpoint.String(), pkce.KeyRingServiceUser),
 				ServiceName: pkce.KeyRingServiceName,
 			}).Build(ctx)
 		if err != nil {

diff --git a/cmd/core/cmd_test.go b/cmd/core/cmd_test.go
@@ -0,0 +1,27 @@
+package cmdcore
+
+import (
+	"context"
+	"net/url"
+	"testing"
+
+	"github.com/flyteorg/flyteidl/clients/go/admin"
+	"github.com/flyteorg/flytestdlib/config"
+
+	"github.com/spf13/cobra"
+	"github.com/stretchr/testify/assert"
+)
+
+func testCommandFunc(ctx context.Context, args []string, cmdCtx CommandContext) error {
+	return nil
+}
+
+func TestGenerateCommandFunc(t *testing.T) {
+	adminCfg := admin.GetConfig(context.Background())
+	adminCfg.Endpoint = config.URL{URL: url.URL{Host: "dummyHost"}}
+	adminCfg.AuthType = admin.AuthTypePkce
+	rootCmd := &cobra.Command{}
+	cmdEntry := CommandEntry{CmdFunc: testCommandFunc, ProjectDomainNotRequired: true}
+	fn := generateCommandFunc(cmdEntry)
+	assert.Nil(t, fn(rootCmd, []string{}))
+}