From 19bcc2e8dc492e08558a68de20652580f2eafdfe Mon Sep 17 00:00:00 2001
From: xdrm-brackets <xdrm.brackets.dev@gmail.com>
Date: Mon, 21 Jun 2021 21:46:03 +0200
Subject: [PATCH] test: cover api context

---
 api/context_test.go | 79 +++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 79 insertions(+)
 create mode 100644 api/context_test.go

diff --git a/api/context_test.go b/api/context_test.go
new file mode 100644
index 0000000..40568ce
--- /dev/null
+++ b/api/context_test.go
@@ -0,0 +1,79 @@
+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)
+	}
+}