diff --git a/scraper/tweet_trove.go b/scraper/tweet_trove.go index 5ebe711..298e688 100644 --- a/scraper/tweet_trove.go +++ b/scraper/tweet_trove.go @@ -4,6 +4,8 @@ type TweetTrove struct { Tweets map[TweetID]Tweet Users map[UserID]User Retweets map[TweetID]Retweet + + TombstoneUsers []UserHandle } func NewTweetTrove() TweetTrove { @@ -11,6 +13,7 @@ func NewTweetTrove() TweetTrove { ret.Tweets = make(map[TweetID]Tweet) ret.Users = make(map[UserID]User) ret.Retweets = make(map[TweetID]Retweet) + ret.TombstoneUsers = []UserHandle{} return ret } @@ -45,4 +48,6 @@ func (t1 *TweetTrove) MergeWith(t2 TweetTrove) { for id, val := range t2.Retweets { t1.Retweets[id] = val } + + t1.TombstoneUsers = append(t1.TombstoneUsers, t2.TombstoneUsers...) } diff --git a/scraper/tweet_trove_test.go b/scraper/tweet_trove_test.go index c6ba01e..67e667c 100644 --- a/scraper/tweet_trove_test.go +++ b/scraper/tweet_trove_test.go @@ -27,6 +27,8 @@ func TestMergeTweetTroves(t *testing.T) { trove1.Retweets[1] = r1 + trove1.TombstoneUsers = []UserHandle{"a", "b"} + trove2 := NewTweetTrove() trove2.Tweets[3] = t3 @@ -36,10 +38,12 @@ func TestMergeTweetTroves(t *testing.T) { trove2.Retweets[2] = r2 trove2.Retweets[3] = r3 + trove2.TombstoneUsers = []UserHandle{"c"} trove1.MergeWith(trove2) assert.Equal(3, len(trove1.Tweets)) assert.Equal(2, len(trove1.Users)) assert.Equal(3, len(trove1.Retweets)) + assert.Equal(3, len(trove1.TombstoneUsers)) }