2fas-server/e2e-tests/mobile/mobile_notifications_test.go

package tests

import (
	"net/http"
	"testing"

	"github.com/google/uuid"
	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/suite"
	"github.com/twofas/2fas-server/e2e-tests"
	query "github.com/twofas/2fas-server/internal/api/mobile/app/queries"
)

func TestMobileNotificationsTestSuite(t *testing.T) {
	suite.Run(t, new(MobileNotificationsTestSuite))
}

type MobileNotificationsTestSuite struct {
	suite.Suite
}

func (s *MobileNotificationsTestSuite) SetupTest() {
	e2e_tests.RemoveAllMobileNotifications(s.T())
}

func (s *MobileNotificationsTestSuite) TestCreateMobileNotification() {
	payload := []byte(`{"icon":"features","platform":"android","link":"2fas.com","message":"demo","version":"0.1"}`)

	var notification *query.MobileNotificationPresenter

	e2e_tests.DoAdminAPISuccessPost(s.T(), "mobile/notifications", payload, &notification)

	assert.Equal(s.T(), "android", notification.Platform)
	assert.Equal(s.T(), "0.1", notification.Version)
	assert.Equal(s.T(), "2fas.com", notification.Link)
	assert.Equal(s.T(), "demo", notification.Message)
	assert.Equal(s.T(), "features", notification.Icon)
}

func (s *MobileNotificationsTestSuite) TestUpdateMobileNotification() {
	payload := []byte(`{"icon":"features","platform":"android","link":"2fas.com","message":"demo","version":"0.1"}`)
	var notification *query.MobileNotificationPresenter
	e2e_tests.DoAdminAPISuccessPost(s.T(), "mobile/notifications", payload, &notification)

	payload = []byte(`{"icon":"youtube","platform":"ios","link":"new-2fas.com","message":"new-demo","version":"1.1"}`)
	var updatedNotification *query.MobileNotificationPresenter
	e2e_tests.DoAdminSuccessPut(s.T(), "mobile/notifications/"+notification.Id, payload, &updatedNotification)

	assert.Equal(s.T(), "ios", updatedNotification.Platform)
	assert.Equal(s.T(), "1.1", updatedNotification.Version)
	assert.Equal(s.T(), "new-2fas.com", updatedNotification.Link)
	assert.Equal(s.T(), "new-demo", updatedNotification.Message)
	assert.Equal(s.T(), "youtube", updatedNotification.Icon)
}

func (s *MobileNotificationsTestSuite) TestDeleteMobileNotification() {
	payload := []byte(`{"icon":"features","platform":"android","link":"2fas.com","message":"demo","version":"0.1"}`)
	var notification *query.MobileNotificationPresenter
	e2e_tests.DoAdminAPISuccessPost(s.T(), "mobile/notifications", payload, &notification)

	e2e_tests.DoAdminSuccessDelete(s.T(), "mobile/notifications/"+notification.Id)

	response := e2e_tests.DoAPIGet(s.T(), "mobile/notifications/"+notification.Id, nil)
	assert.Equal(s.T(), 404, response.StatusCode)
}

func (s *MobileNotificationsTestSuite) TestDeleteNotExistingMobileNotification() {
	id := uuid.New()

	response := e2e_tests.DoAPIRequest(s.T(), "mobile/notifications/"+id.String(), http.MethodDelete, nil /*payload*/, nil /*resp*/)

	assert.Equal(s.T(), 404, response.StatusCode)
}

func (s *MobileNotificationsTestSuite) TestFindAllNotifications() {
	payload1 := []byte(`{"icon":"features","platform":"android","link":"2fas.com","message":"demo","version":"0.1"}`)
	var notification1 *query.MobileNotificationPresenter
	e2e_tests.DoAdminAPISuccessPost(s.T(), "mobile/notifications", payload1, &notification1)

	payload2 := []byte(`{"icon":"youtube","platform":"android","link":"2fas.com","message":"demo2","version":"1.1"}`)
	var notification2 *query.MobileNotificationPresenter
	e2e_tests.DoAdminAPISuccessPost(s.T(), "mobile/notifications", payload2, &notification2)

	var collection []*query.MobileNotificationPresenter
	e2e_tests.DoAPISuccessGet(s.T(), "mobile/notifications", &collection)

	assert.Len(s.T(), collection, 2)
}

func (s *MobileNotificationsTestSuite) TestDoNotFindNotifications() {
	var collection []*query.MobileNotificationPresenter

	e2e_tests.DoAPISuccessGet(s.T(), "mobile/notifications", &collection)

	assert.Len(s.T(), collection, 0)
}

func (s *MobileNotificationsTestSuite) TestPublishNotification() {
	payload := []byte(`{"icon":"features","platform":"android","link":"2fas.com","message":"demo","version":"0.1"}`)
	var notification *query.MobileNotificationPresenter
	e2e_tests.DoAdminAPISuccessPost(s.T(), "mobile/notifications", payload, &notification)

	var publishedNotification *query.MobileNotificationPresenter
	e2e_tests.DoAdminAPISuccessPost(s.T(), "mobile/notifications/"+notification.Id+"/commands/publish", payload, &publishedNotification)

	assert.NotEmpty(s.T(), "published_at", notification.PublishedAt)
}