mirror of
https://codeberg.org/forgejo/forgejo
synced 2024-06-27 20:30:33 +02:00
73776d6195
- Follow up for: #540, #802 - Add API routes for user blocking from user and organization perspective. - The new routes have integration testing. - The new model functions have unit tests. - Actually quite boring to write and to read this pull request. (cherry picked from commitf3afaf15c7) (cherry picked from commit6d754db3e5) (cherry picked from commitd0fc8bc9d3) (cherry picked from commit9a53b0d1a0) (cherry picked from commit44a2a4fd48) (cherry picked from commit182025db9c) (cherry picked from commit558a35963e)
76 lines
2.4 KiB
Go
76 lines
2.4 KiB
Go
// Copyright 2023 The Forgejo 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 TestIsBlocked(t *testing.T) {
|
|
assert.NoError(t, unittest.PrepareTestDatabase())
|
|
assert.True(t, user_model.IsBlocked(db.DefaultContext, 4, 1))
|
|
|
|
// Simple test cases to ensure the function can also respond with false.
|
|
assert.False(t, user_model.IsBlocked(db.DefaultContext, 1, 1))
|
|
assert.False(t, user_model.IsBlocked(db.DefaultContext, 3, 2))
|
|
}
|
|
|
|
func TestIsBlockedMultiple(t *testing.T) {
|
|
assert.NoError(t, unittest.PrepareTestDatabase())
|
|
assert.True(t, user_model.IsBlockedMultiple(db.DefaultContext, []int64{4}, 1))
|
|
assert.True(t, user_model.IsBlockedMultiple(db.DefaultContext, []int64{4, 3, 4, 5}, 1))
|
|
|
|
// Simple test cases to ensure the function can also respond with false.
|
|
assert.False(t, user_model.IsBlockedMultiple(db.DefaultContext, []int64{1}, 1))
|
|
assert.False(t, user_model.IsBlockedMultiple(db.DefaultContext, []int64{3, 4, 1}, 2))
|
|
}
|
|
|
|
func TestUnblockUser(t *testing.T) {
|
|
assert.NoError(t, unittest.PrepareTestDatabase())
|
|
assert.True(t, user_model.IsBlocked(db.DefaultContext, 4, 1))
|
|
|
|
assert.NoError(t, user_model.UnblockUser(db.DefaultContext, 4, 1))
|
|
|
|
// Simple test cases to ensure the function can also respond with false.
|
|
assert.False(t, user_model.IsBlocked(db.DefaultContext, 4, 1))
|
|
}
|
|
|
|
func TestListBlockedUsers(t *testing.T) {
|
|
assert.NoError(t, unittest.PrepareTestDatabase())
|
|
|
|
blockedUsers, err := user_model.ListBlockedUsers(db.DefaultContext, 4, db.ListOptions{})
|
|
assert.NoError(t, err)
|
|
if assert.Len(t, blockedUsers, 1) {
|
|
assert.EqualValues(t, 1, blockedUsers[0].ID)
|
|
}
|
|
}
|
|
|
|
func TestListBlockedByUsersID(t *testing.T) {
|
|
assert.NoError(t, unittest.PrepareTestDatabase())
|
|
|
|
blockedByUserIDs, err := user_model.ListBlockedByUsersID(db.DefaultContext, 1)
|
|
assert.NoError(t, err)
|
|
if assert.Len(t, blockedByUserIDs, 1) {
|
|
assert.EqualValues(t, 4, blockedByUserIDs[0])
|
|
}
|
|
}
|
|
|
|
func TestCountBlockedUsers(t *testing.T) {
|
|
assert.NoError(t, unittest.PrepareTestDatabase())
|
|
|
|
count, err := user_model.CountBlockedUsers(db.DefaultContext, 4)
|
|
assert.NoError(t, err)
|
|
assert.EqualValues(t, 1, count)
|
|
|
|
count, err = user_model.CountBlockedUsers(db.DefaultContext, 1)
|
|
assert.NoError(t, err)
|
|
assert.EqualValues(t, 0, count)
|
|
}
|