From 34e59c21f071662e2357a6ad26c8635f71fbf1c1 Mon Sep 17 00:00:00 2001
From: Joe Chen <jc@unknwon.io>
Date: Thu, 16 Nov 2023 22:55:33 -0500
Subject: [PATCH] orgsSetMemberVisibility
---
internal/db/organizations_test.go | 25 +++++++++++++++++++++++++
1 file changed, 25 insertions(+)
diff --git a/internal/db/organizations_test.go b/internal/db/organizations_test.go
index 01c012623..6b6e9e356 100644
--- a/internal/db/organizations_test.go
+++ b/internal/db/organizations_test.go
@@ -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)
+}