orgsHasMember

jc/db-migrate-orgs
Joe Chen 2023-11-15 22:26:20 -05:00
parent 7d4c5b47a2
commit b2a0ce6b53
No known key found for this signature in database
GPG Key ID: 0BDE5280C552FF60
1 changed files with 14 additions and 0 deletions

View File

@ -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)
}