125 lines
2.5 KiB
Go
125 lines
2.5 KiB
Go
|
package geddit
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"errors"
|
||
|
|
"fmt"
|
||
|
|
"io/ioutil"
|
||
|
|
"net/http"
|
||
|
|
"net/http/httptest"
|
||
|
|
"os"
|
||
|
|
"reflect"
|
||
|
|
"testing"
|
||
|
|
)
|
||
|
|
|
||
|
|
var (
|
||
|
|
mux *http.ServeMux
|
||
|
|
ctx = context.Background()
|
||
|
|
client *Client
|
||
|
|
server *httptest.Server
|
||
|
|
)
|
||
|
|
|
||
|
|
func setup() {
|
||
|
|
mux = http.NewServeMux()
|
||
|
|
server = httptest.NewServer(mux)
|
||
|
|
|
||
|
|
mux.HandleFunc("/api/v1/access_token", func(w http.ResponseWriter, r *http.Request) {
|
||
|
|
response := `
|
||
|
|
{
|
||
|
|
"access_token": "token1",
|
||
|
|
"token_type": "bearer",
|
||
|
|
"expires_in": 3600,
|
||
|
|
"scope": "*"
|
||
|
|
}
|
||
|
|
`
|
||
|
|
w.Header().Add(headerContentType, mediaTypeJSON)
|
||
|
|
fmt.Fprint(w, response)
|
||
|
|
})
|
||
|
|
|
||
|
|
client, _ = NewClient(nil,
|
||
|
|
WithOAuth2("id1", "secret1", "user1", "password1"),
|
||
|
|
WithBaseURL(server.URL),
|
||
|
|
WithTokenURL(server.URL+"/api/v1/access_token"),
|
||
|
|
)
|
||
|
|
}
|
||
|
|
|
||
|
|
func teardown() {
|
||
|
|
server.Close()
|
||
|
|
}
|
||
|
|
|
||
|
|
func testMethod(t *testing.T, r *http.Request, expected string) {
|
||
|
|
if expected != r.Method {
|
||
|
|
t.Fatalf("Request method = %v, expected %v", r.Method, expected)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func readFileContents(t *testing.T, filepath string) string {
|
||
|
|
file, err := os.Open(filepath)
|
||
|
|
if err != nil {
|
||
|
|
t.Fatalf("got unexpected error: %v", err)
|
||
|
|
}
|
||
|
|
defer file.Close()
|
||
|
|
|
||
|
|
bytes, err := ioutil.ReadAll(file)
|
||
|
|
if err != nil {
|
||
|
|
t.Fatalf("got unexpected error: %v", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
return string(bytes)
|
||
|
|
}
|
||
|
|
|
||
|
|
func testClientServices(t *testing.T, c *Client) {
|
||
|
|
services := []string{
|
||
|
|
"Comment",
|
||
|
|
"Flair",
|
||
|
|
"Link",
|
||
|
|
"Listings",
|
||
|
|
"Subreddit",
|
||
|
|
"User",
|
||
|
|
"Vote",
|
||
|
|
}
|
||
|
|
|
||
|
|
cp := reflect.ValueOf(c)
|
||
|
|
cv := reflect.Indirect(cp)
|
||
|
|
|
||
|
|
for _, s := range services {
|
||
|
|
if cv.FieldByName(s).IsNil() {
|
||
|
|
t.Fatalf("c.%s shouldn't be nil", s)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func testClientDefaultUserAgent(t *testing.T, c *Client) {
|
||
|
|
if expect, actual := fmt.Sprintf("golang:%s:v%s (by /u/)", libraryName, libraryVersion), c.userAgent; expect != actual {
|
||
|
|
t.Fatalf("got unexpected value\nexpect: %+v\nactual: %+v", Stringify(expect), Stringify(actual))
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func testClientDefaults(t *testing.T, c *Client) {
|
||
|
|
testClientDefaultUserAgent(t, c)
|
||
|
|
testClientServices(t, c)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestNewClient(t *testing.T) {
|
||
|
|
c, err := NewClient(nil)
|
||
|
|
if err != nil {
|
||
|
|
t.Fatalf("got unexpected error: %+v", err)
|
||
|
|
}
|
||
|
|
testClientDefaults(t, c)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestNewClient_Error(t *testing.T) {
|
||
|
|
errorOpt := func(c *Client) error {
|
||
|
|
return errors.New("foo")
|
||
|
|
}
|
||
|
|
|
||
|
|
_, err := NewClient(nil, errorOpt)
|
||
|
|
if err == nil {
|
||
|
|
t.Fatal("expected error, got nothing instead")
|
||
|
|
}
|
||
|
|
|
||
|
|
if expect, actual := "foo", err.Error(); expect != actual {
|
||
|
|
t.Fatalf("got unexpected error\nexpect: %+v\nactual: %+v", Stringify(expect), Stringify(actual))
|
||
|
|
}
|
||
|
|
}
|