fix pr review requests for teams

mercimat · mercimat · commit a612f06deec8 · 2021-05-24T17:00:25.000+02:00
diff --git a/api/queries_pr.go b/api/queries_pr.go
@@ -151,17 +151,27 @@ type PullRequestFile struct {
 type ReviewRequests struct {
 	Nodes []struct {
 		RequestedReviewer struct {
-			TypeName string `json:"__typename"`
-			Login    string `json:"login"`
-			Name     string `json:"name"`
+			TypeName     string `json:"__typename"`
+			Login        string `json:"login"`
+			Name         string `json:"name"`
+			Slug         string `json:"slug"`
+			Organization struct {
+				Login string `json:"login"`
+			}
 		}
 	}
 }
 
+const teamTypeName = "Team"
+
 func (r ReviewRequests) Logins() []string {
 	logins := make([]string, len(r.Nodes))
 	for i, a := range r.Nodes {
-		logins[i] = a.RequestedReviewer.Login
+		if a.RequestedReviewer.TypeName == teamTypeName {
+			logins[i] = a.RequestedReviewer.Organization.Login + "/" + a.RequestedReviewer.Slug
+		} else {
+			logins[i] = a.RequestedReviewer.Login
+		}
 	}
 	return logins
 }
diff --git a/api/queries_pr_test.go b/api/queries_pr_test.go
@@ -1,6 +1,7 @@
 package api
 
 import (
+	"encoding/json"
 	"reflect"
 	"testing"
 
@@ -158,3 +159,88 @@ func Test_determinePullRequestFeatures(t *testing.T) {
 		})
 	}
 }
+
+func Test_Logins(t *testing.T) {
+	rr := ReviewRequests{}
+	var tests = []struct {
+		name             string
+		requestedReviews string
+		want             []string
+	}{
+		{
+			name: "no requested reviewers",
+			requestedReviews: `{"nodes": []}`,
+			want: []string{},
+		},
+		{
+			name: "user",
+			requestedReviews: `{"nodes": [
+				{
+					"requestedreviewer": {
+						"__typename": "User", "login": "testuser"
+					}
+				}
+			]}`,
+			want: []string{"testuser"},
+		},
+		{
+			name: "team",
+			requestedReviews: `{"nodes": [
+				{
+					"requestedreviewer": {
+						"__typename": "Team",
+						"name": "Test Team",
+						"slug": "test-team",
+						"organization": {"login": "myorg"}
+					}
+				}
+			]}`,
+			want: []string{"myorg/test-team"},
+		},
+		{
+			name: "multiple users and teams",
+			requestedReviews: `{"nodes": [
+				{
+					"requestedreviewer": {
+						"__typename": "User", "login": "user1"
+					}
+				},
+				{
+					"requestedreviewer": {
+						"__typename": "User", "login": "user2"
+					}
+				},
+				{
+					"requestedreviewer": {
+						"__typename": "Team",
+						"name": "Test Team",
+						"slug": "test-team",
+						"organization": {"login": "myorg"}
+					}
+				},
+				{
+					"requestedreviewer": {
+						"__typename": "Team",
+						"name": "Dev Team",
+						"slug": "dev-team",
+						"organization": {"login": "myorg"}
+					}
+				}
+			]}`,
+			want: []string{"user1", "user2", "myorg/test-team", "myorg/dev-team"},
+		},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			err := json.Unmarshal([]byte(tt.requestedReviews), &rr)
+			if err != nil {
+				t.Fatalf("Failed to unmarshal json string as ReviewRequests: %v", tt.requestedReviews)
+			}
+			got := rr.Logins()
+			if !reflect.DeepEqual(got, tt.want) {
+				t.Fatalf("Unexpected results: expected %v but got %v", tt.want, got)
+			}
+		})
+	}
+}
diff --git a/api/query_builder.go b/api/query_builder.go
@@ -41,7 +41,11 @@ var prReviewRequests = shortenQuery(`
 			requestedReviewer {
 				__typename,
 				...on User{login},
-				...on Team{name}
+				...on Team{
+					organization{login}
+					name,
+					slug
+				}
 			}
 		}
 	}

Original file line number	Diff line number	Diff line change
@@ -41,7 +41,11 @@ var prReviewRequests = shortenQuery(`
`41`	`41`	`requestedReviewer {`
`42`	`42`	`__typename,`
`43`	`43`	`...on User{login},`
`44`		`- ...on Team{name}`
	`44`	`+ ...on Team{`
	`45`	`+ organization{login}`
	`46`	`+ name,`
	`47`	`+ slug`
	`48`	`+ }`
`45`	`49`	`}`
`46`	`50`	`}`
`47`	`51`	`}`