diff options
author | 2023-03-20 19:10:08 +0100 | |
---|---|---|
committer | 2023-03-20 18:10:08 +0000 | |
commit | e8595f0c64f527af0913d1a426b697e67ff74ac9 (patch) | |
tree | a5d45b1ad8b96318944408a23fda91f008643900 /internal/db/bundb/statusfave_test.go | |
parent | [chore]: Bump github.com/miekg/dns from 1.1.51 to 1.1.52 (#1636) (diff) | |
download | gotosocial-e8595f0c64f527af0913d1a426b697e67ff74ac9.tar.xz |
[chore] Refactor account deleting/block logic, tidy up some other processing things (#1599)
* start refactoring account deletion
* update to use state.DB
* further messing about
* some more tidying up
* more tidying, cleaning, nice-making
* further adventures in refactoring and the woes of technical debt
* update fr accept/reject
* poking + prodding
* fix up deleting
* create fave uri
* don't log using requestingAccount.ID because it might be nil
* move getBookmarks function
* use exists query to check for status bookmark
* use deletenotifications func
* fiddle
* delete follow request notif
* split up some db functions
* Fix possible nil pointer panic
* fix more possible nil pointers
* fix license headers
* warn when follow missing (target) account
* return wrapped err when bookmark/fave models can't be retrieved
* simplify self account delete
* warn log likely race condition
* de-sillify status delete loop
* move error check due north
* warn when unfollowSideEffects has no target account
* warn when no boost account is found
* warn + dump follow when no account
* more warnings
* warn on fave account not set
* move for loop inside anonymous function
* fix funky logic
* don't remove mutual account items on block;
do make sure unfollow occurs in both directions!
Diffstat (limited to 'internal/db/bundb/statusfave_test.go')
-rw-r--r-- | internal/db/bundb/statusfave_test.go | 148 |
1 files changed, 148 insertions, 0 deletions
diff --git a/internal/db/bundb/statusfave_test.go b/internal/db/bundb/statusfave_test.go new file mode 100644 index 000000000..98e495bf3 --- /dev/null +++ b/internal/db/bundb/statusfave_test.go @@ -0,0 +1,148 @@ +/* + 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 <http://www.gnu.org/licenses/>. +*/ + +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.GetStatusFaveByAccountID(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.DeleteStatusFave(ctx, testFave.ID); err != nil { + suite.FailNow(err.Error()) + } + + fave, err := suite.db.GetStatusFave(ctx, testFave.ID) + suite.ErrorIs(err, db.ErrNoEntries) + suite.Nil(fave) +} + +func (suite *StatusFaveTestSuite) TestDeleteStatusFaveNonExisting() { + err := suite.db.DeleteStatusFave(context.Background(), "01GVAV715K6Y2SG9ZKS9ZA8G7G") + suite.NoError(err) +} + +func TestStatusFaveTestSuite(t *testing.T) { + suite.Run(t, new(StatusFaveTestSuite)) +} |