guy176251/reddit-go
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add streamer config, tests

Browse Source 
Signed-off-by: Vartan Benohanian <vartanbeno@gmail.com>
This commit is contained in:
Vartan Benohanian
2020-08-22 14:26:50 -04:00
parent 49fa672619
commit da689412a1
3 changed files with 401 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files
reddit/stream_test.go
+314
View File
@@ -0,0 +1,314 @@
package reddit
import (
"fmt"
"net/http"
"testing"
"time"
"github.com/stretchr/testify/require"
)
func TestStreamService_Posts(t *testing.T) {
setup()
defer teardown()
var counter int
mux.HandleFunc("/r/testsubreddit/new", func(w http.ResponseWriter, r *http.Request) {
require.Equal(t, http.MethodGet, r.Method)
defer func() { counter++ }()
switch counter {
case 0:
fmt.Fprint(w, `{
"kind": "Listing",
"data": {
"children": [
{
"kind": "t3",
"data": {
"name": "t3_post1"
}
},
{
"kind": "t3",
"data": {
"name": "t3_post2"
}
}
]
}
}`)
case 1:
fmt.Fprint(w, `{
"kind": "Listing",
"data": {
"children": [
{
"kind": "t3",
"data": {
"name": "t3_post3"
}
},
{
"kind": "t3",
"data": {
"name": "t3_post1"
}
}
]
}
}`)
case 2:
fmt.Fprint(w, `{
"kind": "Listing",
"data": {
"children": [
{
"kind": "t3",
"data": {
"name": "t3_post4"
}
},
{
"kind": "t3",
"data": {
"name": "t3_post5"
}
},
{
"kind": "t3",
"data": {
"name": "t3_post6"
}
}
]
}
}`)
case 3:
fmt.Fprint(w, `{
"kind": "Listing",
"data": {
"children": [
{
"kind": "t3",
"data": {
"name": "t3_post7"
}
},
{
"kind": "t3",
"data": {
"name": "t3_post8"
}
},
{
"kind": "t3",
"data": {
"name": "t3_post9"
}
},
{
"kind": "t3",
"data": {
"name": "t3_post10"
}
},
{
"kind": "t3",
"data": {
"name": "t3_post11"
}
},
{
"kind": "t3",
"data": {
"name": "t3_post12"
}
}
]
}
}`)
default:
fmt.Fprint(w, `{}`)
}
})
posts, errs, stop := client.Stream.Posts("testsubreddit", StreamInterval(time.Millisecond*10), StreamMaxRequests(4))
defer stop()
expectedPostIDs := []string{"t3_post1", "t3_post2", "t3_post3", "t3_post4", "t3_post5", "t3_post6", "t3_post7", "t3_post8", "t3_post9", "t3_post10", "t3_post11", "t3_post12"}
var i int
loop:
for i != len(expectedPostIDs) {
select {
case post, ok := <-posts:
if !ok {
break loop
}
require.Equal(t, expectedPostIDs[i], post.FullID)
case err, ok := <-errs:
if !ok {
break loop
}
require.NoError(t, err)
}
i++
}
require.Len(t, expectedPostIDs, i)
}
func TestStreamService_Posts_DiscardInitial(t *testing.T) {
setup()
defer teardown()
var counter int
mux.HandleFunc("/r/testsubreddit/new", func(w http.ResponseWriter, r *http.Request) {
require.Equal(t, http.MethodGet, r.Method)
defer func() { counter++ }()
switch counter {
case 0:
fmt.Fprint(w, `{
"kind": "Listing",
"data": {
"children": [
{
"kind": "t3",
"data": {
"name": "t3_post1"
}
},
{
"kind": "t3",
"data": {
"name": "t3_post2"
}
}
]
}
}`)
case 1:
fmt.Fprint(w, `{
"kind": "Listing",
"data": {
"children": [
{
"kind": "t3",
"data": {
"name": "t3_post3"
}
},
{
"kind": "t3",
"data": {
"name": "t3_post1"
}
}
]
}
}`)
case 2:
fmt.Fprint(w, `{
"kind": "Listing",
"data": {
"children": [
{
"kind": "t3",
"data": {
"name": "t3_post4"
}
},
{
"kind": "t3",
"data": {
"name": "t3_post5"
}
},
{
"kind": "t3",
"data": {
"name": "t3_post6"
}
}
]
}
}`)
case 3:
fmt.Fprint(w, `{
"kind": "Listing",
"data": {
"children": [
{
"kind": "t3",
"data": {
"name": "t3_post7"
}
},
{
"kind": "t3",
"data": {
"name": "t3_post8"
}
},
{
"kind": "t3",
"data": {
"name": "t3_post9"
}
},
{
"kind": "t3",
"data": {
"name": "t3_post10"
}
},
{
"kind": "t3",
"data": {
"name": "t3_post11"
}
},
{
"kind": "t3",
"data": {
"name": "t3_post12"
}
}
]
}
}`)
default:
fmt.Fprint(w, `{}`)
}
})
posts, errs, stop := client.Stream.Posts("testsubreddit", StreamInterval(time.Millisecond*10), StreamMaxRequests(4), StreamDiscardInitial)
defer stop()
expectedPostIDs := []string{"t3_post3", "t3_post4", "t3_post5", "t3_post6", "t3_post7", "t3_post8", "t3_post9", "t3_post10", "t3_post11", "t3_post12"}
var i int
loop:
for i != len(expectedPostIDs) {
select {
case post, ok := <-posts:
if !ok {
break loop
}
require.Equal(t, expectedPostIDs[i], post.FullID)
case err, ok := <-errs:
if !ok {
break loop
}
require.NoError(t, err)
}
i++
}
require.Len(t, expectedPostIDs, i)
}