|
@@ -48,6 +48,7 @@ func TestOrganizations(t *testing.T) {
|
|
{"DeleteByID", orgsDeleteByID},
|
|
{"DeleteByID", orgsDeleteByID},
|
|
{"AddMember", orgsAddMember},
|
|
{"AddMember", orgsAddMember},
|
|
{"RemoveMember", orgsRemoveMember},
|
|
{"RemoveMember", orgsRemoveMember},
|
|
|
|
+ {"HasMember", orgsHasMember},
|
|
} {
|
|
} {
|
|
t.Run(tc.name, func(t *testing.T) {
|
|
t.Run(tc.name, func(t *testing.T) {
|
|
t.Cleanup(func() {
|
|
t.Cleanup(func() {
|
|
@@ -551,3 +552,16 @@ func orgsRemoveMember(t *testing.T, db *organizations) {
|
|
require.NoError(t, err)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, 1, gotOrg.NumMembers)
|
|
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)
|
|
|
|
+}
|