Update test data since the previous private-user unprivated

This commit is contained in:
Alessio 2022-01-30 15:16:04 -08:00
parent 68a8549f34
commit f264aff265
2 changed files with 47 additions and 2 deletions

View File

@ -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

View File

@ -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))
}