114 lines
2.4 KiB
Go
114 lines
2.4 KiB
Go
package reddit
|
|
|
|
import (
|
|
"fmt"
|
|
"net/http"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
var expectedUserFlairs = []*Flair{
|
|
{
|
|
ID: "b8a1c822-3feb-11e8-88e1-0e5f55d58ce0",
|
|
Type: "text",
|
|
Text: "Beginner",
|
|
|
|
Color: "dark",
|
|
BackgroundColor: "",
|
|
CSSClass: "Beginner1",
|
|
|
|
Editable: false,
|
|
ModOnly: false,
|
|
},
|
|
{
|
|
ID: "b8ea0fce-3feb-11e8-af7a-0e263a127cf8",
|
|
Text: "Moderator",
|
|
Type: "text",
|
|
|
|
Color: "dark",
|
|
BackgroundColor: "",
|
|
CSSClass: "Moderator1",
|
|
|
|
Editable: false,
|
|
ModOnly: true,
|
|
},
|
|
}
|
|
|
|
var expectedPostFlairs = []*Flair{
|
|
{
|
|
ID: "305b503e-da60-11ea-9681-0e9f1d580d2d",
|
|
Type: "richtext",
|
|
Text: "test",
|
|
|
|
Color: "light",
|
|
BackgroundColor: "#373c3f",
|
|
CSSClass: "test",
|
|
|
|
Editable: false,
|
|
ModOnly: true,
|
|
},
|
|
}
|
|
|
|
var expectedListUserFlairs = []*FlairSummary{
|
|
{
|
|
User: "TestUser1",
|
|
Text: "TestFlair1",
|
|
},
|
|
{
|
|
User: "TestUser2",
|
|
Text: "TestFlair2",
|
|
},
|
|
}
|
|
|
|
func TestFlairService_GetUserFlairs(t *testing.T) {
|
|
setup()
|
|
defer teardown()
|
|
|
|
blob, err := readFileContents("testdata/flair/user-flairs.json")
|
|
assert.NoError(t, err)
|
|
|
|
mux.HandleFunc("/r/testsubreddit/api/user_flair_v2", func(w http.ResponseWriter, r *http.Request) {
|
|
assert.Equal(t, http.MethodGet, r.Method)
|
|
fmt.Fprint(w, blob)
|
|
})
|
|
|
|
userFlairs, _, err := client.Flair.GetUserFlairs(ctx, "testsubreddit")
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, expectedUserFlairs, userFlairs)
|
|
}
|
|
|
|
func TestFlairService_GetPostFlairs(t *testing.T) {
|
|
setup()
|
|
defer teardown()
|
|
|
|
blob, err := readFileContents("testdata/flair/post-flairs.json")
|
|
assert.NoError(t, err)
|
|
|
|
mux.HandleFunc("/r/testsubreddit/api/link_flair_v2", func(w http.ResponseWriter, r *http.Request) {
|
|
assert.Equal(t, http.MethodGet, r.Method)
|
|
fmt.Fprint(w, blob)
|
|
})
|
|
|
|
postFlairs, _, err := client.Flair.GetPostFlairs(ctx, "testsubreddit")
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, expectedPostFlairs, postFlairs)
|
|
}
|
|
|
|
func TestFlairService_ListUserFlairs(t *testing.T) {
|
|
setup()
|
|
defer teardown()
|
|
|
|
blob, err := readFileContents("testdata/flair/list-user-flairs.json")
|
|
assert.NoError(t, err)
|
|
|
|
mux.HandleFunc("/r/testsubreddit/api/flairlist", func(w http.ResponseWriter, r *http.Request) {
|
|
assert.Equal(t, http.MethodGet, r.Method)
|
|
fmt.Fprint(w, blob)
|
|
})
|
|
|
|
userFlairs, _, err := client.Flair.ListUserFlairs(ctx, "testsubreddit")
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, expectedListUserFlairs, userFlairs)
|
|
}
|