lain/twitter-scrapper
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Separate test package

Browse source
This commit is contained in:
Alexander Sheiko 2021-12-07 10:18:01 +02:00
parent 55a0c83804
commit 4c9f06806a
7 changed files with 58 additions and 44 deletions
Download patch file Download diff file Expand all files Collapse all files

27
tweets_test.go
View file

@ -1,4 +1,4 @@
package twitterscraper
package twitterscraper_test
import (
"context"
@ -7,19 +7,20 @@ import (
"github.com/google/go-cmp/cmp"
"github.com/google/go-cmp/cmp/cmpopts"
twitterscraper "github.com/n0madic/twitter-scraper"
)
var cmpOptions = cmp.Options{
cmpopts.IgnoreFields(Tweet{}, "Likes"),
cmpopts.IgnoreFields(Tweet{}, "Replies"),
cmpopts.IgnoreFields(Tweet{}, "Retweets"),
cmpopts.IgnoreFields(twitterscraper.Tweet{}, "Likes"),
cmpopts.IgnoreFields(twitterscraper.Tweet{}, "Replies"),
cmpopts.IgnoreFields(twitterscraper.Tweet{}, "Retweets"),
}
func TestGetTweets(t *testing.T) {
count := 0
maxTweetsNbr := 300
dupcheck := make(map[string]bool)
for tweet := range GetTweets(context.Background(), "Twitter", maxTweetsNbr) {
for tweet := range twitterscraper.GetTweets(context.Background(), "Twitter", maxTweetsNbr) {
if tweet.Error != nil {
t.Error(tweet.Error)
} else {
@ -70,7 +71,7 @@ func TestGetTweets(t *testing.T) {
}
func TestGetTweet(t *testing.T) {
sample := Tweet{
sample := twitterscraper.Tweet{
HTML: "That thing you didn’t Tweet but wanted to but didn’t but got so close but then were like nah. <br><br>We have a place for that now—Fleets! <br><br>Rolling out to everyone starting today. <br><a href=\"https://t.co/auQAHXZMfH\"><img src=\"https://pbs.twimg.com/amplify_video_thumb/1328684333599756289/img/cP5KwbIXbGunNSBy.jpg\"/></a>",
ID: "1328684389388185600",
PermanentURL: "https://twitter.com/Twitter/status/1328684389388185600",
@ -80,13 +81,13 @@ func TestGetTweet(t *testing.T) {
Timestamp: 1605618018,
UserID: "783214",
Username: "Twitter",
Videos: []Video{{
Videos: []twitterscraper.Video{{
ID: "1328684333599756289",
Preview: "https://pbs.twimg.com/amplify_video_thumb/1328684333599756289/img/cP5KwbIXbGunNSBy.jpg",
URL: "https://video.twimg.com/amplify_video/1328684333599756289/vid/960x720/PcL8yv8KhgQ48Qpt.mp4?tag=13",
}},
}
tweet, err := defaultScraper.GetTweet("1328684389388185600")
tweet, err := twitterscraper.GetTweet("1328684389388185600")
if err != nil {
t.Error(err)
} else {
@ -97,7 +98,7 @@ func TestGetTweet(t *testing.T) {
}
func TestQuotedAndReply(t *testing.T) {
sample := &Tweet{
sample := &twitterscraper.Tweet{
HTML: "The Easiest Problem Everyone Gets Wrong <br><br>[new video] --&gt; <a href=\"https://youtu.be/ytfCdqWhmdg\">https://t.co/YdaeDYmPAU</a> <br><a href=\"https://t.co/iKu4Xs6o2V\"><img src=\"https://pbs.twimg.com/media/ESsZa9AXgAIAYnF.jpg\"/></a>",
ID: "1237110546383724547",
Likes: 485,
@ -112,7 +113,7 @@ func TestQuotedAndReply(t *testing.T) {
UserID: "978944851",
Username: "VsauceTwo",
}
tweet, err := defaultScraper.GetTweet("1237110897597976576")
tweet, err := twitterscraper.GetTweet("1237110897597976576")
if err != nil {
t.Error(err)
} else {
@ -123,7 +124,7 @@ func TestQuotedAndReply(t *testing.T) {
t.Error("Resulting quote does not match the sample", diff)
}
}
tweet, err = defaultScraper.GetTweet("1237111868445134850")
tweet, err = twitterscraper.GetTweet("1237111868445134850")
if err != nil {
t.Error(err)
} else {
@ -137,7 +138,7 @@ func TestQuotedAndReply(t *testing.T) {
}
func TestRetweet(t *testing.T) {
sample := &Tweet{
sample := &twitterscraper.Tweet{
HTML: "We’ve seen an increase in attacks against Asian communities and individuals around the world. It’s important to know that this isn’t new; throughout history, Asians have experienced violence and exclusion. However, their diverse lived experiences have largely been overlooked.",
ID: "1359151057872580612",
Likes: 6683,
@ -150,7 +151,7 @@ func TestRetweet(t *testing.T) {
UserID: "773578328498372608",
Username: "TwitterTogether",
}
tweet, err := defaultScraper.GetTweet("1362849141248974853")
tweet, err := twitterscraper.GetTweet("1362849141248974853")
if err != nil {
t.Error(err)
} else {