mirror of https://github.com/go-gitea/gitea.git
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
101 lines
2.5 KiB
101 lines
2.5 KiB
// Copyright 2024 The Gitea Authors. All rights reserved. |
|
// SPDX-License-Identifier: MIT |
|
|
|
package oauth2 |
|
|
|
import ( |
|
"testing" |
|
|
|
"code.gitea.io/gitea/models/auth" |
|
"code.gitea.io/gitea/models/unittest" |
|
user_model "code.gitea.io/gitea/models/user" |
|
|
|
"github.com/stretchr/testify/assert" |
|
) |
|
|
|
func TestSource(t *testing.T) { |
|
assert.NoError(t, unittest.PrepareTestDatabase()) |
|
|
|
source := &Source{ |
|
Provider: "fake", |
|
ConfigBase: auth.ConfigBase{ |
|
AuthSource: &auth.Source{ |
|
ID: 12, |
|
Type: auth.OAuth2, |
|
Name: "fake", |
|
IsActive: true, |
|
IsSyncEnabled: true, |
|
}, |
|
}, |
|
} |
|
|
|
user := &user_model.User{ |
|
LoginName: "external", |
|
LoginType: auth.OAuth2, |
|
LoginSource: source.AuthSource.ID, |
|
Name: "test", |
|
Email: "external@example.com", |
|
} |
|
|
|
err := user_model.CreateUser(t.Context(), user, &user_model.Meta{}, &user_model.CreateUserOverwriteOptions{}) |
|
assert.NoError(t, err) |
|
|
|
e := &user_model.ExternalLoginUser{ |
|
ExternalID: "external", |
|
UserID: user.ID, |
|
LoginSourceID: user.LoginSource, |
|
RefreshToken: "valid", |
|
} |
|
err = user_model.LinkExternalToUser(t.Context(), user, e) |
|
assert.NoError(t, err) |
|
|
|
provider, err := createProvider(source.AuthSource.Name, source) |
|
assert.NoError(t, err) |
|
|
|
t.Run("refresh", func(t *testing.T) { |
|
t.Run("valid", func(t *testing.T) { |
|
err := source.refresh(t.Context(), provider, e) |
|
assert.NoError(t, err) |
|
|
|
e := &user_model.ExternalLoginUser{ |
|
ExternalID: e.ExternalID, |
|
LoginSourceID: e.LoginSourceID, |
|
} |
|
|
|
ok, err := user_model.GetExternalLogin(t.Context(), e) |
|
assert.NoError(t, err) |
|
assert.True(t, ok) |
|
assert.Equal(t, "refresh", e.RefreshToken) |
|
assert.Equal(t, "token", e.AccessToken) |
|
|
|
u, err := user_model.GetUserByID(t.Context(), user.ID) |
|
assert.NoError(t, err) |
|
assert.True(t, u.IsActive) |
|
}) |
|
|
|
t.Run("expired", func(t *testing.T) { |
|
err := source.refresh(t.Context(), provider, &user_model.ExternalLoginUser{ |
|
ExternalID: "external", |
|
UserID: user.ID, |
|
LoginSourceID: user.LoginSource, |
|
RefreshToken: "expired", |
|
}) |
|
assert.NoError(t, err) |
|
|
|
e := &user_model.ExternalLoginUser{ |
|
ExternalID: e.ExternalID, |
|
LoginSourceID: e.LoginSourceID, |
|
} |
|
|
|
ok, err := user_model.GetExternalLogin(t.Context(), e) |
|
assert.NoError(t, err) |
|
assert.True(t, ok) |
|
assert.Empty(t, e.RefreshToken) |
|
assert.Empty(t, e.AccessToken) |
|
|
|
u, err := user_model.GetUserByID(t.Context(), user.ID) |
|
assert.NoError(t, err) |
|
assert.False(t, u.IsActive) |
|
}) |
|
}) |
|
}
|
|
|