mirror of https://github.com/go-gitea/gitea.git
62 lines
1.6 KiB
Go
62 lines
1.6 KiB
Go
// Copyright 2025 The Gitea Authors. All rights reserved.
|
|
// SPDX-License-Identifier: MIT
|
|
|
|
package user_test
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"code.gitea.io/gitea/models/db"
|
|
"code.gitea.io/gitea/models/unittest"
|
|
user_model "code.gitea.io/gitea/models/user"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestGetBadgeUsers(t *testing.T) {
|
|
assert.NoError(t, unittest.PrepareTestDatabase())
|
|
|
|
// Create a test badge
|
|
badge := &user_model.Badge{
|
|
Slug: "test-badge",
|
|
Description: "Test Badge",
|
|
ImageURL: "test.png",
|
|
}
|
|
assert.NoError(t, user_model.CreateBadge(db.DefaultContext, badge))
|
|
|
|
// Create test users and assign badges
|
|
user1 := unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: 1})
|
|
user2 := unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: 2})
|
|
|
|
assert.NoError(t, user_model.AddUserBadge(db.DefaultContext, user1, badge))
|
|
assert.NoError(t, user_model.AddUserBadge(db.DefaultContext, user2, badge))
|
|
|
|
// Test getting users with pagination
|
|
opts := &user_model.GetBadgeUsersOptions{
|
|
Badge: badge,
|
|
ListOptions: db.ListOptions{
|
|
Page: 1,
|
|
PageSize: 1,
|
|
},
|
|
}
|
|
|
|
users, count, err := user_model.GetBadgeUsers(db.DefaultContext, opts)
|
|
assert.NoError(t, err)
|
|
assert.EqualValues(t, 2, count)
|
|
assert.Len(t, users, 1)
|
|
|
|
// Test second page
|
|
opts.Page = 2
|
|
users, count, err = user_model.GetBadgeUsers(db.DefaultContext, opts)
|
|
assert.NoError(t, err)
|
|
assert.EqualValues(t, 2, count)
|
|
assert.Len(t, users, 1)
|
|
|
|
// Test with non-existent badge
|
|
opts.Badge = &user_model.Badge{Slug: "non-existent"}
|
|
users, count, err = user_model.GetBadgeUsers(db.DefaultContext, opts)
|
|
assert.NoError(t, err)
|
|
assert.EqualValues(t, 0, count)
|
|
assert.Empty(t, users)
|
|
}
|