74 lines
1.5 KiB
Go
74 lines
1.5 KiB
Go
package reddit
|
|
|
|
import (
|
|
"fmt"
|
|
"net/http"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
var expectedFlairs = []Flair{
|
|
{
|
|
ID: "b8a1c822-3feb-11e8-88e1-0e5f55d58ce0",
|
|
Text: "Beginner",
|
|
Type: "text",
|
|
CSS: "Beginner1",
|
|
},
|
|
{
|
|
ID: "b8ea0fce-3feb-11e8-af7a-0e263a127cf8",
|
|
Text: "Beginner",
|
|
Type: "text",
|
|
CSS: "Beginner2",
|
|
},
|
|
}
|
|
|
|
var expectedFlairsV2 = []FlairV2{
|
|
{
|
|
ID: "b8a1c822-3feb-11e8-88e1-0e5f55d58ce0",
|
|
Text: "Beginner",
|
|
Type: "text",
|
|
CSS: "Beginner1",
|
|
ModOnly: false,
|
|
},
|
|
{
|
|
ID: "b8ea0fce-3feb-11e8-af7a-0e263a127cf8",
|
|
Text: "Moderator",
|
|
Type: "text",
|
|
CSS: "Moderator1",
|
|
ModOnly: true,
|
|
},
|
|
}
|
|
|
|
func TestFlairService_GetFlairs(t *testing.T) {
|
|
setup()
|
|
defer teardown()
|
|
|
|
blob := readFileContents(t, "testdata/flairs.json")
|
|
|
|
mux.HandleFunc("/r/subreddit/api/user_flair", func(w http.ResponseWriter, r *http.Request) {
|
|
assert.Equal(t, http.MethodGet, r.Method)
|
|
fmt.Fprint(w, blob)
|
|
})
|
|
|
|
flairs, _, err := client.Flair.GetFromSubreddit(ctx, "subreddit")
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, expectedFlairs, flairs)
|
|
}
|
|
|
|
func TestFlairService_GetFlairsV2(t *testing.T) {
|
|
setup()
|
|
defer teardown()
|
|
|
|
blob := readFileContents(t, "testdata/flairs-v2.json")
|
|
|
|
mux.HandleFunc("/r/subreddit/api/user_flair_v2", func(w http.ResponseWriter, r *http.Request) {
|
|
assert.Equal(t, http.MethodGet, r.Method)
|
|
fmt.Fprint(w, blob)
|
|
})
|
|
|
|
flairs, _, err := client.Flair.GetFromSubredditV2(ctx, "subreddit")
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, expectedFlairsV2, flairs)
|
|
}
|