/* GoToSocial Copyright (C) 2021-2023 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 bundb_test import ( "context" "errors" "testing" "github.com/stretchr/testify/suite" "github.com/superseriousbusiness/gotosocial/internal/db" "github.com/superseriousbusiness/gotosocial/internal/gtsmodel" ) type StatusFaveTestSuite struct { BunDBStandardTestSuite } func (suite *StatusFaveTestSuite) TestGetStatusFaves() { testStatus := suite.testStatuses["admin_account_status_1"] faves, err := suite.db.GetStatusFaves(context.Background(), testStatus.ID) if err != nil { suite.FailNow(err.Error()) } suite.NotEmpty(faves) for _, fave := range faves { suite.NotNil(fave.Account) suite.NotNil(fave.TargetAccount) suite.NotNil(fave.Status) } } func (suite *StatusFaveTestSuite) TestGetStatusFavesNone() { testStatus := suite.testStatuses["admin_account_status_4"] faves, err := suite.db.GetStatusFaves(context.Background(), testStatus.ID) if err != nil { suite.FailNow(err.Error()) } suite.Empty(faves) } func (suite *StatusFaveTestSuite) TestGetStatusFaveByAccountID() { testAccount := suite.testAccounts["local_account_1"] testStatus := suite.testStatuses["admin_account_status_1"] fave, err := suite.db.GetStatusFave(context.Background(), testAccount.ID, testStatus.ID) suite.NoError(err) suite.NotNil(fave) } func (suite *StatusFaveTestSuite) TestDeleteStatusFavesOriginatingFromAccount() { testAccount := suite.testAccounts["local_account_1"] if err := suite.db.DeleteStatusFaves(context.Background(), "", testAccount.ID); err != nil { suite.FailNow(err.Error()) } faves := []*gtsmodel.StatusFave{} if err := suite.db.GetAll(context.Background(), &faves); err != nil && !errors.Is(err, db.ErrNoEntries) { suite.FailNow(err.Error()) } for _, b := range faves { if b.AccountID == testAccount.ID { suite.FailNowf("", "no StatusFaves with account id %s should remain", testAccount.ID) } } } func (suite *StatusFaveTestSuite) TestDeleteStatusFavesTargetingAccount() { testAccount := suite.testAccounts["local_account_1"] if err := suite.db.DeleteStatusFaves(context.Background(), testAccount.ID, ""); err != nil { suite.FailNow(err.Error()) } faves := []*gtsmodel.StatusFave{} if err := suite.db.GetAll(context.Background(), &faves); err != nil && !errors.Is(err, db.ErrNoEntries) { suite.FailNow(err.Error()) } for _, b := range faves { if b.TargetAccountID == testAccount.ID { suite.FailNowf("", "no StatusFaves with target account id %s should remain", testAccount.ID) } } } func (suite *StatusFaveTestSuite) TestDeleteStatusFavesTargetingStatus() { testStatus := suite.testStatuses["local_account_1_status_1"] if err := suite.db.DeleteStatusFavesForStatus(context.Background(), testStatus.ID); err != nil { suite.FailNow(err.Error()) } faves := []*gtsmodel.StatusFave{} if err := suite.db.GetAll(context.Background(), &faves); err != nil && !errors.Is(err, db.ErrNoEntries) { suite.FailNow(err.Error()) } for _, b := range faves { if b.StatusID == testStatus.ID { suite.FailNowf("", "no StatusFaves with status id %s should remain", testStatus.ID) } } } func (suite *StatusFaveTestSuite) TestDeleteStatusFave() { testFave := suite.testFaves["local_account_1_admin_account_status_1"] ctx := context.Background() if err := suite.db.DeleteStatusFaveByID(ctx, testFave.ID); err != nil { suite.FailNow(err.Error()) } fave, err := suite.db.GetStatusFaveByID(ctx, testFave.ID) suite.ErrorIs(err, db.ErrNoEntries) suite.Nil(fave) } func (suite *StatusFaveTestSuite) TestDeleteStatusFaveNonExisting() { err := suite.db.DeleteStatusFaveByID(context.Background(), "01GVAV715K6Y2SG9ZKS9ZA8G7G") suite.NoError(err) } func TestStatusFaveTestSuite(t *testing.T) { suite.Run(t, new(StatusFaveTestSuite)) }