diff options
Diffstat (limited to 'internal/db/bundb/relationship_test.go')
| -rw-r--r-- | internal/db/bundb/relationship_test.go | 28 |
1 files changed, 26 insertions, 2 deletions
diff --git a/internal/db/bundb/relationship_test.go b/internal/db/bundb/relationship_test.go index 00583d175..9e5a71d60 100644 --- a/internal/db/bundb/relationship_test.go +++ b/internal/db/bundb/relationship_test.go @@ -568,6 +568,14 @@ func (suite *RelationshipTestSuite) TestAcceptFollowRequestOK() { 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", @@ -575,10 +583,18 @@ func (suite *RelationshipTestSuite) TestAcceptFollowRequestOK() { TargetAccountID: targetAccount.ID, } - if err := suite.db.Put(ctx, followRequest); err != nil { + 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) + followRequestNotification := >smodel.Notification{ ID: "01GV8MY1Q9KX2ZSWN4FAQ3V1PB", OriginAccountID: account.ID, @@ -586,7 +602,7 @@ func (suite *RelationshipTestSuite) TestAcceptFollowRequestOK() { NotificationType: gtsmodel.NotificationFollowRequest, } - if err := suite.db.Put(ctx, followRequestNotification); err != nil { + if err := suite.db.PutNotification(ctx, followRequestNotification); err != nil { suite.FailNow(err.Error()) } @@ -599,6 +615,14 @@ func (suite *RelationshipTestSuite) TestAcceptFollowRequestOK() { 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() { |