mirror of
https://codeberg.org/forgejo/forgejo.git
synced 2024-11-24 08:57:03 -05:00
072dd9f8bc
Reviewed-on: https://codeberg.org/forgejo/forgejo/pulls/4572 Co-authored-by: TheFox0x7 <thefox0x7@gmail.com> Co-committed-by: TheFox0x7 <thefox0x7@gmail.com>
65 lines
2 KiB
Go
65 lines
2 KiB
Go
// Copyright 2024 The Gitea Authors. All rights reserved.
|
|
// SPDX-License-Identifier: MIT
|
|
|
|
package actions
|
|
|
|
import (
|
|
"net/http"
|
|
"testing"
|
|
|
|
"code.gitea.io/gitea/modules/json"
|
|
"code.gitea.io/gitea/modules/setting"
|
|
|
|
"github.com/golang-jwt/jwt/v5"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestCreateAuthorizationToken(t *testing.T) {
|
|
var taskID int64 = 23
|
|
token, err := CreateAuthorizationToken(taskID, 1, 2)
|
|
require.NoError(t, err)
|
|
assert.NotEqual(t, "", token)
|
|
claims := jwt.MapClaims{}
|
|
_, err = jwt.ParseWithClaims(token, claims, func(t *jwt.Token) (interface{}, error) {
|
|
return setting.GetGeneralTokenSigningSecret(), nil
|
|
})
|
|
require.NoError(t, err)
|
|
scp, ok := claims["scp"]
|
|
assert.True(t, ok, "Has scp claim in jwt token")
|
|
assert.Contains(t, scp, "Actions.Results:1:2")
|
|
taskIDClaim, ok := claims["TaskID"]
|
|
assert.True(t, ok, "Has TaskID claim in jwt token")
|
|
assert.InDelta(t, float64(taskID), taskIDClaim, 0, "Supplied taskid must match stored one")
|
|
acClaim, ok := claims["ac"]
|
|
assert.True(t, ok, "Has ac claim in jwt token")
|
|
ac, ok := acClaim.(string)
|
|
assert.True(t, ok, "ac claim is a string for buildx gha cache")
|
|
scopes := []actionsCacheScope{}
|
|
err = json.Unmarshal([]byte(ac), &scopes)
|
|
require.NoError(t, err, "ac claim is a json list for buildx gha cache")
|
|
assert.GreaterOrEqual(t, len(scopes), 1, "Expected at least one action cache scope for buildx gha cache")
|
|
}
|
|
|
|
func TestParseAuthorizationToken(t *testing.T) {
|
|
var taskID int64 = 23
|
|
token, err := CreateAuthorizationToken(taskID, 1, 2)
|
|
require.NoError(t, err)
|
|
assert.NotEqual(t, "", token)
|
|
headers := http.Header{}
|
|
headers.Set("Authorization", "Bearer "+token)
|
|
rTaskID, err := ParseAuthorizationToken(&http.Request{
|
|
Header: headers,
|
|
})
|
|
require.NoError(t, err)
|
|
assert.Equal(t, taskID, rTaskID)
|
|
}
|
|
|
|
func TestParseAuthorizationTokenNoAuthHeader(t *testing.T) {
|
|
headers := http.Header{}
|
|
rTaskID, err := ParseAuthorizationToken(&http.Request{
|
|
Header: headers,
|
|
})
|
|
require.NoError(t, err)
|
|
assert.Equal(t, int64(0), rTaskID)
|
|
}
|