[usage] Add listing of memberships by UserID

Author: easyCZ

components/usage/pkg/db/team_membership.go

@@ -5,7 +5,10 @@
 package db
 
 import (
+	"context"
+	"fmt"
 	"github.com/google/uuid"
+	"gorm.io/gorm"
 	"time"
 )
 
@@ -36,3 +39,19 @@ const (
 	TeamMembershipRole_Owner  = TeamMembershipRole("owner")
 	TeamMembershipRole_Member = TeamMembershipRole("member")
 )
+
+func ListTeamMembershipsForUserIDs(ctx context.Context, conn *gorm.DB, userIDs []uuid.UUID) ([]TeamMembership, error) {
+	if len(userIDs) == 0 {
+		return nil, nil
+	}
+
+	var memberships []TeamMembership
+	tx := conn.WithContext(ctx).
+		Where("userId IN ?", userIDs).
+		Find(&memberships)
+	if tx.Error != nil {
+		return nil, fmt.Errorf("failed to list team memberships for user IDs: %w", tx.Error)
+	}
+
+	return memberships, nil
+}

components/usage/pkg/db/team_membership_test.go

@@ -5,6 +5,7 @@
 package db_test
 
 import (
+	"context"
 	"github.com/gitpod-io/gitpod/usage/pkg/db"
 	"github.com/google/uuid"
 	"github.com/stretchr/testify/require"
@@ -30,3 +31,70 @@ func TestTeamMembership_WriteRead(t *testing.T) {
 	require.NoError(t, tx.Error)
 	require.Equal(t, membership, read)
 }
+
+func TestListTeamMembershipsForUserIDs(t *testing.T) {
+	conn := db.ConnectForTests(t)
+	userWithTwoTeams := uuid.New()
+	memberships := []db.TeamMembership{
+		{
+			ID:     uuid.New(),
+			TeamID: uuid.New(),
+			UserID: uuid.New(),
+			Role:   db.TeamMembershipRole_Member,
+		},
+		{
+			ID:     uuid.New(),
+			TeamID: uuid.New(),
+			UserID: userWithTwoTeams,
+			Role:   db.TeamMembershipRole_Member,
+		},
+		{
+			ID:     uuid.New(),
+			TeamID: uuid.New(),
+			UserID: userWithTwoTeams,
+			Role:   db.TeamMembershipRole_Member,
+		},
+	}
+
+	tx := conn.Create(&memberships)
+	require.NoError(t, tx.Error)
+
+	for _, scenario := range []struct {
+		Name     string
+		UserIDs  []uuid.UUID
+		Expected int
+	}{
+		{
+			Name:     "no user ids return empty",
+			UserIDs:  nil,
+			Expected: 0,
+		},
+		{
+			Name:     "no matching user IDs returns empty",
+			UserIDs:  []uuid.UUID{uuid.New()},
+			Expected: 0,
+		},
+		{
+			Name:     "one matching, and one not returns only one record",
+			UserIDs:  []uuid.UUID{memberships[0].UserID, uuid.New()},
+			Expected: 1,
+		},
+		{
+			Name:     "all matching returns all records (with duplicate userID in query)",
+			UserIDs:  []uuid.UUID{memberships[0].UserID, memberships[1].UserID, memberships[2].UserID},
+			Expected: 3,
+		},
+		{
+			Name:     "one ID with multiple memberships returns all membership for the ID",
+			UserIDs:  []uuid.UUID{userWithTwoTeams},
+			Expected: 2,
+		},
+	} {
+		t.Run(scenario.Name, func(t *testing.T) {
+			retrieved, err := db.ListTeamMembershipsForUserIDs(context.Background(), conn, scenario.UserIDs)
+			require.NoError(t, err)
+
+			require.Len(t, retrieved, scenario.Expected)
+		})
+	}
+}