From f264aff265eb9d89d0dac2a2e4746bdcb5351b6f Mon Sep 17 00:00:00 2001
From: Alessio <me@playfulpachyderm.com>
Date: Sun, 30 Jan 2022 15:16:04 -0800
Subject: [PATCH] Update test data since the previous private-user unprivated

---
 cmd/tests.sh                |  4 ++--
 scraper/tweet_trove_test.go | 45 +++++++++++++++++++++++++++++++++++++
 2 files changed, 47 insertions(+), 2 deletions(-)
 create mode 100644 scraper/tweet_trove_test.go

diff --git a/cmd/tests.sh b/cmd/tests.sh
index c658a7a..157e7d8 100755
--- a/cmd/tests.sh
+++ b/cmd/tests.sh
@@ -151,8 +151,8 @@ test $(($rts_count + $tweet_count_2 - $tweet_count_1)) -gt "50"
 
 
 # Fetch a privated user
-tw fetch_user HbdNrx
-test $(sqlite3 twitter.db "select is_private from users where handle = 'HbdNrx'") = "1"
+tw fetch_user LandsharkRides
+test $(sqlite3 twitter.db "select is_private from users where handle = 'LandsharkRides'") = "1"
 
 
 # Test tweets with URLs
diff --git a/scraper/tweet_trove_test.go b/scraper/tweet_trove_test.go
new file mode 100644
index 0000000..c6ba01e
--- /dev/null
+++ b/scraper/tweet_trove_test.go
@@ -0,0 +1,45 @@
+package scraper_test
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+
+	. "offline_twitter/scraper"
+)
+
+func TestMergeTweetTroves(t *testing.T) {
+	assert := assert.New(t)
+	t1 := Tweet{Text: "1"}
+	t2 := Tweet{Text: "2"}
+	t3 := Tweet{Text: "3"}
+
+	u1 := User{Handle: "1"}
+	u2 := User{Handle: "2"}
+
+	r1 := Retweet{TweetID: 1}
+	r2 := Retweet{TweetID: 2}
+	r3 := Retweet{TweetID: 3}
+
+	trove1 := NewTweetTrove()
+	trove1.Tweets[1] = t1
+	trove1.Tweets[2] = t2
+
+	trove1.Retweets[1] = r1
+
+	trove2 := NewTweetTrove()
+	trove2.Tweets[3] = t3
+
+	trove2.Users[1] = u1
+	trove2.Users[2] = u2
+
+	trove2.Retweets[2] = r2
+	trove2.Retweets[3] = r3
+
+
+	trove1.MergeWith(trove2)
+
+	assert.Equal(3, len(trove1.Tweets))
+	assert.Equal(2, len(trove1.Users))
+	assert.Equal(3, len(trove1.Retweets))
+}