snoobert/subreddit_test.go

779 lines
20 KiB
Go
Raw Normal View History

package reddit
import (
"fmt"
"net/http"
"net/url"
"testing"
"time"
"github.com/stretchr/testify/assert"
)
var expectedPosts = &Posts{
Posts: []*Post{
{
ID: "agi5zf",
FullID: "t3_agi5zf",
Created: &Timestamp{time.Date(2019, 1, 16, 5, 57, 51, 0, time.UTC)},
Edited: &Timestamp{time.Date(1, 1, 1, 0, 0, 0, 0, time.UTC)},
Permalink: "https://www.reddit.com/r/test/comments/agi5zf/test/",
URL: "https://www.reddit.com/r/test/comments/agi5zf/test/",
Title: "test",
Body: "test",
Score: 253,
UpvoteRatio: 0.99,
NumberOfComments: 1634,
SubredditID: "t5_2qh23",
SubredditName: "test",
SubredditNamePrefixed: "r/test",
AuthorID: "t2_30a5ktgt",
AuthorName: "kmiller0112",
IsSelfPost: true,
Stickied: true,
},
{
ID: "hyhquk",
FullID: "t3_hyhquk",
Created: &Timestamp{time.Date(2020, 7, 27, 0, 5, 10, 0, time.UTC)},
Edited: &Timestamp{time.Date(1, 1, 1, 0, 0, 0, 0, time.UTC)},
Permalink: "https://www.reddit.com/r/test/comments/hyhquk/veggies/",
URL: "https://i.imgur.com/LrN2mPw.jpg",
Title: "Veggies",
Score: 4,
UpvoteRatio: 1,
NumberOfComments: 0,
SubredditID: "t5_2qh23",
SubredditName: "test",
SubredditNamePrefixed: "r/test",
AuthorID: "t2_6fqntbwq",
AuthorName: "MuckleMcDuckle",
},
},
After: "t3_hyhquk",
Before: "",
}
var expectedSubreddit = &Subreddit{
ID: "2rc7j",
FullID: "t5_2rc7j",
Created: &Timestamp{time.Date(2009, 11, 11, 0, 54, 28, 0, time.UTC)},
URL: "/r/golang/",
Name: "golang",
NamePrefixed: "r/golang",
Title: "The Go Programming Language",
Description: "Ask questions and post articles about the Go programming language and related tools, events etc.",
Type: "public",
Subscribers: 116532,
ActiveUserCount: Int(386),
NSFW: false,
UserIsMod: false,
}
var expectedSubreddits = &Subreddits{
After: "t5_2qh0u",
Before: "",
Subreddits: []*Subreddit{
{
ID: "2qs0k",
FullID: "t5_2qs0k",
Created: &Timestamp{time.Date(2009, 1, 25, 2, 25, 57, 0, time.UTC)},
URL: "/r/Home/",
Name: "Home",
NamePrefixed: "r/Home",
Title: "Home",
Type: "public",
Subscribers: 15336,
NSFW: false,
UserIsMod: false,
Favorite: false,
},
{
ID: "2qh1i",
FullID: "t5_2qh1i",
Created: &Timestamp{time.Date(2008, 1, 25, 3, 52, 15, 0, time.UTC)},
URL: "/r/AskReddit/",
Name: "AskReddit",
NamePrefixed: "r/AskReddit",
Title: "Ask Reddit...",
Description: "r/AskReddit is the place to ask and answer thought-provoking questions.",
Type: "public",
Subscribers: 28449174,
NSFW: false,
UserIsMod: false,
Favorite: true,
},
{
ID: "2qh0u",
FullID: "t5_2qh0u",
Created: &Timestamp{time.Date(2008, 1, 25, 0, 31, 9, 0, time.UTC)},
URL: "/r/pics/",
Name: "pics",
NamePrefixed: "r/pics",
Title: "Reddit Pics",
Description: "A place for pictures and photographs.",
Type: "public",
Subscribers: 24987753,
NSFW: false,
UserIsMod: false,
Favorite: false,
},
},
}
var expectedSubredditNames = []string{
"golang",
"golang_infosec",
"GolangJobOfferings",
"golanguage",
"golang2",
"GolangUnofficial",
"golang_jobs",
}
var expectedSearchPosts = &Posts{
Posts: []*Post{
{
ID: "hybow9",
FullID: "t3_hybow9",
Created: &Timestamp{time.Date(2020, 7, 26, 18, 14, 24, 0, time.UTC)},
Edited: &Timestamp{time.Date(1, 1, 1, 0, 0, 0, 0, time.UTC)},
Permalink: "https://www.reddit.com/r/WatchPeopleDieInside/comments/hybow9/pregnancy_test/",
URL: "https://v.redd.it/ra4qnt8bt8d51",
Title: "Pregnancy test",
Score: 103829,
UpvoteRatio: 0.88,
NumberOfComments: 3748,
SubredditID: "t5_3h4zq",
SubredditName: "WatchPeopleDieInside",
SubredditNamePrefixed: "r/WatchPeopleDieInside",
AuthorID: "t2_3p32m02",
AuthorName: "chocolat_ice_cream",
},
{
ID: "hmwhd7",
FullID: "t3_hmwhd7",
Created: &Timestamp{time.Date(2020, 7, 7, 15, 19, 42, 0, time.UTC)},
Edited: &Timestamp{time.Date(1, 1, 1, 0, 0, 0, 0, time.UTC)},
Permalink: "https://www.reddit.com/r/worldnews/comments/hmwhd7/brazilian_president_jair_bolsonaro_tests_positive/",
URL: "https://www.theguardian.com/world/2020/jul/07/jair-bolsonaro-coronavirus-positive-test-brazil-president",
Title: "Brazilian president Jair Bolsonaro tests positive for coronavirus",
Score: 149238,
UpvoteRatio: 0.94,
NumberOfComments: 7415,
SubredditID: "t5_2qh13",
SubredditName: "worldnews",
SubredditNamePrefixed: "r/worldnews",
AuthorID: "t2_wgrkg",
AuthorName: "Jeremy_Martin",
},
},
After: "t3_hmwhd7",
}
var expectedModerators = []Moderator{
{ID: "t2_test1", Name: "testuser1", Permissions: []string{"all"}},
{ID: "t2_test2", Name: "testuser2", Permissions: []string{"all"}},
}
var expectedRandomSubreddit = &Subreddit{
FullID: "t5_2wi4l",
Created: &Timestamp{time.Date(2013, 3, 1, 4, 4, 18, 0, time.UTC)},
URL: "/r/GalaxyS8/",
Name: "GalaxyS8",
NamePrefixed: "r/GalaxyS8",
Title: "Samsung Galaxy S8",
Description: "The only place for news, discussion, photos, and everything else Samsung Galaxy S8.",
Type: "public",
Subscribers: 52357,
}
func TestSubredditService_Hot(t *testing.T) {
setup()
defer teardown()
blob, err := readFileContents("testdata/subreddit/posts.json")
assert.NoError(t, err)
mux.HandleFunc("/r/test/hot", func(w http.ResponseWriter, r *http.Request) {
assert.Equal(t, http.MethodGet, r.Method)
fmt.Fprint(w, blob)
})
posts, _, err := client.Subreddit.Hot(ctx, "test")
assert.NoError(t, err)
assert.Equal(t, expectedPosts, posts)
}
func TestSubredditService_New(t *testing.T) {
setup()
defer teardown()
blob, err := readFileContents("testdata/subreddit/posts.json")
assert.NoError(t, err)
mux.HandleFunc("/r/test/new", func(w http.ResponseWriter, r *http.Request) {
assert.Equal(t, http.MethodGet, r.Method)
fmt.Fprint(w, blob)
})
posts, _, err := client.Subreddit.New(ctx, "test")
assert.NoError(t, err)
assert.Equal(t, expectedPosts, posts)
}
func TestSubredditService_Rising(t *testing.T) {
setup()
defer teardown()
blob, err := readFileContents("testdata/subreddit/posts.json")
assert.NoError(t, err)
mux.HandleFunc("/r/test/rising", func(w http.ResponseWriter, r *http.Request) {
assert.Equal(t, http.MethodGet, r.Method)
fmt.Fprint(w, blob)
})
posts, _, err := client.Subreddit.Rising(ctx, "test")
assert.NoError(t, err)
assert.Equal(t, expectedPosts, posts)
}
func TestSubredditService_Controversial(t *testing.T) {
setup()
defer teardown()
blob, err := readFileContents("testdata/subreddit/posts.json")
assert.NoError(t, err)
mux.HandleFunc("/r/test/controversial", func(w http.ResponseWriter, r *http.Request) {
assert.Equal(t, http.MethodGet, r.Method)
fmt.Fprint(w, blob)
})
posts, _, err := client.Subreddit.Controversial(ctx, "test")
assert.NoError(t, err)
assert.Equal(t, expectedPosts, posts)
}
func TestSubredditService_Top(t *testing.T) {
setup()
defer teardown()
blob, err := readFileContents("testdata/subreddit/posts.json")
assert.NoError(t, err)
mux.HandleFunc("/r/test/top", func(w http.ResponseWriter, r *http.Request) {
assert.Equal(t, http.MethodGet, r.Method)
fmt.Fprint(w, blob)
})
posts, _, err := client.Subreddit.Top(ctx, "test")
assert.NoError(t, err)
assert.Equal(t, expectedPosts, posts)
}
func TestSubredditService_Get(t *testing.T) {
setup()
defer teardown()
blob, err := readFileContents("testdata/subreddit/about.json")
assert.NoError(t, err)
mux.HandleFunc("/r/golang/about", func(w http.ResponseWriter, r *http.Request) {
assert.Equal(t, http.MethodGet, r.Method)
fmt.Fprint(w, blob)
})
_, _, err = client.Subreddit.Get(ctx, "")
assert.EqualError(t, err, "name: cannot be empty")
subreddit, _, err := client.Subreddit.Get(ctx, "golang")
assert.NoError(t, err)
assert.Equal(t, expectedSubreddit, subreddit)
}
func TestSubredditService_GetPopular(t *testing.T) {
setup()
defer teardown()
blob, err := readFileContents("testdata/subreddit/list.json")
assert.NoError(t, err)
mux.HandleFunc("/subreddits/popular", func(w http.ResponseWriter, r *http.Request) {
assert.Equal(t, http.MethodGet, r.Method)
fmt.Fprint(w, blob)
})
subreddits, _, err := client.Subreddit.GetPopular(ctx, nil)
assert.NoError(t, err)
assert.Equal(t, expectedSubreddits, subreddits)
}
func TestSubredditService_GetNew(t *testing.T) {
setup()
defer teardown()
blob, err := readFileContents("testdata/subreddit/list.json")
assert.NoError(t, err)
mux.HandleFunc("/subreddits/new", func(w http.ResponseWriter, r *http.Request) {
assert.Equal(t, http.MethodGet, r.Method)
fmt.Fprint(w, blob)
})
subreddits, _, err := client.Subreddit.GetNew(ctx, nil)
assert.NoError(t, err)
assert.Equal(t, expectedSubreddits, subreddits)
}
func TestSubredditService_GetGold(t *testing.T) {
setup()
defer teardown()
blob, err := readFileContents("testdata/subreddit/list.json")
assert.NoError(t, err)
mux.HandleFunc("/subreddits/gold", func(w http.ResponseWriter, r *http.Request) {
assert.Equal(t, http.MethodGet, r.Method)
fmt.Fprint(w, blob)
})
subreddits, _, err := client.Subreddit.GetGold(ctx, nil)
assert.NoError(t, err)
assert.Equal(t, expectedSubreddits, subreddits)
}
func TestSubredditService_GetDefault(t *testing.T) {
setup()
defer teardown()
blob, err := readFileContents("testdata/subreddit/list.json")
assert.NoError(t, err)
mux.HandleFunc("/subreddits/default", func(w http.ResponseWriter, r *http.Request) {
assert.Equal(t, http.MethodGet, r.Method)
fmt.Fprint(w, blob)
})
subreddits, _, err := client.Subreddit.GetDefault(ctx, nil)
assert.NoError(t, err)
assert.Equal(t, expectedSubreddits, subreddits)
}
func TestSubredditService_GetSubscribed(t *testing.T) {
setup()
defer teardown()
blob, err := readFileContents("testdata/subreddit/list.json")
assert.NoError(t, err)
mux.HandleFunc("/subreddits/mine/subscriber", func(w http.ResponseWriter, r *http.Request) {
assert.Equal(t, http.MethodGet, r.Method)
fmt.Fprint(w, blob)
})
subreddits, _, err := client.Subreddit.GetSubscribed(ctx, nil)
assert.NoError(t, err)
assert.Equal(t, expectedSubreddits, subreddits)
}
func TestSubredditService_GetApproved(t *testing.T) {
setup()
defer teardown()
blob, err := readFileContents("testdata/subreddit/list.json")
assert.NoError(t, err)
mux.HandleFunc("/subreddits/mine/contributor", func(w http.ResponseWriter, r *http.Request) {
assert.Equal(t, http.MethodGet, r.Method)
fmt.Fprint(w, blob)
})
subreddits, _, err := client.Subreddit.GetApproved(ctx, nil)
assert.NoError(t, err)
assert.Equal(t, expectedSubreddits, subreddits)
}
func TestSubredditService_GetModerated(t *testing.T) {
setup()
defer teardown()
blob, err := readFileContents("testdata/subreddit/list.json")
assert.NoError(t, err)
mux.HandleFunc("/subreddits/mine/moderator", func(w http.ResponseWriter, r *http.Request) {
assert.Equal(t, http.MethodGet, r.Method)
fmt.Fprint(w, blob)
})
subreddits, _, err := client.Subreddit.GetModerated(ctx, nil)
assert.NoError(t, err)
assert.Equal(t, expectedSubreddits, subreddits)
}
func TestSubredditService_GetSticky1(t *testing.T) {
setup()
defer teardown()
blob, err := readFileContents("testdata/post/post.json")
assert.NoError(t, err)
mux.HandleFunc("/r/test/about/sticky", func(w http.ResponseWriter, r *http.Request) {
assert.Equal(t, http.MethodGet, r.Method)
err := r.ParseForm()
assert.NoError(t, err)
assert.Equal(t, "1", r.Form.Get("num"))
fmt.Fprint(w, blob)
})
postAndComments, _, err := client.Subreddit.GetSticky1(ctx, "test")
assert.NoError(t, err)
assert.Equal(t, expectedPostAndComments, postAndComments)
}
func TestSubredditService_GetSticky2(t *testing.T) {
setup()
defer teardown()
blob, err := readFileContents("testdata/post/post.json")
assert.NoError(t, err)
mux.HandleFunc("/r/test/about/sticky", func(w http.ResponseWriter, r *http.Request) {
assert.Equal(t, http.MethodGet, r.Method)
err := r.ParseForm()
assert.NoError(t, err)
assert.Equal(t, "2", r.Form.Get("num"))
fmt.Fprint(w, blob)
})
postAndComments, _, err := client.Subreddit.GetSticky2(ctx, "test")
assert.NoError(t, err)
assert.Equal(t, expectedPostAndComments, postAndComments)
}
func TestSubredditService_Subscribe(t *testing.T) {
setup()
defer teardown()
mux.HandleFunc("/api/subscribe", func(w http.ResponseWriter, r *http.Request) {
assert.Equal(t, http.MethodPost, r.Method)
form := url.Values{}
form.Set("action", "sub")
form.Set("sr_name", "test,golang,nba")
err := r.ParseForm()
assert.NoError(t, err)
assert.Equal(t, form, r.Form)
})
_, err := client.Subreddit.Subscribe(ctx, "test", "golang", "nba")
assert.NoError(t, err)
}
func TestSubredditService_SubscribeByID(t *testing.T) {
setup()
defer teardown()
mux.HandleFunc("/api/subscribe", func(w http.ResponseWriter, r *http.Request) {
assert.Equal(t, http.MethodPost, r.Method)
form := url.Values{}
form.Set("action", "sub")
form.Set("sr", "t5_test1,t5_test2,t5_test3")
err := r.ParseForm()
assert.NoError(t, err)
assert.Equal(t, form, r.Form)
})
_, err := client.Subreddit.SubscribeByID(ctx, "t5_test1", "t5_test2", "t5_test3")
assert.NoError(t, err)
}
func TestSubredditService_Unsubscribe(t *testing.T) {
setup()
defer teardown()
mux.HandleFunc("/api/subscribe", func(w http.ResponseWriter, r *http.Request) {
assert.Equal(t, http.MethodPost, r.Method)
form := url.Values{}
form.Set("action", "unsub")
form.Set("sr_name", "test,golang,nba")
err := r.ParseForm()
assert.NoError(t, err)
assert.Equal(t, form, r.Form)
})
_, err := client.Subreddit.Unsubscribe(ctx, "test", "golang", "nba")
assert.NoError(t, err)
}
func TestSubredditService_UnsubscribeByID(t *testing.T) {
setup()
defer teardown()
mux.HandleFunc("/api/subscribe", func(w http.ResponseWriter, r *http.Request) {
assert.Equal(t, http.MethodPost, r.Method)
form := url.Values{}
form.Set("action", "unsub")
form.Set("sr", "t5_test1,t5_test2,t5_test3")
err := r.ParseForm()
assert.NoError(t, err)
assert.Equal(t, form, r.Form)
})
_, err := client.Subreddit.UnsubscribeByID(ctx, "t5_test1", "t5_test2", "t5_test3")
assert.NoError(t, err)
}
func TestSubredditService_Search(t *testing.T) {
setup()
defer teardown()
blob, err := readFileContents("testdata/subreddit/list.json")
assert.NoError(t, err)
mux.HandleFunc("/subreddits/search", func(w http.ResponseWriter, r *http.Request) {
assert.Equal(t, http.MethodGet, r.Method)
form := url.Values{}
form.Set("q", "golang")
err := r.ParseForm()
assert.NoError(t, err)
assert.Equal(t, form, r.Form)
fmt.Fprint(w, blob)
})
subreddits, _, err := client.Subreddit.Search(ctx, "golang")
assert.NoError(t, err)
assert.Equal(t, expectedSubreddits, subreddits)
}
func TestSubredditService_SearchNames(t *testing.T) {
setup()
defer teardown()
blob, err := readFileContents("testdata/subreddit/search-names.json")
assert.NoError(t, err)
mux.HandleFunc("/api/search_reddit_names", func(w http.ResponseWriter, r *http.Request) {
assert.Equal(t, http.MethodGet, r.Method)
form := url.Values{}
form.Set("query", "golang")
err := r.ParseForm()
assert.NoError(t, err)
assert.Equal(t, form, r.Form)
fmt.Fprint(w, blob)
})
names, _, err := client.Subreddit.SearchNames(ctx, "golang")
assert.NoError(t, err)
assert.Equal(t, expectedSubredditNames, names)
}
func TestSubredditService_SearchPosts(t *testing.T) {
setup()
defer teardown()
blob, err := readFileContents("testdata/subreddit/search-posts.json")
assert.NoError(t, err)
mux.HandleFunc("/r/all/search", func(w http.ResponseWriter, r *http.Request) {
assert.Equal(t, http.MethodGet, r.Method)
form := url.Values{}
form.Set("q", "test")
form.Set("after", "t3_testpost")
err := r.ParseForm()
assert.NoError(t, err)
assert.Equal(t, form, r.Form)
fmt.Fprint(w, blob)
})
posts, _, err := client.Subreddit.SearchPosts(ctx, "test", "", SetAfter("t3_testpost"))
assert.NoError(t, err)
assert.Equal(t, expectedSearchPosts, posts)
}
func TestSubredditService_SearchPosts_InSubreddit(t *testing.T) {
setup()
defer teardown()
blob, err := readFileContents("testdata/subreddit/search-posts.json")
assert.NoError(t, err)
mux.HandleFunc("/r/test/search", func(w http.ResponseWriter, r *http.Request) {
assert.Equal(t, http.MethodGet, r.Method)
form := url.Values{}
form.Set("q", "test")
form.Set("restrict_sr", "true")
form.Set("after", "t3_testpost")
err := r.ParseForm()
assert.NoError(t, err)
assert.Equal(t, form, r.Form)
fmt.Fprint(w, blob)
})
posts, _, err := client.Subreddit.SearchPosts(ctx, "test", "test", SetAfter("t3_testpost"))
assert.NoError(t, err)
assert.Equal(t, expectedSearchPosts, posts)
}
func TestSubredditService_SearchPosts_InSubreddits(t *testing.T) {
setup()
defer teardown()
blob, err := readFileContents("testdata/subreddit/search-posts.json")
assert.NoError(t, err)
mux.HandleFunc("/r/test+golang+nba/search", func(w http.ResponseWriter, r *http.Request) {
assert.Equal(t, http.MethodGet, r.Method)
form := url.Values{}
form.Set("q", "test")
form.Set("restrict_sr", "true")
form.Set("after", "t3_testpost")
err := r.ParseForm()
assert.NoError(t, err)
assert.Equal(t, form, r.Form)
fmt.Fprint(w, blob)
})
posts, _, err := client.Subreddit.SearchPosts(ctx, "test", "test+golang+nba", SetAfter("t3_testpost"))
assert.NoError(t, err)
assert.Equal(t, expectedSearchPosts, posts)
}
func TestSubredditService_Moderators(t *testing.T) {
setup()
defer teardown()
blob, err := readFileContents("testdata/subreddit/moderators.json")
assert.NoError(t, err)
mux.HandleFunc("/r/test/about/moderators", func(w http.ResponseWriter, r *http.Request) {
assert.Equal(t, http.MethodGet, r.Method)
fmt.Fprint(w, blob)
})
moderators, _, err := client.Subreddit.Moderators(ctx, "test")
assert.NoError(t, err)
assert.Equal(t, expectedModerators, moderators)
}
func TestSubredditService_Random(t *testing.T) {
setup()
defer teardown()
blob, err := readFileContents("testdata/subreddit/random.json")
assert.NoError(t, err)
mux.HandleFunc("/r/random", func(w http.ResponseWriter, r *http.Request) {
assert.Equal(t, http.MethodGet, r.Method)
err := r.ParseForm()
assert.NoError(t, err)
assert.Equal(t, "true", r.Form.Get("sr_detail"))
assert.Equal(t, "1", r.Form.Get("limit"))
fmt.Fprint(w, blob)
})
subreddit, _, err := client.Subreddit.Random(ctx)
assert.NoError(t, err)
assert.Equal(t, expectedRandomSubreddit, subreddit)
}
func TestSubredditService_RandomNSFW(t *testing.T) {
setup()
defer teardown()
blob, err := readFileContents("testdata/subreddit/random.json")
assert.NoError(t, err)
mux.HandleFunc("/r/randnsfw", func(w http.ResponseWriter, r *http.Request) {
assert.Equal(t, http.MethodGet, r.Method)
err := r.ParseForm()
assert.NoError(t, err)
assert.Equal(t, "true", r.Form.Get("sr_detail"))
assert.Equal(t, "1", r.Form.Get("limit"))
fmt.Fprint(w, blob)
})
subreddit, _, err := client.Subreddit.RandomNSFW(ctx)
assert.NoError(t, err)
assert.Equal(t, expectedRandomSubreddit, subreddit)
}
func TestSubredditService_SubmissionText(t *testing.T) {
setup()
defer teardown()
mux.HandleFunc("/r/test/api/submit_text", func(w http.ResponseWriter, r *http.Request) {
assert.Equal(t, http.MethodGet, r.Method)
fmt.Fprint(w, `{
"submit_text": "this is a test",
"submit_text_html": ""
}`)
})
text, _, err := client.Subreddit.SubmissionText(ctx, "test")
assert.NoError(t, err)
assert.Equal(t, "this is a test", text)
}