Merge pull request go-auth0#221 from mattoddie/feature/google-apps-scopes

Add scope getter/setter to google-apps connection options

Author: Yvo

management/connection.go

@@ -711,6 +711,7 @@ type ConnectionOptionsGoogleApps struct {
 	ClientID     *string `json:"client_id,omitempty"`
 	ClientSecret *string `json:"client_secret,omitempty"`
 	Domain       *string `json:"domain,omitempty"`
+	TenantDomain *string `json:"tenant_domain,omitempty"`
 
 	EnableUsersAPI  *bool `json:"api_enable_users,omitempty"`
 	BasicProfile    *bool `json:"basic_profile,omitempty" scope:"basic_profile"`
@@ -727,6 +728,14 @@ type ConnectionOptionsGoogleApps struct {
 	LogoURL       *string       `json:"icon_url,omitempty"`
 }
 
+func (c *ConnectionOptionsGoogleApps) Scopes() []string {
+	return tag.Scopes(c)
+}
+
+func (c *ConnectionOptionsGoogleApps) SetScopes(enable bool, scopes ...string) {
+	tag.SetScopes(c, enable, scopes...)
+}
+
 type ConnectionManager struct {
 	*Management
 }

management/connection_test.go

@@ -153,6 +153,7 @@ func TestConnection(t *testing.T) {
 		t.Logf("%v\n", cs)
 	})
 }
+
 func TestConnectionOptions(t *testing.T) {
 
 	t.Run("GoogleOAuth2", func(t *testing.T) {
@@ -222,6 +223,72 @@ func TestConnectionOptions(t *testing.T) {
 		t.Logf("%s\n", g)
 	})
 
+	t.Run("GoogleApps", func(t *testing.T) {
+		g := &Connection{
+			Name:     auth0.Stringf("Test-Connection-%d", time.Now().Unix()),
+			Strategy: auth0.String("google-apps"),
+			Options: &ConnectionOptionsGoogleApps{
+				Domain:          auth0.String("example.com"),
+				TenantDomain:    auth0.String("example.com"),
+				BasicProfile:    auth0.Bool(true),
+				ExtendedProfile: auth0.Bool(true),
+				Groups:          auth0.Bool(true),
+			},
+		}
+
+		defer func() {
+			m.Connection.Delete(g.GetID())
+			assertDeleted(t, g)
+
+		}()
+
+		err := m.Connection.Create(g)
+		if err != nil {
+			t.Fatal(err)
+		}
+		c, err := m.Connection.Read(g.GetID())
+		if err != nil {
+			t.Fatal(err)
+		}
+		o, ok := c.Options.(*ConnectionOptionsGoogleApps)
+		if !ok {
+			t.Fatalf("unexpected type %T", o)
+		}
+		expect.Expect(t, o.GetBasicProfile(), true)
+		expect.Expect(t, o.GetExtendedProfile(), true)
+		expect.Expect(t, o.GetGroups(), true)
+		expect.Expect(t, o.GetAdmin(), false)
+		expect.Expect(t, o.Scopes(), []string{"basic_profile", "ext_profile", "ext_groups"})
+
+		o.NonPersistentAttrs = &[]string{"gender", "ethnicity", "favorite_color"}
+		err = m.Connection.Update(g.GetID(), &Connection{
+			Options: o,
+		})
+		if err != nil {
+			t.Fatal(err)
+		}
+		c, _ = m.Connection.Read(g.GetID())
+		o, ok = c.Options.(*ConnectionOptionsGoogleApps)
+		expect.Expect(t, o.GetNonPersistentAttrs(), []string{"gender", "ethnicity", "favorite_color"})
+
+		o.NonPersistentAttrs = &[]string{""}
+		err = m.Connection.Update(g.GetID(), &Connection{
+			Options: o,
+		})
+		if err != nil {
+			log.Fatal(err)
+		}
+
+		c, err = m.Connection.Read(g.GetID())
+		if err != nil {
+			log.Fatal(err)
+		}
+		o, ok = c.Options.(*ConnectionOptionsGoogleApps)
+		expect.Expect(t, o.GetNonPersistentAttrs(), []string{""})
+
+		t.Logf("%s\n", g)
+	})
+
 	t.Run("OIDC", func(t *testing.T) {
 		o := &ConnectionOptionsOIDC{}
 		expect.Expect(t, len(o.Scopes()), 0)