116 lines
2.8 KiB
Go
116 lines
2.8 KiB
Go
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"))
|
|
}
|
|
}
|