Bläddra i källkod

orgsHasMember

Joe Chen 1 år sedan
förälder
incheckning
b2a0ce6b53
1 ändrade filer med 14 tillägg och 0 borttagningar
  1. 14 0
      internal/db/organizations_test.go

+ 14 - 0
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)
+}