orgsSetMemberVisibility

jc/db-migrate-orgs
Joe Chen 2023-11-16 22:55:33 -05:00
parent 22526be6cf
commit 34e59c21f0
No known key found for this signature in database
GPG Key ID: 0BDE5280C552FF60
1 changed files with 25 additions and 0 deletions

View File

@ -51,6 +51,7 @@ func TestOrganizations(t *testing.T) {
{"HasMember", orgsHasMember},
{"ListMembers", orgsListMembers},
{"IsOwnedBy", orgsIsOwnedBy},
{"SetMemberVisibility", orgsSetMemberVisibility},
} {
t.Run(tc.name, func(t *testing.T) {
t.Cleanup(func() {
@ -623,3 +624,27 @@ func orgsIsOwnedBy(t *testing.T, db *organizations) {
got = db.IsOwnedBy(ctx, org1.ID, cindy.ID)
assert.False(t, got)
}
func orgsSetMemberVisibility(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)
tempPictureAvatarUploadPath := filepath.Join(os.TempDir(), "orgsListMembers-tempPictureAvatarUploadPath")
conf.SetMockPicture(t, conf.PictureOpts{AvatarUploadPath: tempPictureAvatarUploadPath})
org1, err := db.Create(ctx, "org1", alice.ID, CreateOrganizationOptions{})
require.NoError(t, err)
got, err := db.List(ctx, ListOrganizationsOptions{MemberID: alice.ID})
require.NoError(t, err)
assert.Len(t, got, 0)
err = db.SetMemberVisibility(ctx, org1.ID, alice.ID, true)
require.NoError(t, err)
got, err = db.List(ctx, ListOrganizationsOptions{MemberID: alice.ID})
require.NoError(t, err)
assert.Len(t, got, 1)
}