1 files changed, 15 insertions, 15 deletions
diff --git a/internal/processing/timeline/public_test.go b/internal/processing/timeline/public_test.go
index 5aa09f138..3320a45da 100644
--- a/internal/processing/timeline/public_test.go
+++ b/internal/processing/timeline/public_test.go
@@ -24,7 +24,6 @@ import (
 	"code.superseriousbusiness.org/gotosocial/internal/gtsmodel"
 	"code.superseriousbusiness.org/gotosocial/internal/id"
 	"code.superseriousbusiness.org/gotosocial/internal/paging"
-	"code.superseriousbusiness.org/gotosocial/internal/util"
 	"github.com/stretchr/testify/suite"
 )
 
@@ -110,24 +109,20 @@ func (suite *PublicTestSuite) TestPublicTimelineGetHideFiltered() {
 		filteredStatus      = suite.testStatuses["admin_account_status_2"]
 		filteredStatusFound = false
 		filterID            = id.NewULID()
-		filter              = &gtsmodel.Filter{
+		filterStatusID      = id.NewULID()
+		filterStatus        = &gtsmodel.FilterStatus{
+			ID:       filterStatusID,
+			FilterID: filterID,
+			StatusID: filteredStatus.ID,
+		}
+		filter = &gtsmodel.Filter{
 			ID:        filterID,
 			AccountID: requester.ID,
 			Title:     "timeline filtering test",
 			Action:    gtsmodel.FilterActionHide,
-			Statuses: []*gtsmodel.FilterStatus{
-				{
-					ID:        id.NewULID(),
-					AccountID: requester.ID,
-					FilterID:  filterID,
-					StatusID:  filteredStatus.ID,
-				},
-			},
-			ContextHome:          util.Ptr(false),
-			ContextNotifications: util.Ptr(false),
-			ContextPublic:        util.Ptr(true),
-			ContextThread:        util.Ptr(false),
-			ContextAccount:       util.Ptr(false),
+			Statuses:  []*gtsmodel.FilterStatus{filterStatus},
+			StatusIDs: []string{filterStatusID},
+			Contexts:  gtsmodel.FilterContexts(gtsmodel.FilterContextPublic),
 		}
 	)
 
@@ -153,6 +148,11 @@ func (suite *PublicTestSuite) TestPublicTimelineGetHideFiltered() {
 		suite.FailNow("precondition failed: status we would filter isn't present in unfiltered timeline")
 	}
 
+	// Create the filter status associated with the main filter.
+	if err := suite.db.PutFilterStatus(ctx, filterStatus); err != nil {
+		suite.FailNow(err.Error())
+	}
+
 	// Create a filter to hide one status on the timeline.
 	if err := suite.db.PutFilter(ctx, filter); err != nil {
 		suite.FailNow(err.Error())