|  | 
| 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