mirror of
https://github.com/mattermost/mattermost.git
synced 2025-02-25 18:55:24 -06:00
efaa6264cc
It was a good decision in hindsight to keep the public module as 0.x because this would have been a breaking change again. https://mattermost.atlassian.net/browse/MM-53032 ```release-note Changed the Go module path from github.com/mattermost/mattermost-server/server/v8 to github.com/mattermost/mattermost/server/v8. For the public facing module, it's path is also changed from github.com/mattermost/mattermost-server/server/public to github.com/mattermost/mattermost/server/public ```
246 lines
9.1 KiB
Go
246 lines
9.1 KiB
Go
// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
|
|
// See LICENSE.txt for license information.
|
|
|
|
package api4
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/mattermost/mattermost/server/public/model"
|
|
)
|
|
|
|
func TestGetUserStatus(t *testing.T) {
|
|
th := Setup(t).InitBasic()
|
|
defer th.TearDown()
|
|
client := th.Client
|
|
|
|
t.Run("offline status", func(t *testing.T) {
|
|
userStatus, _, err := client.GetUserStatus(context.Background(), th.BasicUser.Id, "")
|
|
require.NoError(t, err)
|
|
assert.Equal(t, "offline", userStatus.Status)
|
|
})
|
|
|
|
t.Run("online status", func(t *testing.T) {
|
|
th.App.SetStatusOnline(th.BasicUser.Id, true)
|
|
userStatus, _, err := client.GetUserStatus(context.Background(), th.BasicUser.Id, "")
|
|
require.NoError(t, err)
|
|
assert.Equal(t, "online", userStatus.Status)
|
|
})
|
|
|
|
t.Run("away status", func(t *testing.T) {
|
|
th.App.SetStatusAwayIfNeeded(th.BasicUser.Id, true)
|
|
userStatus, _, err := client.GetUserStatus(context.Background(), th.BasicUser.Id, "")
|
|
require.NoError(t, err)
|
|
assert.Equal(t, "away", userStatus.Status)
|
|
})
|
|
|
|
t.Run("dnd status", func(t *testing.T) {
|
|
th.App.SetStatusDoNotDisturb(th.BasicUser.Id)
|
|
userStatus, _, err := client.GetUserStatus(context.Background(), th.BasicUser.Id, "")
|
|
require.NoError(t, err)
|
|
assert.Equal(t, "dnd", userStatus.Status)
|
|
})
|
|
|
|
t.Run("dnd status timed", func(t *testing.T) {
|
|
th.App.SetStatusDoNotDisturbTimed(th.BasicUser.Id, time.Now().Add(10*time.Minute).Unix())
|
|
userStatus, _, err := client.GetUserStatus(context.Background(), th.BasicUser.Id, "")
|
|
require.NoError(t, err)
|
|
assert.Equal(t, "dnd", userStatus.Status)
|
|
})
|
|
|
|
t.Run("dnd status timed restore after time interval", func(t *testing.T) {
|
|
task := model.CreateRecurringTaskFromNextIntervalTime("Unset DND Statuses From Test", th.App.UpdateDNDStatusOfUsers, 1*time.Second)
|
|
defer task.Cancel()
|
|
th.App.SetStatusOnline(th.BasicUser.Id, true)
|
|
userStatus, _, err := client.GetUserStatus(context.Background(), th.BasicUser.Id, "")
|
|
require.NoError(t, err)
|
|
assert.Equal(t, "online", userStatus.Status)
|
|
th.App.SetStatusDoNotDisturbTimed(th.BasicUser.Id, time.Now().Add(2*time.Second).Unix())
|
|
userStatus, _, err = client.GetUserStatus(context.Background(), th.BasicUser.Id, "")
|
|
require.NoError(t, err)
|
|
assert.Equal(t, "dnd", userStatus.Status)
|
|
time.Sleep(3 * time.Second)
|
|
userStatus, _, err = client.GetUserStatus(context.Background(), th.BasicUser.Id, "")
|
|
require.NoError(t, err)
|
|
assert.Equal(t, "online", userStatus.Status)
|
|
})
|
|
|
|
t.Run("back to offline status", func(t *testing.T) {
|
|
th.App.SetStatusOffline(th.BasicUser.Id, true)
|
|
userStatus, _, err := client.GetUserStatus(context.Background(), th.BasicUser.Id, "")
|
|
require.NoError(t, err)
|
|
assert.Equal(t, "offline", userStatus.Status)
|
|
})
|
|
|
|
t.Run("get other user status", func(t *testing.T) {
|
|
//Get user2 status logged as user1
|
|
userStatus, _, err := client.GetUserStatus(context.Background(), th.BasicUser2.Id, "")
|
|
require.NoError(t, err)
|
|
assert.Equal(t, "offline", userStatus.Status)
|
|
})
|
|
|
|
t.Run("get status from logged out user", func(t *testing.T) {
|
|
client.Logout(context.Background())
|
|
_, resp, err := client.GetUserStatus(context.Background(), th.BasicUser2.Id, "")
|
|
require.Error(t, err)
|
|
CheckUnauthorizedStatus(t, resp)
|
|
})
|
|
|
|
t.Run("get status from other user", func(t *testing.T) {
|
|
th.LoginBasic2()
|
|
userStatus, _, err := client.GetUserStatus(context.Background(), th.BasicUser2.Id, "")
|
|
require.NoError(t, err)
|
|
assert.Equal(t, "offline", userStatus.Status)
|
|
})
|
|
}
|
|
|
|
func TestGetUsersStatusesByIds(t *testing.T) {
|
|
th := Setup(t).InitBasic()
|
|
defer th.TearDown()
|
|
client := th.Client
|
|
|
|
usersIds := []string{th.BasicUser.Id, th.BasicUser2.Id}
|
|
|
|
t.Run("empty userIds list", func(t *testing.T) {
|
|
_, resp, err := client.GetUsersStatusesByIds(context.Background(), []string{})
|
|
require.Error(t, err)
|
|
CheckBadRequestStatus(t, resp)
|
|
})
|
|
|
|
t.Run("completely invalid userIds list", func(t *testing.T) {
|
|
_, resp, err := client.GetUsersStatusesByIds(context.Background(), []string{"invalid_user_id", "invalid_user_id"})
|
|
require.Error(t, err)
|
|
CheckBadRequestStatus(t, resp)
|
|
})
|
|
|
|
t.Run("partly invalid userIds list", func(t *testing.T) {
|
|
_, resp, err := client.GetUsersStatusesByIds(context.Background(), []string{th.BasicUser.Id, "invalid_user_id"})
|
|
require.Error(t, err)
|
|
CheckBadRequestStatus(t, resp)
|
|
})
|
|
|
|
t.Run("offline status", func(t *testing.T) {
|
|
usersStatuses, _, err := client.GetUsersStatusesByIds(context.Background(), usersIds)
|
|
require.NoError(t, err)
|
|
for _, userStatus := range usersStatuses {
|
|
assert.Equal(t, "offline", userStatus.Status)
|
|
}
|
|
})
|
|
|
|
t.Run("online status", func(t *testing.T) {
|
|
th.App.SetStatusOnline(th.BasicUser.Id, true)
|
|
th.App.SetStatusOnline(th.BasicUser2.Id, true)
|
|
usersStatuses, _, err := client.GetUsersStatusesByIds(context.Background(), usersIds)
|
|
require.NoError(t, err)
|
|
for _, userStatus := range usersStatuses {
|
|
assert.Equal(t, "online", userStatus.Status)
|
|
}
|
|
})
|
|
|
|
t.Run("away status", func(t *testing.T) {
|
|
th.App.SetStatusAwayIfNeeded(th.BasicUser.Id, true)
|
|
th.App.SetStatusAwayIfNeeded(th.BasicUser2.Id, true)
|
|
usersStatuses, _, err := client.GetUsersStatusesByIds(context.Background(), usersIds)
|
|
require.NoError(t, err)
|
|
for _, userStatus := range usersStatuses {
|
|
assert.Equal(t, "away", userStatus.Status)
|
|
}
|
|
})
|
|
|
|
t.Run("dnd status", func(t *testing.T) {
|
|
th.App.SetStatusDoNotDisturb(th.BasicUser.Id)
|
|
th.App.SetStatusDoNotDisturb(th.BasicUser2.Id)
|
|
usersStatuses, _, err := client.GetUsersStatusesByIds(context.Background(), usersIds)
|
|
require.NoError(t, err)
|
|
for _, userStatus := range usersStatuses {
|
|
assert.Equal(t, "dnd", userStatus.Status)
|
|
}
|
|
})
|
|
|
|
t.Run("dnd status", func(t *testing.T) {
|
|
th.App.SetStatusDoNotDisturbTimed(th.BasicUser.Id, time.Now().Add(10*time.Minute).Unix())
|
|
th.App.SetStatusDoNotDisturbTimed(th.BasicUser2.Id, time.Now().Add(15*time.Minute).Unix())
|
|
usersStatuses, _, err := client.GetUsersStatusesByIds(context.Background(), usersIds)
|
|
require.NoError(t, err)
|
|
for _, userStatus := range usersStatuses {
|
|
assert.Equal(t, "dnd", userStatus.Status)
|
|
}
|
|
})
|
|
|
|
t.Run("get statuses from logged out user", func(t *testing.T) {
|
|
client.Logout(context.Background())
|
|
|
|
_, resp, err := client.GetUsersStatusesByIds(context.Background(), usersIds)
|
|
require.Error(t, err)
|
|
CheckUnauthorizedStatus(t, resp)
|
|
})
|
|
}
|
|
|
|
func TestUpdateUserStatus(t *testing.T) {
|
|
th := Setup(t).InitBasic()
|
|
defer th.TearDown()
|
|
client := th.Client
|
|
|
|
t.Run("set online status", func(t *testing.T) {
|
|
toUpdateUserStatus := &model.Status{Status: "online", UserId: th.BasicUser.Id}
|
|
updateUserStatus, _, err := client.UpdateUserStatus(context.Background(), th.BasicUser.Id, toUpdateUserStatus)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, "online", updateUserStatus.Status)
|
|
})
|
|
|
|
t.Run("set away status", func(t *testing.T) {
|
|
toUpdateUserStatus := &model.Status{Status: "away", UserId: th.BasicUser.Id}
|
|
updateUserStatus, _, err := client.UpdateUserStatus(context.Background(), th.BasicUser.Id, toUpdateUserStatus)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, "away", updateUserStatus.Status)
|
|
})
|
|
|
|
t.Run("set dnd status timed", func(t *testing.T) {
|
|
toUpdateUserStatus := &model.Status{Status: "dnd", UserId: th.BasicUser.Id, DNDEndTime: time.Now().Add(10 * time.Minute).Unix()}
|
|
updateUserStatus, _, err := client.UpdateUserStatus(context.Background(), th.BasicUser.Id, toUpdateUserStatus)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, "dnd", updateUserStatus.Status)
|
|
})
|
|
|
|
t.Run("set offline status", func(t *testing.T) {
|
|
toUpdateUserStatus := &model.Status{Status: "offline", UserId: th.BasicUser.Id}
|
|
updateUserStatus, _, err := client.UpdateUserStatus(context.Background(), th.BasicUser.Id, toUpdateUserStatus)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, "offline", updateUserStatus.Status)
|
|
})
|
|
|
|
t.Run("set status for other user as regular user", func(t *testing.T) {
|
|
toUpdateUserStatus := &model.Status{Status: "online", UserId: th.BasicUser2.Id}
|
|
_, resp, err := client.UpdateUserStatus(context.Background(), th.BasicUser2.Id, toUpdateUserStatus)
|
|
require.Error(t, err)
|
|
CheckForbiddenStatus(t, resp)
|
|
})
|
|
|
|
t.Run("set status for other user as admin user", func(t *testing.T) {
|
|
toUpdateUserStatus := &model.Status{Status: "online", UserId: th.BasicUser2.Id}
|
|
updateUserStatus, _, _ := th.SystemAdminClient.UpdateUserStatus(context.Background(), th.BasicUser2.Id, toUpdateUserStatus)
|
|
assert.Equal(t, "online", updateUserStatus.Status)
|
|
})
|
|
|
|
t.Run("not matching status user id and the user id passed in the function", func(t *testing.T) {
|
|
toUpdateUserStatus := &model.Status{Status: "online", UserId: th.BasicUser2.Id}
|
|
_, resp, err := client.UpdateUserStatus(context.Background(), th.BasicUser.Id, toUpdateUserStatus)
|
|
require.Error(t, err)
|
|
CheckBadRequestStatus(t, resp)
|
|
})
|
|
|
|
t.Run("get statuses from logged out user", func(t *testing.T) {
|
|
toUpdateUserStatus := &model.Status{Status: "online", UserId: th.BasicUser2.Id}
|
|
client.Logout(context.Background())
|
|
|
|
_, resp, err := client.UpdateUserStatus(context.Background(), th.BasicUser2.Id, toUpdateUserStatus)
|
|
require.Error(t, err)
|
|
CheckUnauthorizedStatus(t, resp)
|
|
})
|
|
}
|