add test

Author: 6543

models/user/search_test.go

@@ -0,0 +1,66 @@
+// Copyright 2021 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"
+	"code.gitea.io/gitea/models/user"
+
+	"github.com/stretchr/testify/assert"
+	"xorm.io/builder"
+)
+
+func TestBuildCanSeeUserCondition(t *testing.T) {
+	assert.NoError(t, unittest.PrepareTestDatabase())
+
+	getIDs := func(cond builder.Cond) (ids []int64) {
+		db.GetEngine(db.DefaultContext).Select("id").Table(`user`).
+			Where(builder.Eq{"is_active": true}.And(cond)).Asc("id").Find(&ids)
+		return ids
+	}
+
+	getUser := func(t *testing.T, id int64) *user.User {
+		user, err := user.GetUserByID(db.DefaultContext, id)
+		assert.NoError(t, err)
+		if !assert.NotNil(t, user) {
+			t.FailNow()
+		}
+		return user
+	}
+
+	// no login
+	cond := user.BuildCanSeeUserCondition(nil)
+	assert.NotNil(t, cond)
+	ids := getIDs(cond)
+	assert.Len(t, ids, 24)
+	assert.EqualValues(t, []int64([]int64{1, 2, 4, 5, 8, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 24, 27, 28, 29, 30, 32, 34}), ids)
+
+	// normal user
+	cond = user.BuildCanSeeUserCondition(getUser(t, 5))
+	ids = getIDs(cond)
+	assert.Len(t, ids, 28)
+	assert.EqualValues(t, []int64{1, 2, 4, 5, 8, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 27, 28, 29, 30, 32, 33, 34, 36}, ids)
+
+	// admin
+	cond = user.BuildCanSeeUserCondition(getUser(t, 1))
+	assert.Nil(t, cond)
+	ids = getIDs(cond)
+	assert.Len(t, ids, 30)
+	assert.EqualValues(t, []int64{1, 2, 4, 5, 8, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36}, ids)
+
+	// private user
+	cond = user.BuildCanSeeUserCondition(getUser(t, 31))
+	ids = getIDs(cond)
+	assert.Len(t, ids, 28)
+	assert.EqualValues(t, []int64{1, 2, 4, 5, 8, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 24, 27, 28, 29, 30, 31, 32, 33, 34, 36}, ids)
+
+	// restricted user
+	cond = user.BuildCanSeeUserCondition(getUser(t, 29))
+	ids = getIDs(cond)
+	assert.Len(t, ids, 2)
+	assert.EqualValues(t, []int64{17, 29}, ids)
+}