From f5c004d67d4ed66b6c6df100afec47174aa14ae0 Mon Sep 17 00:00:00 2001
From: tobi <31960611+tsmethurst@users.noreply.github.com>
Date: Thu, 25 May 2023 10:37:38 +0200
Subject: [feature] Add List functionality (#1802)

* start working on lists

* further list work

* test list db functions nicely

* more work on lists

* peepoopeepoo

* poke

* start list timeline func

* we're getting there lads

* couldn't be me working on stuff... could it?

* hook up handlers

* fiddling

* weeee

* woah

* screaming, pissing

* fix streaming being a whiny baby

* lint, small test fix, swagger

* tidying up, testing

* fucked! by the linter

* move timelines to state like a boss

* add timeline start to tests using state

* invalidate lists
---
 internal/db/bundb/relationship_test.go | 15 +++++++++++++--
 1 file changed, 13 insertions(+), 2 deletions(-)

(limited to 'internal/db/bundb/relationship_test.go')

diff --git a/internal/db/bundb/relationship_test.go b/internal/db/bundb/relationship_test.go
index 0e38d19fe..63fdb9632 100644
--- a/internal/db/bundb/relationship_test.go
+++ b/internal/db/bundb/relationship_test.go
@@ -807,16 +807,27 @@ func (suite *RelationshipTestSuite) TestUnfollowExisting() {
 	follow, err := suite.db.GetFollow(context.Background(), originAccount.ID, targetAccount.ID)
 	suite.NoError(err)
 	suite.NotNil(follow)
+	followID := follow.ID
 
-	err = suite.db.DeleteFollowByID(context.Background(), follow.ID)
+	// We should have list entries for this follow.
+	listEntries, err := suite.db.GetListEntriesForFollowID(context.Background(), followID)
+	suite.NoError(err)
+	suite.NotEmpty(listEntries)
+
+	err = suite.db.DeleteFollowByID(context.Background(), followID)
 	suite.NoError(err)
 
 	follow, err = suite.db.GetFollow(context.Background(), originAccount.ID, targetAccount.ID)
 	suite.EqualError(err, db.ErrNoEntries.Error())
 	suite.Nil(follow)
+
+	// ListEntries pertaining to this follow should be deleted too.
+	listEntries, err = suite.db.GetListEntriesForFollowID(context.Background(), followID)
+	suite.NoError(err)
+	suite.Empty(listEntries)
 }
 
-func (suite *RelationshipTestSuite) TestUnfollowNotExisting() {
+func (suite *RelationshipTestSuite) TestGetFollowNotExisting() {
 	originAccount := suite.testAccounts["local_account_1"]
 	targetAccountID := "01GTVD9N484CZ6AM90PGGNY7GQ"
 
-- 
cgit v1.2.3