Support a default context in the kubecfg template

kuberos.go

@@ -338,6 +338,7 @@ func populateUser(cfg *api.Config, p *extractor.OIDCAuthenticationParams) api.Co
 	c.AuthInfos = make(map[string]*api.AuthInfo)
 	c.Clusters = make(map[string]*api.Cluster)
 	c.Contexts = make(map[string]*api.Context)
+	c.CurrentContext = cfg.CurrentContext
 	c.AuthInfos[p.Username] = &api.AuthInfo{
 		AuthProvider: &api.AuthProviderConfig{
 			Name: templateAuthProvider,

kuberos_test.go

@@ -127,6 +127,49 @@ func TestPopulateUser(t *testing.T) {
 				},
 			},
 		},
+		{
+			name: "MultiClusterWithContext",
+			cfg: &api.Config{
+				Clusters: map[string]*api.Cluster{
+					"a": &api.Cluster{Server: "https://example.org", CertificateAuthorityData: []byte("PAM")},
+					"b": &api.Cluster{Server: "https://example.net", CertificateAuthorityData: []byte("PAM")},
+				},
+				CurrentContext: "a",
+			},
+			params: &extractor.OIDCAuthenticationParams{
+				Username:     "[email protected]",
+				ClientID:     "id",
+				ClientSecret: "secret",
+				IDToken:      "token",
+				RefreshToken: "refresh",
+				IssuerURL:    "https://example.org",
+			},
+			want: api.Config{
+				Clusters: map[string]*api.Cluster{
+					"a": &api.Cluster{Server: "https://example.org", CertificateAuthorityData: []byte("PAM")},
+					"b": &api.Cluster{Server: "https://example.net", CertificateAuthorityData: []byte("PAM")},
+				},
+				Contexts: map[string]*api.Context{
+					"a": &api.Context{AuthInfo: "[email protected]", Cluster: "a"},
+					"b": &api.Context{AuthInfo: "[email protected]", Cluster: "b"},
+				},
+				AuthInfos: map[string]*api.AuthInfo{
+					"[email protected]": &api.AuthInfo{
+						AuthProvider: &api.AuthProviderConfig{
+							Name: templateAuthProvider,
+							Config: map[string]string{
+								templateOIDCClientID:     "id",
+								templateOIDCClientSecret: "secret",
+								templateOIDCIDToken:      "token",
+								templateOIDCRefreshToken: "refresh",
+								templateOIDCIssuer:       "https://example.org",
+							},
+						},
+					},
+				},
+				CurrentContext: "a",
+			},
+		},
 	}
 
 	for _, tt := range cases {