124 lines
3.7 KiB
Go
124 lines
3.7 KiB
Go
// Copyright (C) 2020 Storj Labs, Inc.
|
|
// See LICENSE for copying information.
|
|
|
|
package consoleapi_test
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"net/http"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"storj.io/common/testcontext"
|
|
"storj.io/storj/private/testplanet"
|
|
"storj.io/storj/storagenode/notifications"
|
|
)
|
|
|
|
func TestNotificationsApi(t *testing.T) {
|
|
testplanet.Run(t,
|
|
testplanet.Config{
|
|
SatelliteCount: 2,
|
|
StorageNodeCount: 1,
|
|
},
|
|
func(t *testing.T, ctx *testcontext.Context, planet *testplanet.Planet) {
|
|
satellite := planet.Satellites[0]
|
|
satellite2 := planet.Satellites[1]
|
|
sno := planet.StorageNodes[0]
|
|
console := sno.Console
|
|
notificationsDB := sno.DB.Notifications()
|
|
baseURL := fmt.Sprintf("http://%s/api/notifications", console.Listener.Addr())
|
|
|
|
newNotification1 := notifications.NewNotification{
|
|
SenderID: satellite.ID(),
|
|
Type: 0,
|
|
Title: "title1",
|
|
Message: "title1",
|
|
}
|
|
newNotification2 := notifications.NewNotification{
|
|
SenderID: satellite2.ID(),
|
|
Type: 0,
|
|
Title: "title2",
|
|
Message: "title2",
|
|
}
|
|
notif1, err := notificationsDB.Insert(ctx, newNotification1)
|
|
require.NoError(t, err)
|
|
require.Equal(t, newNotification1.Title, notif1.Title)
|
|
require.Equal(t, newNotification1.Type, notif1.Type)
|
|
require.Equal(t, newNotification1.SenderID, notif1.SenderID)
|
|
|
|
notif2, err := notificationsDB.Insert(ctx, newNotification2)
|
|
require.NoError(t, err)
|
|
require.Equal(t, newNotification2.Title, notif2.Title)
|
|
require.Equal(t, newNotification2.Type, notif2.Type)
|
|
require.Equal(t, newNotification2.SenderID, notif2.SenderID)
|
|
|
|
t.Run("test ListNotifications", func(t *testing.T) {
|
|
// should return notifications list.
|
|
url := fmt.Sprintf("%s/list?limit=3&page=1", baseURL)
|
|
res, err := http.Get(url)
|
|
require.NoError(t, err)
|
|
require.NotNil(t, res)
|
|
require.Equal(t, http.StatusOK, res.StatusCode)
|
|
|
|
expected, err := json.Marshal([]notifications.Notification{notif2, notif1})
|
|
require.NoError(t, err)
|
|
|
|
defer func() {
|
|
err = res.Body.Close()
|
|
require.NoError(t, err)
|
|
}()
|
|
body, err := ioutil.ReadAll(res.Body)
|
|
require.NoError(t, err)
|
|
|
|
require.Equal(t, "{\"page\":{\"notifications\":"+string(expected)+",\"offset\":0,\"limit\":3,\"currentPage\":1,\"pageCount\":1},\"unreadCount\":2,\"totalCount\":2}"+"\n", string(body))
|
|
})
|
|
|
|
t.Run("test ReadNotification", func(t *testing.T) {
|
|
// should change status of notification by id to read.
|
|
url := fmt.Sprintf("%s/%s/read", baseURL, notif1.ID.String())
|
|
res, err := http.Post(url, "applicationJSON", nil)
|
|
require.NoError(t, err)
|
|
require.NotNil(t, res)
|
|
require.Equal(t, http.StatusOK, res.StatusCode)
|
|
|
|
defer func() {
|
|
err = res.Body.Close()
|
|
require.NoError(t, err)
|
|
}()
|
|
|
|
notificationList, err := notificationsDB.List(ctx, notifications.Cursor{
|
|
Limit: 2,
|
|
Page: 1,
|
|
})
|
|
require.NoError(t, err)
|
|
require.NotEqual(t, nil, notificationList.Notifications[1].ReadAt)
|
|
})
|
|
|
|
t.Run("test ReadAllNotifications", func(t *testing.T) {
|
|
// should change status of notification by id to read.
|
|
url := fmt.Sprintf("%s/readall", baseURL)
|
|
res, err := http.Post(url, "applicationJSON", nil)
|
|
require.NoError(t, err)
|
|
require.NotNil(t, res)
|
|
require.Equal(t, http.StatusOK, res.StatusCode)
|
|
|
|
defer func() {
|
|
err = res.Body.Close()
|
|
require.NoError(t, err)
|
|
}()
|
|
|
|
notificationList, err := notificationsDB.List(ctx, notifications.Cursor{
|
|
Limit: 2,
|
|
Page: 1,
|
|
})
|
|
require.NoError(t, err)
|
|
require.NotEqual(t, nil, notificationList.Notifications[1].ReadAt)
|
|
require.NotEqual(t, nil, notificationList.Notifications[0].ReadAt)
|
|
})
|
|
},
|
|
)
|
|
}
|