Implement GET teams/{team_id}/stats - apiv4 (#5453)

* api4 GET teams/{team_id}/stats

* api4 GET teams/{team_id}/stats

* moved TeamStats router to team.go

3 files changed, 86 insertions, 1 deletions

diff --git a/api4/apitestlib.go b/api4/apitestlib.go
index 27bf83f10..84f87d494 100644
--- a/api4/apitestlib.go
+++ b/api4/apitestlib.go
@@ -260,6 +260,20 @@ func (me *TestHelper) LoginSystemAdminWithClient(client *model.Client4) {
 	utils.EnableDebugLogForTest()
 }
 
+func (me *TestHelper) UpdateActiveUser(user *model.User, active bool) {
+	utils.DisableDebugLogForTest()
+
+	_, err := app.UpdateActive(user, active)
+	if err != nil {
+		l4g.Error(err.Error())
+		l4g.Close()
+		time.Sleep(time.Second)
+		panic(err)
+	}
+
+	utils.EnableDebugLogForTest()
+}
+
 func LinkUserToTeam(user *model.User, team *model.Team) {
 	utils.DisableDebugLogForTest()
 
diff --git a/api4/team.go b/api4/team.go
index 8564029b0..d1db268dc 100644
--- a/api4/team.go
+++ b/api4/team.go
@@ -19,9 +19,9 @@ func InitTeam() {
 	BaseRoutes.TeamsForUser.Handle("", ApiSessionRequired(getTeamsForUser)).Methods("GET")
 
 	BaseRoutes.Team.Handle("", ApiSessionRequired(getTeam)).Methods("GET")
+	BaseRoutes.Team.Handle("/stats", ApiHandler(getTeamStats)).Methods("GET")
 
 	BaseRoutes.TeamMember.Handle("", ApiSessionRequired(getTeamMember)).Methods("GET")
-
 }
 
 func createTeam(c *Context, w http.ResponseWriter, r *http.Request) {
@@ -104,3 +104,23 @@ func getTeamMember(c *Context, w http.ResponseWriter, r *http.Request) {
 		return
 	}
 }
+
+func getTeamStats(c *Context, w http.ResponseWriter, r *http.Request) {
+	c.RequireTeamId()
+	if c.Err != nil {
+		return
+	}
+
+	if !app.SessionHasPermissionToTeam(c.Session, c.Params.TeamId, model.PERMISSION_VIEW_TEAM) {
+		c.SetPermissionError(model.PERMISSION_VIEW_TEAM)
+		return
+	}
+
+	if stats, err := app.GetTeamStats(c.Params.TeamId); err != nil {
+		c.Err = err
+		return
+	} else {
+		w.Write([]byte(stats.ToJson()))
+		return
+	}
+}
diff --git a/api4/team_test.go b/api4/team_test.go
index 5c6d64ace..5f7bc7b98 100644
--- a/api4/team_test.go
+++ b/api4/team_test.go
@@ -192,3 +192,54 @@ func TestGetTeamMember(t *testing.T) {
 	_, resp = th.SystemAdminClient.GetTeamMember(team.Id, user.Id, "")
 	CheckNoError(t, resp)
 }
+
+func TestGetTeamStats(t *testing.T) {
+	th := Setup().InitBasic().InitSystemAdmin()
+	defer TearDown()
+	Client := th.Client
+	team := th.BasicTeam
+
+	rstats, resp := Client.GetTeamStats(team.Id, "")
+	CheckNoError(t, resp)
+
+	if rstats.TeamId != team.Id {
+		t.Fatal("wrong team id")
+	}
+
+	if rstats.TotalMemberCount != 3 {
+		t.Fatal("wrong count")
+	}
+
+	if rstats.ActiveMemberCount != 3 {
+		t.Fatal("wrong count")
+	}
+
+	_, resp = Client.GetTeamStats("junk", "")
+	CheckBadRequestStatus(t, resp)
+
+	_, resp = Client.GetTeamStats(model.NewId(), "")
+	CheckForbiddenStatus(t, resp)
+
+	_, resp = th.SystemAdminClient.GetTeamStats(team.Id, "")
+	CheckNoError(t, resp)
+
+	// deactivate BasicUser2
+	th.UpdateActiveUser(th.BasicUser2, false)
+
+	rstats, resp = th.SystemAdminClient.GetTeamStats(team.Id, "")
+	CheckNoError(t, resp)
+
+	if rstats.TotalMemberCount != 3 {
+		t.Fatal("wrong count")
+	}
+
+	if rstats.ActiveMemberCount != 2 {
+		t.Fatal("wrong count")
+	}
+
+	// login with different user and test if forbidden
+	user := th.CreateUser()
+	Client.Login(user.Email, user.Password)
+	_, resp = Client.GetTeamStats(th.BasicTeam.Id, "")
+	CheckForbiddenStatus(t, resp)
+}