diff options
Diffstat (limited to 'internal/processing/status/bookmark_test.go')
| -rw-r--r-- | internal/processing/status/bookmark_test.go | 22 |
1 files changed, 21 insertions, 1 deletions
diff --git a/internal/processing/status/bookmark_test.go b/internal/processing/status/bookmark_test.go index bfb652279..a05e19e8b 100644 --- a/internal/processing/status/bookmark_test.go +++ b/internal/processing/status/bookmark_test.go @@ -36,13 +36,33 @@ func (suite *StatusBookmarkTestSuite) TestBookmark() { bookmarkingAccount1 := suite.testAccounts["local_account_1"] targetStatus1 := suite.testStatuses["admin_account_status_1"] - bookmark1, err := suite.status.Bookmark(ctx, bookmarkingAccount1, targetStatus1.ID) + bookmark1, err := suite.status.BookmarkCreate(ctx, bookmarkingAccount1, targetStatus1.ID) suite.NoError(err) suite.NotNil(bookmark1) suite.True(bookmark1.Bookmarked) suite.Equal(targetStatus1.ID, bookmark1.ID) } +func (suite *StatusBookmarkTestSuite) TestUnbookmark() { + ctx := context.Background() + + // bookmark a status + bookmarkingAccount1 := suite.testAccounts["local_account_1"] + targetStatus1 := suite.testStatuses["admin_account_status_1"] + + bookmark1, err := suite.status.BookmarkCreate(ctx, bookmarkingAccount1, targetStatus1.ID) + suite.NoError(err) + suite.NotNil(bookmark1) + suite.True(bookmark1.Bookmarked) + suite.Equal(targetStatus1.ID, bookmark1.ID) + + bookmark2, err := suite.status.BookmarkRemove(ctx, bookmarkingAccount1, targetStatus1.ID) + suite.NoError(err) + suite.NotNil(bookmark2) + suite.False(bookmark2.Bookmarked) + suite.Equal(targetStatus1.ID, bookmark1.ID) +} + func TestStatusBookmarkTestSuite(t *testing.T) { suite.Run(t, new(StatusBookmarkTestSuite)) } |