|
4 | 4 | package organization_test
|
5 | 5 |
|
6 | 6 | import (
|
| 7 | + "sort" |
7 | 8 | "testing"
|
8 | 9 |
|
9 | 10 | "code.gitea.io/gitea/models/db"
|
@@ -103,7 +104,7 @@ func TestUser_GetTeams(t *testing.T) {
|
103 | 104 | func TestUser_GetMembers(t *testing.T) {
|
104 | 105 | assert.NoError(t, unittest.PrepareTestDatabase())
|
105 | 106 | org := unittest.AssertExistsAndLoadBean(t, &organization.Organization{ID: 3})
|
106 |
| - members, _, err := org.GetMembers(db.DefaultContext) |
| 107 | + members, _, err := org.GetMembers(db.DefaultContext, &user_model.User{IsAdmin: true}) |
107 | 108 | assert.NoError(t, err)
|
108 | 109 | if assert.Len(t, members, 3) {
|
109 | 110 | assert.Equal(t, int64(2), members[0].ID)
|
@@ -210,37 +211,42 @@ func TestFindOrgs(t *testing.T) {
|
210 | 211 | func TestGetOrgUsersByOrgID(t *testing.T) {
|
211 | 212 | assert.NoError(t, unittest.PrepareTestDatabase())
|
212 | 213 |
|
213 |
| - orgUsers, err := organization.GetOrgUsersByOrgID(db.DefaultContext, &organization.FindOrgMembersOpts{ |
214 |
| - ListOptions: db.ListOptions{}, |
215 |
| - OrgID: 3, |
216 |
| - PublicOnly: false, |
| 214 | + opts := &organization.FindOrgMembersOpts{ |
| 215 | + Doer: &user_model.User{IsAdmin: true}, |
| 216 | + OrgID: 3, |
| 217 | + } |
| 218 | + assert.False(t, opts.PublicOnly()) |
| 219 | + orgUsers, err := organization.GetOrgUsersByOrgID(db.DefaultContext, opts) |
| 220 | + assert.NoError(t, err) |
| 221 | + sort.Slice(orgUsers, func(i, j int) bool { |
| 222 | + return orgUsers[i].ID < orgUsers[j].ID |
217 | 223 | })
|
| 224 | + assert.EqualValues(t, []*organization.OrgUser{{ |
| 225 | + ID: 1, |
| 226 | + OrgID: 3, |
| 227 | + UID: 2, |
| 228 | + IsPublic: true, |
| 229 | + }, { |
| 230 | + ID: 2, |
| 231 | + OrgID: 3, |
| 232 | + UID: 4, |
| 233 | + IsPublic: false, |
| 234 | + }, { |
| 235 | + ID: 9, |
| 236 | + OrgID: 3, |
| 237 | + UID: 28, |
| 238 | + IsPublic: true, |
| 239 | + }}, orgUsers) |
| 240 | + |
| 241 | + opts = &organization.FindOrgMembersOpts{OrgID: 3} |
| 242 | + assert.True(t, opts.PublicOnly()) |
| 243 | + orgUsers, err = organization.GetOrgUsersByOrgID(db.DefaultContext, opts) |
218 | 244 | assert.NoError(t, err)
|
219 |
| - if assert.Len(t, orgUsers, 3) { |
220 |
| - assert.Equal(t, organization.OrgUser{ |
221 |
| - ID: orgUsers[0].ID, |
222 |
| - OrgID: 3, |
223 |
| - UID: 2, |
224 |
| - IsPublic: true, |
225 |
| - }, *orgUsers[0]) |
226 |
| - assert.Equal(t, organization.OrgUser{ |
227 |
| - ID: orgUsers[1].ID, |
228 |
| - OrgID: 3, |
229 |
| - UID: 4, |
230 |
| - IsPublic: false, |
231 |
| - }, *orgUsers[1]) |
232 |
| - assert.Equal(t, organization.OrgUser{ |
233 |
| - ID: orgUsers[2].ID, |
234 |
| - OrgID: 3, |
235 |
| - UID: 28, |
236 |
| - IsPublic: true, |
237 |
| - }, *orgUsers[2]) |
238 |
| - } |
| 245 | + assert.Len(t, orgUsers, 2) |
239 | 246 |
|
240 | 247 | orgUsers, err = organization.GetOrgUsersByOrgID(db.DefaultContext, &organization.FindOrgMembersOpts{
|
241 | 248 | ListOptions: db.ListOptions{},
|
242 | 249 | OrgID: unittest.NonexistentID,
|
243 |
| - PublicOnly: false, |
244 | 250 | })
|
245 | 251 | assert.NoError(t, err)
|
246 | 252 | assert.Len(t, orgUsers, 0)
|
|
0 commit comments