2024-02-18 12:39:04 -05:00
|
|
|
// Copyright 2024 The Gitea Authors. All rights reserved.
|
|
|
|
// SPDX-License-Identifier: MIT
|
|
|
|
|
|
|
|
package generate
|
|
|
|
|
|
|
|
import (
|
|
|
|
"encoding/base64"
|
|
|
|
"strings"
|
|
|
|
"testing"
|
|
|
|
|
|
|
|
"github.com/stretchr/testify/assert"
|
2024-07-30 14:41:10 -05:00
|
|
|
"github.com/stretchr/testify/require"
|
2024-02-18 12:39:04 -05:00
|
|
|
)
|
|
|
|
|
|
|
|
func TestDecodeJwtSecret(t *testing.T) {
|
|
|
|
_, err := DecodeJwtSecret("abcd")
|
2024-07-30 14:41:10 -05:00
|
|
|
require.ErrorContains(t, err, "invalid base64 decoded length")
|
2024-02-18 12:39:04 -05:00
|
|
|
_, err = DecodeJwtSecret(strings.Repeat("a", 64))
|
2024-07-30 14:41:10 -05:00
|
|
|
require.ErrorContains(t, err, "invalid base64 decoded length")
|
2024-02-18 12:39:04 -05:00
|
|
|
|
|
|
|
str32 := strings.Repeat("x", 32)
|
|
|
|
encoded32 := base64.RawURLEncoding.EncodeToString([]byte(str32))
|
|
|
|
decoded32, err := DecodeJwtSecret(encoded32)
|
2024-07-30 14:41:10 -05:00
|
|
|
require.NoError(t, err)
|
2024-02-18 12:39:04 -05:00
|
|
|
assert.Equal(t, str32, string(decoded32))
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestNewJwtSecret(t *testing.T) {
|
|
|
|
secret, encoded, err := NewJwtSecret()
|
2024-07-30 14:41:10 -05:00
|
|
|
require.NoError(t, err)
|
2024-02-18 12:39:04 -05:00
|
|
|
assert.Len(t, secret, 32)
|
|
|
|
decoded, err := DecodeJwtSecret(encoded)
|
2024-07-30 14:41:10 -05:00
|
|
|
require.NoError(t, err)
|
2024-02-18 12:39:04 -05:00
|
|
|
assert.Equal(t, secret, decoded)
|
|
|
|
}
|