add func for check membership

teamcity/user.go

@@ -3,6 +3,7 @@ package teamcity
 import (
  "fmt"
  "net/http"
+ "strings"
 
  "github.com/dghubble/sling"
 )
@@ -181,3 +182,31 @@ func (s *UserService) groupDeleteByKey(locator Locator, groupKey string) (*Group
  }
  return &out, nil
 }
+
+// IsGroupMemberByID - checks the user's group membership by ID
+func (s *UserService) IsGroupMemberByID(id int, key string) (bool, error) {
+ return s.isGroupMemberByLocator(LocatorID(fmt.Sprint(id)), key)
+}
+
+// IsGroupMemberByUsername - checks the user's group membership by Username
+func (s *UserService) IsGroupMemberByUsername(username, key string) (bool, error) {
+ return s.isGroupMemberByLocator(LocatorUsername(username), key)
+}
+
+// IsGroupMemberByName - checks the user's group membership by Name
+func (s *UserService) IsGroupMemberByName(name, key string) (bool, error) {
+ return s.isGroupMemberByLocator(LocatorName(name), key)
+}
+
+func (s *UserService) isGroupMemberByLocator(locator Locator, key string) (bool, error) {
+ var out Group
+ err := s.restHelper.get(fmt.Sprintf("%s/groups/%s", locator, LocatorKey(key)), &out, "User")
+ if err != nil {
+ strErr := err.Error()
+ if strings.Contains(strErr, "status code: 404") {
+ return false, nil
+ }
+ return false, err
+ }
+ return true, nil
+}

teamcity/user_test.go

@@ -152,6 +152,26 @@ func TestUser_Group(t *testing.T) {
 
 }
 
+func TestUser_Member(t *testing.T) {
+ client := setup()
+ admin, err := client.Users.GetByUsername("admin")
+ require.NoError(t, err)
+ groupKey := "ALL_USERS_GROUP"
+
+ member, err := client.Users.IsGroupMemberByID(admin.ID, groupKey)
+ require.NoError(t, err)
+ require.True(t, member)
+
+ member, err = client.Users.IsGroupMemberByID(admin.ID, "INVALID_GROUP")
+ require.NoError(t, err)
+ require.False(t, member)
+
+ member, err = client.Users.IsGroupMemberByUsername("INVALIDUSERNAME", groupKey)
+ require.NoError(t, err)
+ require.False(t, member)
+
+}
+
 func cleanUpUser(t *testing.T, client *teamcity.Client, id int) {
  client.Users.DeleteByID(id)
 }