From f8ba05d99a3b55393e63542cefbb07fa0dbbd1fc Mon Sep 17 00:00:00 2001
From: Carl Jackson <carl@avtok.com>
Date: Mon, 21 Apr 2014 19:44:00 +0200
Subject: [PATCH] Options middleware tests

They're really verbose. I should probably DRY them up later.
---
 web/middleware/options_test.go | 112 +++++++++++++++++++++++++++++++++
 1 file changed, 112 insertions(+)
 create mode 100644 web/middleware/options_test.go

diff --git a/web/middleware/options_test.go b/web/middleware/options_test.go
new file mode 100644
index 0000000..74f5481
--- /dev/null
+++ b/web/middleware/options_test.go
@@ -0,0 +1,112 @@
+package middleware
+
+import (
+	"net/http"
+	"net/http/httptest"
+	"testing"
+
+	"github.com/zenazn/goji/web"
+)
+
+func testOptions(r *http.Request, f func(*web.C, http.ResponseWriter, *http.Request)) *httptest.ResponseRecorder {
+	var c web.C
+
+	h := func(w http.ResponseWriter, r *http.Request) {
+		f(&c, w, r)
+	}
+	m := AutomaticOptions(&c, http.HandlerFunc(h))
+	w := httptest.NewRecorder()
+	m.ServeHTTP(w, r)
+
+	return w
+}
+
+var optionsTestEnv = map[string]interface{}{
+	web.ValidMethodsKey: []string{
+		"hello",
+		"world",
+	},
+}
+
+func TestAutomaticOptions(t *testing.T) {
+	t.Parallel()
+
+	// Shouldn't interfere with normal requests
+	r, _ := http.NewRequest("GET", "/", nil)
+	rr := testOptions(r,
+		func(c *web.C, w http.ResponseWriter, r *http.Request) {
+			w.Write([]byte{'h', 'i'})
+		},
+	)
+	if rr.Code != http.StatusOK {
+		t.Errorf("status is %d, not 200", rr.Code)
+	}
+	if rr.Body.String() != "hi" {
+		t.Errorf("body was %q, should be %q", rr.Body.String(), "hi")
+	}
+	allow := rr.HeaderMap.Get("Allow")
+	if allow != "" {
+		t.Errorf("Allow header was set to %q, should be empty", allow)
+	}
+
+	// If we respond non-404 to an OPTIONS request, also don't interfere
+	r, _ = http.NewRequest("OPTIONS", "/", nil)
+	rr = testOptions(r,
+		func(c *web.C, w http.ResponseWriter, r *http.Request) {
+			c.Env = optionsTestEnv
+			w.Write([]byte{'h', 'i'})
+		},
+	)
+	if rr.Code != http.StatusOK {
+		t.Errorf("status is %d, not 200", rr.Code)
+	}
+	if rr.Body.String() != "hi" {
+		t.Errorf("body was %q, should be %q", rr.Body.String(), "hi")
+	}
+	allow = rr.HeaderMap.Get("Allow")
+	if allow != "" {
+		t.Errorf("Allow header was set to %q, should be empty", allow)
+	}
+
+	// Provide options if we 404. Make sure we nom the output bytes
+	r, _ = http.NewRequest("OPTIONS", "/", nil)
+	rr = testOptions(r,
+		func(c *web.C, w http.ResponseWriter, r *http.Request) {
+			c.Env = optionsTestEnv
+			w.WriteHeader(http.StatusNotFound)
+			w.Write([]byte{'h', 'i'})
+		},
+	)
+	if rr.Code != http.StatusOK {
+		t.Errorf("status is %d, not 200", rr.Code)
+	}
+	if rr.Body.Len() != 0 {
+		t.Errorf("body was %q, should be empty", rr.Body.String())
+	}
+	allow = rr.HeaderMap.Get("Allow")
+	correctHeaders := "hello, world, OPTIONS"
+	if allow != "hello, world, OPTIONS" {
+		t.Errorf("Allow header should be %q, was %q", correctHeaders,
+			allow)
+	}
+
+	// If we somehow 404 without giving a list of valid options, don't do
+	// anything
+	r, _ = http.NewRequest("OPTIONS", "/", nil)
+	rr = testOptions(r,
+		func(c *web.C, w http.ResponseWriter, r *http.Request) {
+			w.WriteHeader(http.StatusNotFound)
+			w.Write([]byte{'h', 'i'})
+		},
+	)
+	if rr.Code != http.StatusNotFound {
+		t.Errorf("status is %d, not 404", rr.Code)
+	}
+	if rr.Body.String() != "hi" {
+		t.Errorf("body was %q, should be %q", rr.Body.String(), "hi")
+	}
+	allow = rr.HeaderMap.Get("Allow")
+	if allow != "" {
+		t.Errorf("Allow header was set to %q, should be empty", allow)
+	}
+}