hanayo/vendor/github.com/RangelReale/osin/response_json_test.go

116 lines
2.8 KiB
Go
Raw Normal View History

2019-02-23 13:29:15 +00:00
package osin
import (
"encoding/json"
"net/http"
"net/http/httptest"
"testing"
)
func TestResponseJSON(t *testing.T) {
req, err := http.NewRequest("GET", "http://localhost:14000/appauth", nil)
if err != nil {
t.Fatal(err)
}
w := httptest.NewRecorder()
r := NewResponse(NewTestingStorage())
r.Output["access_token"] = "1234"
r.Output["token_type"] = "5678"
err = OutputJSON(r, w, req)
if err != nil {
t.Fatalf("Error outputting json: %s", err)
}
//fmt.Printf("%d - %s - %+v", w.Code, w.Body.String(), w.HeaderMap)
if w.Code != 200 {
t.Fatalf("Invalid response code for output: %d", w.Code)
}
if w.HeaderMap.Get("Content-Type") != "application/json" {
t.Fatalf("Result from json must be application/json")
}
// parse output json
output := make(map[string]interface{})
if err := json.Unmarshal(w.Body.Bytes(), &output); err != nil {
t.Fatalf("Could not decode output json: %s", err)
}
if d, ok := output["access_token"]; !ok || d != "1234" {
t.Fatalf("Invalid or not found output data: access_token=%s", d)
}
if d, ok := output["token_type"]; !ok || d != "5678" {
t.Fatalf("Invalid or not found output data: token_type=%s", d)
}
}
func TestErrorResponseJSON(t *testing.T) {
req, err := http.NewRequest("GET", "http://localhost:14000/appauth", nil)
if err != nil {
t.Fatal(err)
}
w := httptest.NewRecorder()
r := NewResponse(NewTestingStorage())
r.ErrorStatusCode = 500
r.SetError(E_INVALID_REQUEST, "")
err = OutputJSON(r, w, req)
if err != nil {
t.Fatalf("Error outputting json: %s", err)
}
//fmt.Printf("%d - %s - %+v", w.Code, w.Body.String(), w.HeaderMap)
if w.Code != 500 {
t.Fatalf("Invalid response code for error output: %d", w.Code)
}
if w.HeaderMap.Get("Content-Type") != "application/json" {
t.Fatalf("Result from json must be application/json")
}
// parse output json
output := make(map[string]interface{})
if err := json.Unmarshal(w.Body.Bytes(), &output); err != nil {
t.Fatalf("Could not decode output json: %s", err)
}
if d, ok := output["error"]; !ok || d != E_INVALID_REQUEST {
t.Fatalf("Invalid or not found output data: error=%s", d)
}
}
func TestRedirectResponseJSON(t *testing.T) {
req, err := http.NewRequest("GET", "http://localhost:14000/appauth", nil)
if err != nil {
t.Fatal(err)
}
w := httptest.NewRecorder()
r := NewResponse(NewTestingStorage())
r.SetRedirect("http://localhost:14000")
err = OutputJSON(r, w, req)
if err != nil {
t.Fatalf("Error outputting json: %s", err)
}
//fmt.Printf("%d - %s - %+v", w.Code, w.Body.String(), w.HeaderMap)
if w.Code != 302 {
t.Fatalf("Invalid response code for redirect output: %d", w.Code)
}
if w.HeaderMap.Get("Location") != "http://localhost:14000" {
t.Fatalf("Invalid response location url: %s", w.HeaderMap.Get("Location"))
}
}