Unit test for the IfMiddleware

Author: ant0ine

rest/if_test.go

@@ -0,0 +1,51 @@
+package rest
+
+import (
+	"github.com/ant0ine/go-json-rest/rest/test"
+	"testing"
+)
+
+func TestIfMiddleware(t *testing.T) {
+
+	api := NewApi()
+
+	// the middleware to test
+	api.Use(&IfMiddleware{
+		Condition: func(r *Request) bool {
+			if r.URL.Path == "/true" {
+				return true
+			}
+			return false
+		},
+		IfTrue: MiddlewareSimple(func(handler HandlerFunc) HandlerFunc {
+			return func(w ResponseWriter, r *Request) {
+				r.Env["TRUE_MIDDLEWARE"] = true
+				handler(w, r)
+			}
+		}),
+		IfFalse: MiddlewareSimple(func(handler HandlerFunc) HandlerFunc {
+			return func(w ResponseWriter, r *Request) {
+				r.Env["FALSE_MIDDLEWARE"] = true
+				handler(w, r)
+			}
+		}),
+	})
+
+	// a simple app
+	api.SetApp(AppSimple(func(w ResponseWriter, r *Request) {
+		w.WriteJson(r.Env)
+	}))
+
+	// wrap all
+	handler := api.MakeHandler()
+
+	recorded := test.RunRequest(t, handler, test.MakeSimpleRequest("GET", "http://localhost/", nil))
+	recorded.CodeIs(200)
+	recorded.ContentTypeIsJson()
+	recorded.BodyIs("{\"FALSE_MIDDLEWARE\":true}")
+
+	recorded = test.RunRequest(t, handler, test.MakeSimpleRequest("GET", "http://localhost/true", nil))
+	recorded.CodeIs(200)
+	recorded.ContentTypeIsJson()
+	recorded.BodyIs("{\"TRUE_MIDDLEWARE\":true}")
+}