Issue 70: Demonstration of error in InstanceGroups

plugin/authorizer_client_gcp.go

@@ -19,28 +19,31 @@ type gcpClient struct {
 	iamSvc     *iam.Service
 }
 
+func genExtractZonesFn(igz map[string][]string, boundInstanceGroups []string) func(*compute.InstanceGroupAggregatedList) error {
+	return func(l *compute.InstanceGroupAggregatedList) error {
+		for k, v := range l.Items {
+			zone, err := zoneFromSelfLink(k)
+			if err != nil {
+				return err
+			}
+			for _, g := range v.InstanceGroups {
+				if strutil.StrListContains(boundInstanceGroups, g.Name) {
+					igz[zone] = append(igz[zone], g.Name)
+				}
+			}
+		}
+		return nil
+	}
+}
+
 func (c *gcpClient) InstanceGroups(ctx context.Context, project string, boundInstanceGroups []string) (map[string][]string, error) {
 	// map of zone names to a slice of instance group names in that zone.
 	igz := make(map[string][]string)
 
 	if err := c.computeSvc.InstanceGroups.
 		AggregatedList(project).
 		Fields("items/*/instanceGroups/name").
-		Pages(ctx, func(l *compute.InstanceGroupAggregatedList) error {
-			for k, v := range l.Items {
-				zone, err := zoneFromSelfLink(k)
-				if err != nil {
-					return err
-				}
-
-				for _, g := range v.InstanceGroups {
-					if strutil.StrListContains(boundInstanceGroups, g.Name) {
-						igz[zone] = append(igz[zone], g.Name)
-					}
-				}
-			}
-			return nil
-		}); err != nil {
+		Pages(ctx, genExtractZonesFn(igz, boundInstanceGroups)); err != nil {
 		return nil, err
 	}
 

plugin/authorizer_client_gcp_test.go

@@ -0,0 +1,34 @@
+package gcpauth
+
+import (
+	"testing"
+	"google.golang.org/api/compute/v1"
+)
+
+func TestInstanceGroups(t *testing.T) {
+
+	igz := make(map[string][]string)
+	boundInstanceGroups := []string{"foo-us1-baz-group"}
+
+	extractZonesFn := genExtractZonesFn(igz, boundInstanceGroups)
+
+
+	InstanceGroupAggregatedList := compute.InstanceGroupAggregatedList{
+		Items: map[string]compute.InstanceGroupsScopedList{
+			"zones/us-central1-c": compute.InstanceGroupsScopedList{
+				InstanceGroups: []*compute.InstanceGroup{&compute.InstanceGroup{Name: "foo-us1-bar-group"}, &compute.InstanceGroup{Name: "foo-us1-baz-group"}},
+			},
+			"regions/us-central1": compute.InstanceGroupsScopedList{},
+		},
+	}
+
+	err := extractZonesFn(&InstanceGroupAggregatedList)
+
+	if err != nil {
+		t.Fatal("error not expected")
+	}
+
+	if igz["us-central1-c"][0] != "foo-us1-baz-group" {
+		t.Fatal("expected value not found")
+	}
+}