From b2a0ce6b530f7ef5adc15f898b7b3c7796564837 Mon Sep 17 00:00:00 2001 From: Joe Chen Date: Wed, 15 Nov 2023 22:26:20 -0500 Subject: [PATCH] orgsHasMember --- internal/db/organizations_test.go | 14 ++++++++++++++ 1 file changed, 14 insertions(+) diff --git a/internal/db/organizations_test.go b/internal/db/organizations_test.go index 1aefe4969..8e9876e8a 100644 --- a/internal/db/organizations_test.go +++ b/internal/db/organizations_test.go @@ -48,6 +48,7 @@ func TestOrganizations(t *testing.T) { {"DeleteByID", orgsDeleteByID}, {"AddMember", orgsAddMember}, {"RemoveMember", orgsRemoveMember}, + {"HasMember", orgsHasMember}, } { t.Run(tc.name, func(t *testing.T) { t.Cleanup(func() { @@ -551,3 +552,16 @@ func orgsRemoveMember(t *testing.T, db *organizations) { require.NoError(t, err) assert.Equal(t, 1, gotOrg.NumMembers) } + +func orgsHasMember(t *testing.T, db *organizations) { + ctx := context.Background() + + got, _ := db.HasMember(ctx, 1, 1) + assert.False(t, got) + + err := db.AddMember(ctx, 1, 1) + require.NoError(t, err) + + got, _ = db.HasMember(ctx, 1, 1) + assert.True(t, got) +}