80 lines
1.9 KiB
Go
80 lines
1.9 KiB
Go
|
package api_test
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"net/http"
|
||
|
|
"net/http/httptest"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/xdrm-io/aicra/api"
|
||
|
|
"github.com/xdrm-io/aicra/internal/ctx"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestContextGetRequest(t *testing.T) {
|
||
|
|
req, err := http.NewRequest(http.MethodGet, "/random", nil)
|
||
|
|
if err != nil {
|
||
|
|
t.Fatalf("cannot create http request: %s", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
// store in bare context
|
||
|
|
c := context.Background()
|
||
|
|
c = context.WithValue(c, ctx.Request, req)
|
||
|
|
|
||
|
|
// fetch from context
|
||
|
|
fetched := api.GetRequest(c)
|
||
|
|
if fetched != req {
|
||
|
|
t.Fatalf("fetched http request %v ; expected %v", fetched, req)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
func TestContextGetNilRequest(t *testing.T) {
|
||
|
|
// fetch from bare context
|
||
|
|
fetched := api.GetRequest(context.Background())
|
||
|
|
if fetched != nil {
|
||
|
|
t.Fatalf("fetched http request %v from empty context; expected nil", fetched)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestContextGetResponseWriter(t *testing.T) {
|
||
|
|
res := httptest.NewRecorder()
|
||
|
|
|
||
|
|
// store in bare context
|
||
|
|
c := context.Background()
|
||
|
|
c = context.WithValue(c, ctx.Response, res)
|
||
|
|
|
||
|
|
// fetch from context
|
||
|
|
fetched := api.GetResponseWriter(c)
|
||
|
|
if fetched != res {
|
||
|
|
t.Fatalf("fetched http response writer %v ; expected %v", fetched, res)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestContextGetNilResponseWriter(t *testing.T) {
|
||
|
|
// fetch from bare context
|
||
|
|
fetched := api.GetResponseWriter(context.Background())
|
||
|
|
if fetched != nil {
|
||
|
|
t.Fatalf("fetched http response writer %v from empty context; expected nil", fetched)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestContextGetAuth(t *testing.T) {
|
||
|
|
auth := &api.Auth{}
|
||
|
|
|
||
|
|
// store in bare context
|
||
|
|
c := context.Background()
|
||
|
|
c = context.WithValue(c, ctx.Auth, auth)
|
||
|
|
|
||
|
|
// fetch from context
|
||
|
|
fetched := api.GetAuth(c)
|
||
|
|
if fetched != auth {
|
||
|
|
t.Fatalf("fetched api auth %v ; expected %v", fetched, auth)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestContextGetNilAuth(t *testing.T) {
|
||
|
|
// fetch from bare context
|
||
|
|
fetched := api.GetAuth(context.Background())
|
||
|
|
if fetched != nil {
|
||
|
|
t.Fatalf("fetched api auth %v from empty context; expected nil", fetched)
|
||
|
|
}
|
||
|
|
}
|