/* GoToSocial Copyright (C) 2021 GoToSocial Authors admin@gotosocial.org This program is free software: you can redistribute it and/or modify it under the terms of the GNU Affero General Public License as published by the Free Software Foundation, either version 3 of the License, or (at your option) any later version. This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU Affero General Public License for more details. You should have received a copy of the GNU Affero General Public License along with this program. If not, see . */ package gtsmodel_test import ( "testing" "time" "github.com/stretchr/testify/suite" "github.com/superseriousbusiness/gotosocial/internal/gtsmodel" ) func happyNotification() *gtsmodel.Notification { return >smodel.Notification{ ID: "01FE91RJR88PSEEE30EV35QR8N", CreatedAt: time.Now(), NotificationType: gtsmodel.NotificationFave, OriginAccountID: "01FE96MAE58MXCE5C4SSMEMCEK", OriginAccount: nil, TargetAccountID: "01FE96MXRHWZHKC0WH5FT82H1A", TargetAccount: nil, StatusID: "01FE96NBPNJNY26730FT6GZTFE", Status: nil, } } type NotificationValidateTestSuite struct { suite.Suite } func (suite *NotificationValidateTestSuite) TestValidateNotificationHappyPath() { // no problem here m := happyNotification() err := gtsmodel.ValidateStruct(*m) suite.NoError(err) } func (suite *NotificationValidateTestSuite) TestValidateNotificationBadID() { m := happyNotification() m.ID = "" err := gtsmodel.ValidateStruct(*m) suite.EqualError(err, "Key: 'Notification.ID' Error:Field validation for 'ID' failed on the 'ulid' tag") m.ID = "01FE96W293ZPRG9FQQP48HK8N001FE96W32AT24VYBGM12WN3GKB" err = gtsmodel.ValidateStruct(*m) suite.EqualError(err, "Key: 'Notification.ID' Error:Field validation for 'ID' failed on the 'ulid' tag") } func (suite *NotificationValidateTestSuite) TestValidateNotificationStatusID() { m := happyNotification() m.StatusID = "" err := gtsmodel.ValidateStruct(*m) suite.EqualError(err, "Key: 'Notification.StatusID' Error:Field validation for 'StatusID' failed on the 'required_if' tag") m.StatusID = "9HZJ76B6VXSKF" err = gtsmodel.ValidateStruct(*m) suite.EqualError(err, "Key: 'Notification.StatusID' Error:Field validation for 'StatusID' failed on the 'ulid' tag") m.StatusID = "aaaaaaaaaaaaaaaaaaaaaaaaaaaaaa!!!!!!!!!!!!" err = gtsmodel.ValidateStruct(*m) suite.EqualError(err, "Key: 'Notification.StatusID' Error:Field validation for 'StatusID' failed on the 'ulid' tag") m.StatusID = "" m.NotificationType = gtsmodel.NotificationFollowRequest err = gtsmodel.ValidateStruct(*m) suite.NoError(err) } func (suite *NotificationValidateTestSuite) TestValidateNotificationNoCreatedAt() { m := happyNotification() m.CreatedAt = time.Time{} err := gtsmodel.ValidateStruct(*m) suite.EqualError(err, "Key: 'Notification.CreatedAt' Error:Field validation for 'CreatedAt' failed on the 'required' tag") } func TestNotificationValidateTestSuite(t *testing.T) { suite.Run(t, new(NotificationValidateTestSuite)) }