diff options
Diffstat (limited to 'api4')
-rw-r--r-- | api4/team.go | 24 | ||||
-rw-r--r-- | api4/team_test.go | 31 |
2 files changed, 55 insertions, 0 deletions
diff --git a/api4/team.go b/api4/team.go index 9071458cb..c88c553fb 100644 --- a/api4/team.go +++ b/api4/team.go @@ -18,6 +18,7 @@ func InitTeam() { BaseRoutes.Teams.Handle("", ApiSessionRequired(createTeam)).Methods("POST") BaseRoutes.Teams.Handle("", ApiSessionRequired(getAllTeams)).Methods("GET") BaseRoutes.TeamsForUser.Handle("", ApiSessionRequired(getTeamsForUser)).Methods("GET") + BaseRoutes.TeamsForUser.Handle("/unread", ApiSessionRequired(getTeamsUnreadForUser)).Methods("GET") BaseRoutes.Team.Handle("", ApiSessionRequired(getTeam)).Methods("GET") BaseRoutes.Team.Handle("/stats", ApiSessionRequired(getTeamStats)).Methods("GET") @@ -104,6 +105,29 @@ func getTeamsForUser(c *Context, w http.ResponseWriter, r *http.Request) { } } +func getTeamsUnreadForUser(c *Context, w http.ResponseWriter, r *http.Request) { + c.RequireUserId() + if c.Err != nil { + return + } + + if c.Session.UserId != c.Params.UserId && !app.SessionHasPermissionTo(c.Session, model.PERMISSION_MANAGE_SYSTEM) { + c.SetPermissionError(model.PERMISSION_MANAGE_SYSTEM) + return + } + + // optional team id to be excluded from the result + teamId := r.URL.Query().Get("exclude_team") + + unreadTeamsList, err := app.GetTeamsUnreadForUser(teamId, c.Params.UserId) + if err != nil { + c.Err = err + return + } + + w.Write([]byte(model.TeamsUnreadToJson(unreadTeamsList))) +} + func getTeamMember(c *Context, w http.ResponseWriter, r *http.Request) { c.RequireTeamId().RequireUserId() if c.Err != nil { diff --git a/api4/team_test.go b/api4/team_test.go index a4ec1a6e2..e54cf4414 100644 --- a/api4/team_test.go +++ b/api4/team_test.go @@ -463,3 +463,34 @@ func TestUpdateTeamMemberRoles(t *testing.T) { _, resp = Client.UpdateTeamMemberRoles(th.BasicTeam.Id, th.BasicUser.Id, TEAM_MEMBER) CheckNoError(t, resp) } + +func TestGetMyTeamsUnread(t *testing.T) { + th := Setup().InitBasic().InitSystemAdmin() + defer TearDown() + Client := th.Client + + user := th.BasicUser + Client.Login(user.Email, user.Password) + + teams, resp := Client.GetTeamsUnreadForUser(user.Id, "") + CheckNoError(t, resp) + if len(teams) == 0 { + t.Fatal("should have results") + } + + teams, resp = Client.GetTeamsUnreadForUser(user.Id, th.BasicTeam.Id) + CheckNoError(t, resp) + if len(teams) != 0 { + t.Fatal("should not have results") + } + + _, resp = Client.GetTeamsUnreadForUser("fail", "") + CheckBadRequestStatus(t, resp) + + _, resp = Client.GetTeamsUnreadForUser(model.NewId(), "") + CheckForbiddenStatus(t, resp) + + Client.Logout() + _, resp = Client.GetTeamsUnreadForUser(user.Id, "") + CheckUnauthorizedStatus(t, resp) +}
\ No newline at end of file |