USEFULL
/
gogs
mirror of https://github.com/gogs/gogs.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

orgsAddMember

jc/db-migrate-orgs
Joe Chen 2023-11-13 22:20:03 -05:00
parent bf92bcdf66
commit fce3e65b2c
No known key found for this signature in database
GPG Key ID: 0BDE5280C552FF60
1 changed files with 36 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

36
internal/db/organizations_test.go
View File

@ -46,6 +46,7 @@ func TestOrganizations(t *testing.T) {
{"CountByUser", orgsCountByUser},
{"Count", orgsCount},
{"DeleteByID", orgsDeleteByID},
{"AddMember", orgsAddMember},
} {
t.Run(tc.name, func(t *testing.T) {
t.Cleanup(func() {
@ -397,3 +398,38 @@ func orgsDeleteByID(t *testing.T, db *organizations) {
wantErr := ErrOrganizationNotExist{errutil.Args{"name": org2.Name}}
assert.Equal(t, wantErr, err)
}
func orgsAddMember(t *testing.T, db *organizations) {
ctx := context.Background()
usersStore := NewUsersStore(db.DB)
alice, err := usersStore.Create(ctx, "alice", "alice@example.com", CreateUserOptions{})
require.NoError(t, err)
bob, err := usersStore.Create(ctx, "bob", "bob@exmaple.com", CreateUserOptions{})
require.NoError(t, err)
tempPictureAvatarUploadPath := filepath.Join(os.TempDir(), "orgsAddMember-tempPictureAvatarUploadPath")
conf.SetMockPicture(t, conf.PictureOpts{AvatarUploadPath: tempPictureAvatarUploadPath})
org1, err := db.Create(ctx, "org1", alice.ID, CreateOrganizationOptions{})
require.NoError(t, err)
// Not yet a member
got, err := db.List(ctx, ListOrganizationsOptions{MemberID: bob.ID, IncludePrivateMembers: true})
require.NoError(t, err)
assert.Len(t, got, 0)
// Add member
err = db.AddMember(ctx, org1.ID, bob.ID)
require.NoError(t, err)
// Now a member
got, err = db.List(ctx, ListOrganizationsOptions{MemberID: bob.ID, IncludePrivateMembers: true})
require.NoError(t, err)
assert.Len(t, got, 1)
assert.Equal(t, org1.ID, got[0].ID)
// Add member again shouldn't fail
err = db.AddMember(ctx, org1.ID, bob.ID)
require.NoError(t, err)
}