Pg to bun (#148)

* start moving to bun

* changing more stuff

* more

* and yet more

* tests passing

* seems stable now

* more big changes

* small fix

* little fixes

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
 }