(MAINT) Add the PE v1 /tokens endpoint (#90)

Author: eoinmcq

pkg/rbac/testdata/apidocs/GenerateToken-request.json

@@ -0,0 +1,5 @@
+{
+  "lifetime": "1y",
+  "description": "A token to be used with joy and care.",
+  "client": "PE console"
+}

pkg/rbac/testdata/apidocs/GenerateToken-response.json

@@ -0,0 +1,3 @@
+{
+  "token": "some token"
+}

pkg/rbac/token.go

@@ -6,6 +6,7 @@ const (
 	requestAuthTokenURI  = "/rbac-api/v1/auth/token"              // #nosec - this is the uri to g et RBAC tokens
 	tokenAuthenticateURI = "/rbac-api/v2/auth/token/authenticate" // #nosec - this is the uri to authenticate RBAC tokens
 	tokenRevokeURI       = "/rbac-api/v2/tokens/"                 // #nosec - this is the uri to revoke individual RBAC tokens
+	tokenGenerateURI     = "/rbac-api/v1/tokens"                  // #nosec - this is the uri to generate a token.
 )
 
 // GetRBACToken returns an auth token given user/password information
@@ -66,6 +67,27 @@ func (c *Client) RevokeRBACToken(token string) error {
 	return nil
 }
 
+// GenerateRBACToken returns an RBAC token or errors otherwise
+func (c *Client) GenerateRBACToken(token string, request TokenRequest) (string, error) {
+	var payload Token
+
+	r, err := c.resty.R().
+		SetHeader("X-Authentication", token).
+		SetResult(&payload).
+		SetBody(request).
+		Post(tokenGenerateURI)
+	if err != nil {
+		return "", FormatError(r, err.Error())
+	}
+	if r.IsError() {
+		if r.Error() != nil {
+			return "", FormatError(r)
+		}
+		return "", FormatError(r)
+	}
+	return payload.Token, nil
+}
+
 // Token is the returned auth token
 type Token struct {
 	Token string `json:"token"`
@@ -95,3 +117,10 @@ type AuthenticateResponse struct {
 	UserID      string `json:"user_id"`
 	DisplayName string `json:"display_name"`
 }
+
+// TokenRequest will hold the details needed by the /tokens endpoint.
+type TokenRequest struct {
+	Lifetime    string `json:"lifetime,omitempty"`
+	Description string `json:"description,omitempty"`
+	Client      string `json:"client,omitempty"`
+}

pkg/rbac/token_test.go

@@ -65,3 +65,25 @@ func TestRevokeRBACToken(t *testing.T) {
 	err = rbacClient.RevokeRBACToken(tokenValue)
 	require.Equal(t, expectedError, err)
 }
+
+func TestGenerateRBACToken(t *testing.T) {
+	tokenValue := "some token"
+
+	tokenRequest := TokenRequest{
+		Description: "A token to be used with joy and care.",
+		Lifetime:    "1y",
+		Client:      "PE console",
+	}
+
+	// Test success
+	setupPostResponder(t, tokenGenerateURI, "GenerateToken-request.json", "GenerateToken-response.json")
+
+	tokenResponse, err := rbacClient.GenerateRBACToken(tokenValue, tokenRequest)
+	require.Nil(t, err)
+	require.Equal(t, tokenValue, tokenResponse)
+
+	// Test error
+	setUpBadRequestResponder(t, http.MethodPost, tokenGenerateURI)
+	_, err = rbacClient.GenerateRBACToken(tokenValue, tokenRequest)
+	require.Equal(t, expectedError, err)
+}