diff options
Diffstat (limited to 'internal/timeline/manager_test.go')
| -rw-r--r-- | internal/timeline/manager_test.go | 37 |
1 files changed, 19 insertions, 18 deletions
diff --git a/internal/timeline/manager_test.go b/internal/timeline/manager_test.go index 00c6dcb4a..ea4dc4c12 100644 --- a/internal/timeline/manager_test.go +++ b/internal/timeline/manager_test.go @@ -19,6 +19,7 @@ package timeline_test import ( + "context" "testing" "github.com/stretchr/testify/suite" @@ -54,85 +55,85 @@ func (suite *ManagerTestSuite) TestManagerIntegration() { testAccount := suite.testAccounts["local_account_1"] // should start at 0 - indexedLen := suite.manager.GetIndexedLength(testAccount.ID) + indexedLen := suite.manager.GetIndexedLength(context.Background(), testAccount.ID) suite.Equal(0, indexedLen) // oldestIndexed should be empty string since there's nothing indexed - oldestIndexed, err := suite.manager.GetOldestIndexedID(testAccount.ID) + oldestIndexed, err := suite.manager.GetOldestIndexedID(context.Background(), testAccount.ID) suite.NoError(err) suite.Empty(oldestIndexed) // trigger status preparation - err = suite.manager.PrepareXFromTop(testAccount.ID, 20) + err = suite.manager.PrepareXFromTop(context.Background(), testAccount.ID, 20) suite.NoError(err) // local_account_1 can see 12 statuses out of the testrig statuses in its home timeline - indexedLen = suite.manager.GetIndexedLength(testAccount.ID) + indexedLen = suite.manager.GetIndexedLength(context.Background(), testAccount.ID) suite.Equal(12, indexedLen) // oldest should now be set - oldestIndexed, err = suite.manager.GetOldestIndexedID(testAccount.ID) + oldestIndexed, err = suite.manager.GetOldestIndexedID(context.Background(), testAccount.ID) suite.NoError(err) suite.Equal("01F8MH75CBF9JFX4ZAD54N0W0R", oldestIndexed) // get hometimeline - statuses, err := suite.manager.HomeTimeline(testAccount.ID, "", "", "", 20, false) + statuses, err := suite.manager.HomeTimeline(context.Background(), testAccount.ID, "", "", "", 20, false) suite.NoError(err) suite.Len(statuses, 12) // now wipe the last status from all timelines, as though it had been deleted by the owner - err = suite.manager.WipeStatusFromAllTimelines("01F8MH75CBF9JFX4ZAD54N0W0R") + err = suite.manager.WipeStatusFromAllTimelines(context.Background(), "01F8MH75CBF9JFX4ZAD54N0W0R") suite.NoError(err) // timeline should be shorter - indexedLen = suite.manager.GetIndexedLength(testAccount.ID) + indexedLen = suite.manager.GetIndexedLength(context.Background(), testAccount.ID) suite.Equal(11, indexedLen) // oldest should now be different - oldestIndexed, err = suite.manager.GetOldestIndexedID(testAccount.ID) + oldestIndexed, err = suite.manager.GetOldestIndexedID(context.Background(), testAccount.ID) suite.NoError(err) suite.Equal("01F8MH82FYRXD2RC6108DAJ5HB", oldestIndexed) // delete the new oldest status specifically from this timeline, as though local_account_1 had muted or blocked it - removed, err := suite.manager.Remove(testAccount.ID, "01F8MH82FYRXD2RC6108DAJ5HB") + removed, err := suite.manager.Remove(context.Background(), testAccount.ID, "01F8MH82FYRXD2RC6108DAJ5HB") suite.NoError(err) suite.Equal(2, removed) // 1 status should be removed, but from both indexed and prepared, so 2 removals total // timeline should be shorter - indexedLen = suite.manager.GetIndexedLength(testAccount.ID) + indexedLen = suite.manager.GetIndexedLength(context.Background(), testAccount.ID) suite.Equal(10, indexedLen) // oldest should now be different - oldestIndexed, err = suite.manager.GetOldestIndexedID(testAccount.ID) + oldestIndexed, err = suite.manager.GetOldestIndexedID(context.Background(), testAccount.ID) suite.NoError(err) suite.Equal("01F8MHAAY43M6RJ473VQFCVH37", oldestIndexed) // now remove all entries by local_account_2 from the timeline - err = suite.manager.WipeStatusesFromAccountID(testAccount.ID, suite.testAccounts["local_account_2"].ID) + err = suite.manager.WipeStatusesFromAccountID(context.Background(), testAccount.ID, suite.testAccounts["local_account_2"].ID) suite.NoError(err) // timeline should be empty now - indexedLen = suite.manager.GetIndexedLength(testAccount.ID) + indexedLen = suite.manager.GetIndexedLength(context.Background(), testAccount.ID) suite.Equal(5, indexedLen) // ingest 1 into the timeline status1 := suite.testStatuses["admin_account_status_1"] - ingested, err := suite.manager.Ingest(status1, testAccount.ID) + ingested, err := suite.manager.Ingest(context.Background(), status1, testAccount.ID) suite.NoError(err) suite.True(ingested) // ingest and prepare another one into the timeline status2 := suite.testStatuses["local_account_2_status_1"] - ingested, err = suite.manager.IngestAndPrepare(status2, testAccount.ID) + ingested, err = suite.manager.IngestAndPrepare(context.Background(), status2, testAccount.ID) suite.NoError(err) suite.True(ingested) // timeline should be longer now - indexedLen = suite.manager.GetIndexedLength(testAccount.ID) + indexedLen = suite.manager.GetIndexedLength(context.Background(), testAccount.ID) suite.Equal(7, indexedLen) // try to ingest status 2 again - ingested, err = suite.manager.IngestAndPrepare(status2, testAccount.ID) + ingested, err = suite.manager.IngestAndPrepare(context.Background(), status2, testAccount.ID) suite.NoError(err) suite.False(ingested) // should be false since it's a duplicate } |