Pass config token as param

slack-go · Aug 4, 2023 · b9a30c0 · b9a30c0
1 parent a33e5ae
commit b9a30c0
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 5 deletions.
diff --git a/tokens.go b/tokens.go
@@ -6,12 +6,16 @@ import (
 )
 
 // RotateTokens exchanges a refresh token for a new app configuration token
-func (api *Client) RotateTokens(refreshToken string) (*TokenResponse, error) {
-	return api.RotateTokensContext(context.Background(), refreshToken)
+func (api *Client) RotateTokens(configToken string, refreshToken string) (*TokenResponse, error) {
+	return api.RotateTokensContext(context.Background(), configToken, refreshToken)
 }
 
 // RotateTokensContext exchanges a refresh token for a new app configuration token with a custom context
-func (api *Client) RotateTokensContext(ctx context.Context, refreshToken string) (*TokenResponse, error) {
+func (api *Client) RotateTokensContext(ctx context.Context, configToken string, refreshToken string) (*TokenResponse, error) {
+	if configToken == "" {
+		configToken = api.configToken
+	}
+
 	if refreshToken == "" {
 		refreshToken = api.configRefreshToken
 	}
@@ -21,7 +25,7 @@ func (api *Client) RotateTokensContext(ctx context.Context, refreshToken string)
 	}
 
 	response := &TokenResponse{}
-	err := api.getMethod(ctx, "tooling.tokens.rotate", api.configToken, values, response)
+	err := api.getMethod(ctx, "tooling.tokens.rotate", configToken, values, response)
 	if err != nil {
 		return nil, err
 	}

diff --git a/tokens_test.go b/tokens_test.go
@@ -14,7 +14,7 @@ func TestRotateTokens(t *testing.T) {
 	once.Do(startServer)
 	api := New("testing-token", OptionAPIURL("http://"+serverAddr+"/"))
 
-	tok, err := api.RotateTokens("old-refresh")
+	tok, err := api.RotateTokens("expired-config", "old-refresh")
 	if err != nil {
 		t.Errorf("Unexpected error: %v", err)
 		return