// GoToSocial // Copyright (C) GoToSocial Authors admin@gotosocial.org // SPDX-License-Identifier: AGPL-3.0-or-later // // 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" "reflect" "testing" "time" "github.com/stretchr/testify/suite" "github.com/superseriousbusiness/gotosocial/internal/db" "github.com/superseriousbusiness/gotosocial/internal/gtsmodel" "github.com/superseriousbusiness/gotosocial/internal/id" "github.com/superseriousbusiness/gotosocial/internal/util" ) type RelationshipTestSuite struct { BunDBStandardTestSuite } func (suite *RelationshipTestSuite) TestGetBlockBy() { t := suite.T() // Create a new context for this test. ctx, cncl := context.WithCancel(context.Background()) defer cncl() // Sentinel error to mark avoiding a test case. sentinelErr := errors.New("sentinel") // isEqual checks if 2 block models are equal. isEqual := func(b1, b2 gtsmodel.Block) bool { // Clear populated sub-models. b1.Account = nil b2.Account = nil b1.TargetAccount = nil b2.TargetAccount = nil // Clear database-set fields. b1.CreatedAt = time.Time{} b2.CreatedAt = time.Time{} b1.UpdatedAt = time.Time{} b2.UpdatedAt = time.Time{} return reflect.DeepEqual(b1, b2) } var testBlocks []*gtsmodel.Block for _, account1 := range suite.testAccounts { for _, account2 := range suite.testAccounts { if account1.ID == account2.ID { // don't block *yourself* ... continue } // Create new account block. block := >smodel.Block{ ID: id.NewULID(), URI: "http://127.0.0.1:8080/" + id.NewULID(), AccountID: account1.ID, TargetAccountID: account2.ID, } // Attempt to place the block in database (if not already). if err := suite.db.PutBlock(ctx, block); err != nil { if err != db.ErrAlreadyExists { // Unrecoverable database error. t.Fatalf("error creating block: %v", err) } // Fetch existing block from database between accounts. block, _ = suite.db.GetBlock(ctx, account1.ID, account2.ID) continue } // Append generated block to test cases. testBlocks = append(testBlocks, block) } } for _, block := range testBlocks { for lookup, dbfunc := range map[string]func() (*gtsmodel.Block, error){ "id": func() (*gtsmodel.Block, error) { return suite.db.GetBlockByID(ctx, block.ID) }, "uri": func() (*gtsmodel.Block, error) { return suite.db.GetBlockByURI(ctx, block.URI) }, "origin_target": func() (*gtsmodel.Block, error) { return suite.db.GetBlock(ctx, block.AccountID, block.TargetAccountID) }, } { // Clear database caches. suite.state.Caches.Init() t.Logf("checking database lookup %q", lookup) // Perform database function. checkBlock, err := dbfunc() if err != nil { if err == sentinelErr { continue } t.Errorf("error encountered for database lookup %q: %v", lookup, err) continue } // Check received block data. if !isEqual(*checkBlock, *block) { t.Errorf("block does not contain expected data: %+v", checkBlock) continue } // Check that block origin account populated. if checkBlock.Account == nil || checkBlock.Account.ID != block.AccountID { t.Errorf("block origin account not correctly populated for: %+v", checkBlock) continue } // Check that block target account populated. if checkBlock.TargetAccount == nil || checkBlock.TargetAccount.ID != block.TargetAccountID { t.Errorf("block target account not correctly populated for: %+v", checkBlock) continue } } } } func (suite *RelationshipTestSuite) TestGetFollowBy() { t := suite.T() // Create a new context for this test. ctx, cncl := context.WithCancel(context.Background()) defer cncl() // Sentinel error to mark avoiding a test case. sentinelErr := errors.New("sentinel") // isEqual checks if 2 follow models are equal. isEqual := func(f1, f2 gtsmodel.Follow) bool { // Clear populated sub-models. f1.Account = nil f2.Account = nil f1.TargetAccount = nil f2.TargetAccount = nil // Clear database-set fields. f1.CreatedAt = time.Time{} f2.CreatedAt = time.Time{} f1.UpdatedAt = time.Time{} f2.UpdatedAt = time.Time{} return reflect.DeepEqual(f1, f2) } var testFollows []*gtsmodel.Follow for _, account1 := range suite.testAccounts { for _, account2 := range suite.testAccounts { if account1.ID == account2.ID { // don't follow *yourself* ... continue } // Create new account follow. follow := >smodel.Follow{ ID: id.NewULID(), URI: "http://127.0.0.1:8080/" + id.NewULID(), AccountID: account1.ID, TargetAccountID: account2.ID, } // Attempt to place the follow in database (if not already). if err := suite.db.PutFollow(ctx, follow); err != nil { if err != db.ErrAlreadyExists { // Unrecoverable database error. t.Fatalf("error creating follow: %v", err) } // Fetch existing follow from database between accounts. follow, _ = suite.db.GetFollow(ctx, account1.ID, account2.ID) continue } // Append generated follow to test cases. testFollows = append(testFollows, follow) } } for _, follow := range testFollows { for lookup, dbfunc := range map[string]func() (*gtsmodel.Follow, error){ "id": func() (*gtsmodel.Follow, error) { return suite.db.GetFollowByID(ctx, follow.ID) }, "uri": func() (*gtsmodel.Follow, error) { return suite.db.GetFollowByURI(ctx, follow.URI) }, "origin_target": func() (*gtsmodel.Follow, error) { return suite.db.GetFollow(ctx, follow.AccountID, follow.TargetAccountID) }, } { // Clear database caches. suite.state.Caches.Init() t.Logf("checking database lookup %q", lookup) // Perform database function. checkFollow, err := dbfunc() if err != nil { if err == sentinelErr { continue } t.Errorf("error encountered for database lookup %q: %v", lookup, err) continue } // Check received follow data. if !isEqual(*checkFollow, *follow) { t.Errorf("follow does not contain expected data: %+v", checkFollow) continue } // Check that follow origin account populated. if checkFollow.Account == nil || checkFollow.Account.ID != follow.AccountID { t.Errorf("follow origin account not correctly populated for: %+v", checkFollow) continue } // Check that follow target account populated. if checkFollow.TargetAccount == nil || checkFollow.TargetAccount.ID != follow.TargetAccountID { t.Errorf("follow target account not correctly populated for: %+v", checkFollow) continue } } } } func (suite *RelationshipTestSuite) TestGetFollowRequestBy() { t := suite.T() // Create a new context for this test. ctx, cncl := context.WithCancel(context.Background()) defer cncl() // Sentinel error to mark avoiding a test case. sentinelErr := errors.New("sentinel") // isEqual checks if 2 follow request models are equal. isEqual := func(f1, f2 gtsmodel.FollowRequest) bool { // Clear populated sub-models. f1.Account = nil f2.Account = nil f1.TargetAccount = nil f2.TargetAccount = nil // Clear database-set fields. f1.CreatedAt = time.Time{} f2.CreatedAt = time.Time{} f1.UpdatedAt = time.Time{} f2.UpdatedAt = time.Time{} return reflect.DeepEqual(f1, f2) } var testFollowReqs []*gtsmodel.FollowRequest for _, account1 := range suite.testAccounts { for _, account2 := range suite.testAccounts { if account1.ID == account2.ID { // don't follow *yourself* ... continue } // Create new account follow request. followReq := >smodel.FollowRequest{ ID: id.NewULID(), URI: "http://127.0.0.1:8080/" + id.NewULID(), AccountID: account1.ID, TargetAccountID: account2.ID, } // Attempt to place the follow in database (if not already). if err := suite.db.PutFollowRequest(ctx, followReq); err != nil { if err != db.ErrAlreadyExists { // Unrecoverable database error. t.Fatalf("error creating follow request: %v", err) } // Fetch existing follow request from database between accounts. followReq, _ = suite.db.GetFollowRequest(ctx, account1.ID, account2.ID) continue } // Append generated follow request to test cases. testFollowReqs = append(testFollowReqs, followReq) } } for _, followReq := range testFollowReqs { for lookup, dbfunc := range map[string]func() (*gtsmodel.FollowRequest, error){ "id": func() (*gtsmodel.FollowRequest, error) { return suite.db.GetFollowRequestByID(ctx, followReq.ID) }, "uri": func() (*gtsmodel.FollowRequest, error) { return suite.db.GetFollowRequestByURI(ctx, followReq.URI) }, "origin_target": func() (*gtsmodel.FollowRequest, error) { return suite.db.GetFollowRequest(ctx, followReq.AccountID, followReq.TargetAccountID) }, } { // Clear database caches. suite.state.Caches.Init() t.Logf("checking database lookup %q", lookup) // Perform database function. checkFollowReq, err := dbfunc() if err != nil { if err == sentinelErr { continue } t.Errorf("error encountered for database lookup %q: %v", lookup, err) continue } // Check received follow request data. if !isEqual(*checkFollowReq, *followReq) { t.Errorf("follow request does not contain expected data: %+v", checkFollowReq) continue } // Check that follow request origin account populated. if checkFollowReq.Account == nil || checkFollowReq.Account.ID != followReq.AccountID { t.Errorf("follow request origin account not correctly populated for: %+v", checkFollowReq) continue } // Check that follow request target account populated. if checkFollowReq.TargetAccount == nil || checkFollowReq.TargetAccount.ID != followReq.TargetAccountID { t.Errorf("follow request target account not correctly populated for: %+v", checkFollowReq) continue } } } } func (suite *RelationshipTestSuite) TestIsBlocked() { ctx := context.Background() account1 := suite.testAccounts["local_account_1"].ID account2 := suite.testAccounts["local_account_2"].ID // no blocks exist between account 1 and account 2 blocked, err := suite.db.IsBlocked(ctx, account1, account2) suite.NoError(err) suite.False(blocked) blocked, err = suite.db.IsBlocked(ctx, account2, account1) suite.NoError(err) suite.False(blocked) // have account1 block account2 if err := suite.db.PutBlock(ctx, >smodel.Block{ ID: "01G202BCSXXJZ70BHB5KCAHH8C", URI: "http://localhost:8080/some_block_uri_1", AccountID: account1, TargetAccountID: account2, }); err != nil { suite.FailNow(err.Error()) } // account 1 now blocks account 2 blocked, err = suite.db.IsBlocked(ctx, account1, account2) suite.NoError(err) suite.True(blocked) // account 2 doesn't block account 1 blocked, err = suite.db.IsBlocked(ctx, account2, account1) suite.NoError(err) suite.False(blocked) // a block exists in either direction between the two blocked, err = suite.db.IsEitherBlocked(ctx, account1, account2) suite.NoError(err) suite.True(blocked) blocked, err = suite.db.IsEitherBlocked(ctx, account2, account1) suite.NoError(err) suite.True(blocked) } func (suite *RelationshipTestSuite) TestDeleteBlockByID() { ctx := context.Background() // put a block in first account1 := suite.testAccounts["local_account_1"].ID account2 := suite.testAccounts["local_account_2"].ID if err := suite.db.PutBlock(ctx, >smodel.Block{ ID: "01G202BCSXXJZ70BHB5KCAHH8C", URI: "http://localhost:8080/some_block_uri_1", AccountID: account1, TargetAccountID: account2, }); err != nil { suite.FailNow(err.Error()) } // make sure the block is in the db block, err := suite.db.GetBlock(ctx, account1, account2) suite.NoError(err) suite.NotNil(block) suite.Equal("01G202BCSXXJZ70BHB5KCAHH8C", block.ID) // delete the block by ID err = suite.db.DeleteBlockByID(ctx, "01G202BCSXXJZ70BHB5KCAHH8C") suite.NoError(err) // block should be gone block, err = suite.db.GetBlock(ctx, account1, account2) suite.ErrorIs(err, db.ErrNoEntries) suite.Nil(block) } func (suite *RelationshipTestSuite) TestDeleteBlockByURI() { ctx := context.Background() // put a block in first account1 := suite.testAccounts["local_account_1"].ID account2 := suite.testAccounts["local_account_2"].ID if err := suite.db.PutBlock(ctx, >smodel.Block{ ID: "01G202BCSXXJZ70BHB5KCAHH8C", URI: "http://localhost:8080/some_block_uri_1", AccountID: account1, TargetAccountID: account2, }); err != nil { suite.FailNow(err.Error()) } // make sure the block is in the db block, err := suite.db.GetBlock(ctx, account1, account2) suite.NoError(err) suite.NotNil(block) suite.Equal("01G202BCSXXJZ70BHB5KCAHH8C", block.ID) // delete the block by uri err = suite.db.DeleteBlockByURI(ctx, "http://localhost:8080/some_block_uri_1") suite.NoError(err) // block should be gone block, err = suite.db.GetBlock(ctx, account1, account2) suite.ErrorIs(err, db.ErrNoEntries) suite.Nil(block) } func (suite *RelationshipTestSuite) TestDeleteAccountBlocks() { ctx := context.Background() // put a block in first account1 := suite.testAccounts["local_account_1"].ID account2 := suite.testAccounts["local_account_2"].ID if err := suite.db.PutBlock(ctx, >smodel.Block{ ID: "01G202BCSXXJZ70BHB5KCAHH8C", URI: "http://localhost:8080/some_block_uri_1", AccountID: account1, TargetAccountID: account2, }); err != nil { suite.FailNow(err.Error()) } // make sure the block is in the db block, err := suite.db.GetBlock(ctx, account1, account2) suite.NoError(err) suite.NotNil(block) suite.Equal("01G202BCSXXJZ70BHB5KCAHH8C", block.ID) // delete the block by originAccountID err = suite.db.DeleteAccountBlocks(ctx, account1) suite.NoError(err) // block should be gone block, err = suite.db.GetBlock(ctx, account1, account2) suite.ErrorIs(err, db.ErrNoEntries) suite.Nil(block) } func (suite *RelationshipTestSuite) TestDeleteAccountMutes() { ctx := context.Background() // Add a mute. accountID1 := suite.testAccounts["local_account_1"].ID accountID2 := suite.testAccounts["local_account_2"].ID muteID := "01HZGZ3F3C7S1TTPE8F9VPZDCB" err := suite.db.PutMute(ctx, >smodel.UserMute{ ID: muteID, AccountID: accountID1, TargetAccountID: accountID2, }) if err != nil { suite.FailNow(err.Error()) } // Make sure the mute is in the DB. mute, err := suite.db.GetMute(ctx, accountID1, accountID2) if err != nil { suite.FailNow(err.Error()) } if suite.NotNil(mute) { suite.Equal(muteID, mute.ID) } // Delete all mutes owned by that account. err = suite.db.DeleteAccountMutes(ctx, accountID1) if err != nil { suite.FailNow(err.Error()) } // Mute should be gone. mute, err = suite.db.GetMute(ctx, accountID1, accountID2) suite.ErrorIs(err, db.ErrNoEntries) suite.Nil(mute) } func (suite *RelationshipTestSuite) TestGetRelationship() { requestingAccount := suite.testAccounts["local_account_1"] targetAccount := suite.testAccounts["admin_account"] relationship, err := suite.db.GetRelationship(context.Background(), requestingAccount.ID, targetAccount.ID) suite.NoError(err) suite.NotNil(relationship) suite.True(relationship.Following) suite.True(relationship.ShowingReblogs) suite.False(relationship.Notifying) suite.True(relationship.FollowedBy) suite.False(relationship.Blocking) suite.False(relationship.BlockedBy) suite.False(relationship.Muting) suite.False(relationship.MutingNotifications) suite.False(relationship.Requested) suite.False(relationship.DomainBlocking) suite.False(relationship.Endorsed) suite.Empty(relationship.Note) } func (suite *RelationshipTestSuite) TestIsFollowingYes() { requestingAccount := suite.testAccounts["local_account_1"] targetAccount := suite.testAccounts["admin_account"] isFollowing, err := suite.db.IsFollowing(context.Background(), requestingAccount.ID, targetAccount.ID) suite.NoError(err) suite.True(isFollowing) } func (suite *RelationshipTestSuite) TestIsFollowingNo() { requestingAccount := suite.testAccounts["admin_account"] targetAccount := suite.testAccounts["local_account_2"] isFollowing, err := suite.db.IsFollowing(context.Background(), requestingAccount.ID, targetAccount.ID) suite.NoError(err) suite.False(isFollowing) } func (suite *RelationshipTestSuite) TestIsMutualFollowing() { requestingAccount := suite.testAccounts["local_account_1"] targetAccount := suite.testAccounts["admin_account"] isMutualFollowing, err := suite.db.IsMutualFollowing(context.Background(), requestingAccount.ID, targetAccount.ID) suite.NoError(err) suite.True(isMutualFollowing) } func (suite *RelationshipTestSuite) TestIsMutualFollowingNo() { requestingAccount := suite.testAccounts["local_account_1"] targetAccount := suite.testAccounts["local_account_2"] isMutualFollowing, err := suite.db.IsMutualFollowing(context.Background(), requestingAccount.ID, targetAccount.ID) suite.NoError(err) suite.True(isMutualFollowing) } func (suite *RelationshipTestSuite) TestAcceptFollowRequestOK() { ctx := context.Background() account := suite.testAccounts["admin_account"] targetAccount := suite.testAccounts["local_account_2"] // Fetch relationship before follow request. relationship, err := suite.db.GetRelationship(ctx, account.ID, targetAccount.ID) if err != nil { suite.FailNow(err.Error()) } suite.False(relationship.Following) suite.False(relationship.Requested) followRequest := >smodel.FollowRequest{ ID: "01GEF753FWHCHRDWR0QEHBXM8W", URI: "http://localhost:8080/weeeeeeeeeeeeeeeee", AccountID: account.ID, TargetAccountID: targetAccount.ID, } if err := suite.db.PutFollowRequest(ctx, followRequest); err != nil { suite.FailNow(err.Error()) } // Fetch relationship while follow requested. relationship, err = suite.db.GetRelationship(ctx, account.ID, targetAccount.ID) if err != nil { suite.FailNow(err.Error()) } suite.False(relationship.Following) suite.True(relationship.Requested) // Check the other way around too; local_account_2 // should have requested_by true for admin now. inverse, err := suite.db.GetRelationship(ctx, targetAccount.ID, account.ID) if err != nil { suite.FailNow(err.Error()) } suite.True(inverse.RequestedBy) followRequestNotification := >smodel.Notification{ ID: "01GV8MY1Q9KX2ZSWN4FAQ3V1PB", OriginAccountID: account.ID, TargetAccountID: targetAccount.ID, NotificationType: gtsmodel.NotificationFollowRequest, } if err := suite.db.PutNotification(ctx, followRequestNotification); err != nil { suite.FailNow(err.Error()) } follow, err := suite.db.AcceptFollowRequest(ctx, account.ID, targetAccount.ID) suite.NoError(err) suite.NotNil(follow) suite.Equal(followRequest.URI, follow.URI) // Ensure notification is deleted. notification, err := suite.db.GetNotificationByID(ctx, followRequestNotification.ID) suite.ErrorIs(err, db.ErrNoEntries) suite.Nil(notification) // Fetch relationship while followed. relationship, err = suite.db.GetRelationship(ctx, account.ID, targetAccount.ID) if err != nil { suite.FailNow(err.Error()) } suite.True(relationship.Following) suite.False(relationship.Requested) } func (suite *RelationshipTestSuite) TestAcceptFollowRequestNoNotification() { ctx := context.Background() account := suite.testAccounts["admin_account"] targetAccount := suite.testAccounts["local_account_2"] followRequest := >smodel.FollowRequest{ ID: "01GEF753FWHCHRDWR0QEHBXM8W", URI: "http://localhost:8080/weeeeeeeeeeeeeeeee", AccountID: account.ID, TargetAccountID: targetAccount.ID, } if err := suite.db.Put(ctx, followRequest); err != nil { suite.FailNow(err.Error()) } // Unlike the above test, don't create a notification. // Follow request accept should still produce no error. follow, err := suite.db.AcceptFollowRequest(ctx, account.ID, targetAccount.ID) suite.NoError(err) suite.NotNil(follow) suite.Equal(followRequest.URI, follow.URI) } func (suite *RelationshipTestSuite) TestAcceptFollowRequestNotExisting() { ctx := context.Background() account := suite.testAccounts["admin_account"] targetAccount := suite.testAccounts["local_account_2"] follow, err := suite.db.AcceptFollowRequest(ctx, account.ID, targetAccount.ID) suite.ErrorIs(err, db.ErrNoEntries) suite.Nil(follow) } func (suite *RelationshipTestSuite) TestAcceptFollowRequestFollowAlreadyExists() { ctx := context.Background() account := suite.testAccounts["local_account_1"] targetAccount := suite.testAccounts["admin_account"] // follow already exists in the db from local_account_1 -> admin_account existingFollow := >smodel.Follow{} if err := suite.db.GetByID(ctx, suite.testFollows["local_account_1_admin_account"].ID, existingFollow); err != nil { suite.FailNow(err.Error()) } followRequest := >smodel.FollowRequest{ ID: "01GEF753FWHCHRDWR0QEHBXM8W", URI: "http://localhost:8080/weeeeeeeeeeeeeeeee", AccountID: account.ID, TargetAccountID: targetAccount.ID, } if err := suite.db.Put(ctx, followRequest); err != nil { suite.FailNow(err.Error()) } follow, err := suite.db.AcceptFollowRequest(ctx, account.ID, targetAccount.ID) suite.NoError(err) suite.NotNil(follow) // uri should be equal to value of new/overlapping follow request suite.NotEqual(followRequest.URI, existingFollow.URI) suite.Equal(followRequest.URI, follow.URI) } func (suite *RelationshipTestSuite) TestRejectFollowRequestOK() { ctx := context.Background() account := suite.testAccounts["admin_account"] targetAccount := suite.testAccounts["local_account_2"] followRequest := >smodel.FollowRequest{ ID: "01GEF753FWHCHRDWR0QEHBXM8W", URI: "http://localhost:8080/weeeeeeeeeeeeeeeee", AccountID: account.ID, TargetAccountID: targetAccount.ID, } if err := suite.db.PutFollowRequest(ctx, followRequest); err != nil { suite.FailNow(err.Error()) } followRequestNotification := >smodel.Notification{ ID: "01GV8MY1Q9KX2ZSWN4FAQ3V1PB", OriginAccountID: account.ID, TargetAccountID: targetAccount.ID, NotificationType: gtsmodel.NotificationFollowRequest, } if err := suite.db.Put(ctx, followRequestNotification); err != nil { suite.FailNow(err.Error()) } err := suite.db.RejectFollowRequest(ctx, account.ID, targetAccount.ID) suite.NoError(err) // Ensure notification is deleted. notification, err := suite.db.GetNotificationByID(ctx, followRequestNotification.ID) suite.ErrorIs(err, db.ErrNoEntries) suite.Nil(notification) } func (suite *RelationshipTestSuite) TestRejectFollowRequestNotExisting() { ctx := context.Background() account := suite.testAccounts["admin_account"] targetAccount := suite.testAccounts["local_account_2"] err := suite.db.RejectFollowRequest(ctx, account.ID, targetAccount.ID) suite.NoError(err) } func (suite *RelationshipTestSuite) TestGetAccountFollowRequests() { ctx := context.Background() account := suite.testAccounts["admin_account"] targetAccount := suite.testAccounts["local_account_2"] followRequest := >smodel.FollowRequest{ ID: "01GEF753FWHCHRDWR0QEHBXM8W", URI: "http://localhost:8080/weeeeeeeeeeeeeeeee", AccountID: account.ID, TargetAccountID: targetAccount.ID, } if err := suite.db.Put(ctx, followRequest); err != nil { suite.FailNow(err.Error()) } followRequests, err := suite.db.GetAccountFollowRequests(ctx, targetAccount.ID, nil) suite.NoError(err) suite.Len(followRequests, 1) } func (suite *RelationshipTestSuite) TestGetAccountFollows() { account := suite.testAccounts["local_account_1"] follows, err := suite.db.GetAccountFollows(context.Background(), account.ID, nil) suite.NoError(err) suite.Len(follows, 2) } func (suite *RelationshipTestSuite) TestGetAccountFollowers() { account := suite.testAccounts["local_account_1"] follows, err := suite.db.GetAccountFollowers(context.Background(), account.ID, nil) suite.NoError(err) suite.Len(follows, 2) } func (suite *RelationshipTestSuite) TestUnfollowExisting() { originAccount := suite.testAccounts["local_account_1"] targetAccount := suite.testAccounts["admin_account"] follow, err := suite.db.GetFollow(context.Background(), originAccount.ID, targetAccount.ID) suite.NoError(err) suite.NotNil(follow) followID := follow.ID // We should have lists that this follow is a part of. lists, err := suite.db.GetListsContainingFollowID(context.Background(), followID) suite.NoError(err) suite.NotEmpty(lists) err = suite.db.DeleteFollowByID(context.Background(), followID) suite.NoError(err) follow, err = suite.db.GetFollow(context.Background(), originAccount.ID, targetAccount.ID) suite.EqualError(err, db.ErrNoEntries.Error()) suite.Nil(follow) // Lists containing this follow should return empty too. lists, err = suite.db.GetListsContainingFollowID(context.Background(), followID) suite.NoError(err) suite.Empty(lists) } func (suite *RelationshipTestSuite) TestGetFollowNotExisting() { originAccount := suite.testAccounts["local_account_1"] targetAccountID := "01GTVD9N484CZ6AM90PGGNY7GQ" follow, err := suite.db.GetFollow(context.Background(), originAccount.ID, targetAccountID) suite.EqualError(err, db.ErrNoEntries.Error()) suite.Nil(follow) } func (suite *RelationshipTestSuite) TestDeleteFollow() { ctx := context.Background() originAccount := suite.testAccounts["local_account_1"] targetAccount := suite.testAccounts["admin_account"] err := suite.db.DeleteFollow(ctx, originAccount.ID, targetAccount.ID) suite.NoError(err) follow, err := suite.db.GetFollow(ctx, originAccount.ID, targetAccount.ID) suite.EqualError(err, db.ErrNoEntries.Error()) suite.Nil(follow) } func (suite *RelationshipTestSuite) TestUnfollowRequestExisting() { ctx := context.Background() originAccount := suite.testAccounts["admin_account"] targetAccount := suite.testAccounts["local_account_2"] followRequest := >smodel.FollowRequest{ ID: "01GEF753FWHCHRDWR0QEHBXM8W", URI: "http://localhost:8080/weeeeeeeeeeeeeeeee", AccountID: originAccount.ID, TargetAccountID: targetAccount.ID, } if err := suite.db.PutFollowRequest(ctx, followRequest); err != nil { suite.FailNow(err.Error()) } followRequest, err := suite.db.GetFollowRequest(context.Background(), originAccount.ID, targetAccount.ID) suite.NoError(err) suite.NotNil(followRequest) err = suite.db.DeleteFollowRequestByID(context.Background(), followRequest.ID) suite.NoError(err) followRequest, err = suite.db.GetFollowRequest(context.Background(), originAccount.ID, targetAccount.ID) suite.EqualError(err, db.ErrNoEntries.Error()) suite.Nil(followRequest) } func (suite *RelationshipTestSuite) TestUnfollowRequestNotExisting() { originAccount := suite.testAccounts["local_account_1"] targetAccountID := "01GTVD9N484CZ6AM90PGGNY7GQ" followRequest, err := suite.db.GetFollowRequest(context.Background(), originAccount.ID, targetAccountID) suite.EqualError(err, db.ErrNoEntries.Error()) suite.Nil(followRequest) } func (suite *RelationshipTestSuite) TestUpdateFollow() { ctx := context.Background() follow := >smodel.Follow{} *follow = *suite.testFollows["local_account_1_admin_account"] follow.Notify = util.Ptr(true) if err := suite.db.UpdateFollow(ctx, follow, "notify"); err != nil { suite.FailNow(err.Error()) } dbFollow, err := suite.db.GetFollowByID(ctx, follow.ID) if err != nil { suite.FailNow(err.Error()) } suite.True(*dbFollow.Notify) relationship, err := suite.db.GetRelationship(ctx, follow.AccountID, follow.TargetAccountID) if err != nil { suite.FailNow(err.Error()) } suite.True(relationship.Notifying) } func (suite *RelationshipTestSuite) TestGetNote() { ctx := context.Background() // Retrieve a fixture note account1 := suite.testAccounts["local_account_1"].ID account2 := suite.testAccounts["local_account_2"].ID expectedNote := suite.testAccountNotes["local_account_2_note_on_1"] note, err := suite.db.GetNote(ctx, account2, account1) suite.NoError(err) suite.NotNil(note) suite.Equal(expectedNote.ID, note.ID) suite.Equal(expectedNote.Comment, note.Comment) } func (suite *RelationshipTestSuite) TestPutNote() { ctx := context.Background() // put a note in account1 := suite.testAccounts["local_account_1"].ID account2 := suite.testAccounts["local_account_2"].ID err := suite.db.PutNote(ctx, >smodel.AccountNote{ ID: "01H539R2NA0M83JX15Y5RWKE97", AccountID: account1, TargetAccountID: account2, Comment: "foo", }) suite.NoError(err) // make sure the note is in the db note, err := suite.db.GetNote(ctx, account1, account2) suite.NoError(err) suite.NotNil(note) suite.Equal("01H539R2NA0M83JX15Y5RWKE97", note.ID) suite.Equal("foo", note.Comment) // update the note note.Comment = "bar" err = suite.db.PutNote(ctx, note) suite.NoError(err) // make sure the comment changes note, err = suite.db.GetNote(ctx, account1, account2) suite.NoError(err) suite.NotNil(note) suite.Equal("bar", note.Comment) } func TestRelationshipTestSuite(t *testing.T) { suite.Run(t, new(RelationshipTestSuite)) }