mirror of
https://github.com/hanzoai/forge.git
synced 2026-10-30 15:00:39 +00:00
zeekayandHanzo Dev
9a07b4d089
2978 files. The fork is 2026-07-15 (rebrand: gitea.dev -> github.com/hanzoai/git), so 2026 is the year, and "Copyright 2026 Hanzo AI, Inc." was already the form used by the handful of files that had ours. Shape is now: // Copyright 2026 Hanzo AI, Inc. All rights reserved. // Copyright 2015 The Gogs Authors. All rights reserved. // Copyright 2019 The Gitea Authors. All rights reserved. // SPDX-License-Identifier: MIT Ours on top, theirs retained verbatim. Retained because MIT requires it, and separately because they are true — those authors did write that code. Not renamed to "The Git Authors": Git is a different project under a different license, and putting its name here would attribute this code to people who have nothing to do with it. Adding our line is the fix; overwriting theirs would have been both a licence violation and a false statement. gofmt still reports import-order drift in some of these files. It predates this change (verified against a stash) and is left alone rather than burying a one-line-per-file legal change under 3000 files of unrelated reordering. Build clean. Co-authored-by: Hanzo Dev <[email protected]>
111 lines
3.0 KiB
Go
111 lines
3.0 KiB
Go
// Copyright 2026 Hanzo AI, Inc. All rights reserved.
|
|
// Copyright 2023 The Gitea Authors. All rights reserved.
|
|
// SPDX-License-Identifier: MIT
|
|
|
|
package auth
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
auth_model "github.com/hanzoai/git/models/auth"
|
|
"github.com/hanzoai/git/models/unittest"
|
|
"github.com/hanzoai/git/modules/timeutil"
|
|
"github.com/hanzoai/git/modules/util"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestCheckAuthToken(t *testing.T) {
|
|
assert.NoError(t, unittest.PrepareTestDatabase())
|
|
|
|
t.Run("Empty", func(t *testing.T) {
|
|
token, err := CheckAuthToken(t.Context(), "")
|
|
assert.NoError(t, err)
|
|
assert.Nil(t, token)
|
|
})
|
|
|
|
t.Run("InvalidFormat", func(t *testing.T) {
|
|
token, err := CheckAuthToken(t.Context(), "dummy")
|
|
assert.ErrorIs(t, err, ErrAuthTokenInvalidFormat)
|
|
assert.Nil(t, token)
|
|
})
|
|
|
|
t.Run("NotFound", func(t *testing.T) {
|
|
token, err := CheckAuthToken(t.Context(), "notexists:dummy")
|
|
assert.ErrorIs(t, err, ErrAuthTokenExpired)
|
|
assert.Nil(t, token)
|
|
})
|
|
|
|
t.Run("Expired", func(t *testing.T) {
|
|
timeutil.MockSet(time.Date(2023, 1, 1, 0, 0, 0, 0, time.UTC))
|
|
|
|
at, token, err := CreateAuthTokenForUserID(t.Context(), 2)
|
|
assert.NoError(t, err)
|
|
assert.NotNil(t, at)
|
|
assert.NotEmpty(t, token)
|
|
|
|
timeutil.MockUnset()
|
|
|
|
at2, err := CheckAuthToken(t.Context(), at.ID+":"+token)
|
|
assert.ErrorIs(t, err, ErrAuthTokenExpired)
|
|
assert.Nil(t, at2)
|
|
|
|
assert.NoError(t, auth_model.DeleteAuthTokenByID(t.Context(), at.ID))
|
|
})
|
|
|
|
t.Run("InvalidHash", func(t *testing.T) {
|
|
at, token, err := CreateAuthTokenForUserID(t.Context(), 2)
|
|
assert.NoError(t, err)
|
|
assert.NotNil(t, at)
|
|
assert.NotEmpty(t, token)
|
|
|
|
at2, err := CheckAuthToken(t.Context(), at.ID+":"+token+"dummy")
|
|
assert.ErrorIs(t, err, ErrAuthTokenInvalidHash)
|
|
assert.Nil(t, at2)
|
|
|
|
// a hash mismatch signals a compromised token, which must be revoked
|
|
_, err = auth_model.GetAuthTokenByID(t.Context(), at.ID)
|
|
assert.ErrorIs(t, err, util.ErrNotExist)
|
|
})
|
|
|
|
t.Run("Valid", func(t *testing.T) {
|
|
at, token, err := CreateAuthTokenForUserID(t.Context(), 2)
|
|
assert.NoError(t, err)
|
|
assert.NotNil(t, at)
|
|
assert.NotEmpty(t, token)
|
|
|
|
at2, err := CheckAuthToken(t.Context(), at.ID+":"+token)
|
|
assert.NoError(t, err)
|
|
assert.NotNil(t, at2)
|
|
|
|
assert.NoError(t, auth_model.DeleteAuthTokenByID(t.Context(), at.ID))
|
|
})
|
|
}
|
|
|
|
func TestRegenerateAuthToken(t *testing.T) {
|
|
assert.NoError(t, unittest.PrepareTestDatabase())
|
|
|
|
timeutil.MockSet(time.Date(2023, 1, 1, 0, 0, 0, 0, time.UTC))
|
|
defer timeutil.MockUnset()
|
|
|
|
at, token, err := CreateAuthTokenForUserID(t.Context(), 2)
|
|
assert.NoError(t, err)
|
|
assert.NotNil(t, at)
|
|
assert.NotEmpty(t, token)
|
|
|
|
timeutil.MockSet(time.Date(2023, 1, 1, 0, 0, 1, 0, time.UTC))
|
|
|
|
at2, token2, err := RegenerateAuthToken(t.Context(), at)
|
|
assert.NoError(t, err)
|
|
assert.NotNil(t, at2)
|
|
assert.NotEmpty(t, token2)
|
|
|
|
assert.Equal(t, at.ID, at2.ID)
|
|
assert.Equal(t, at.UserID, at2.UserID)
|
|
assert.NotEqual(t, token, token2)
|
|
assert.NotEqual(t, at.ExpiresUnix, at2.ExpiresUnix)
|
|
|
|
assert.NoError(t, auth_model.DeleteAuthTokenByID(t.Context(), at.ID))
|
|
}
|