Added test case for session middleware

Signed-off-by: Vishal Rana <[email protected]>

Author: vishr

Gopkg.lock

@@ -18,6 +18,10 @@ type (
 	}
 )
 
+const (
+	key = "_session_store"
+)
+
 var (
 	// DefaultConfig is the default Session middleware config.
 	DefaultConfig = Config{
@@ -27,7 +31,7 @@ var (
 
 // Get returns a named session.
 func Get(name string, c echo.Context) (*sessions.Session, error) {
-	store := c.Get("_session_store").(sessions.Store)
+	store := c.Get(key).(sessions.Store)
 	return store.Get(c.Request(), name)
 }
 
@@ -54,7 +58,7 @@ func MiddlewareWithConfig(config Config) echo.MiddlewareFunc {
 			if config.Skipper(c) {
 				return next(c)
 			}
-			c.Set("_session_store", config.Store)
+			c.Set(key, config.Store)
 			return next(c)
 		}
 	}

session/session.go

@@ -1 +1,51 @@
 package session
+
+import (
+	"net/http"
+	"net/http/httptest"
+	"testing"
+
+	"github.com/gorilla/sessions"
+	"github.com/labstack/echo"
+	"github.com/stretchr/testify/assert"
+)
+
+func TestMiddleware(t *testing.T) {
+	e := echo.New()
+	req := httptest.NewRequest(echo.GET, "/", nil)
+	rec := httptest.NewRecorder()
+	c := e.NewContext(req, rec)
+	handler := func(c echo.Context) error {
+		sess, _ := Get("test", c)
+		sess.Options.Domain = "labstack.com"
+		sess.Values["foo"] = "bar"
+		sess.Save(c.Request(), c.Response())
+		return c.String(http.StatusOK, "test")
+	}
+	store := sessions.NewCookieStore([]byte("secret"))
+	config := Config{
+		Skipper: func(c echo.Context) bool {
+			return true
+		},
+		Store: store,
+	}
+
+	// Skipper
+	mw := MiddlewareWithConfig(config)
+	h := mw(echo.NotFoundHandler)
+	h(c)
+	assert.Nil(t, c.Get(key))
+
+	// Panic
+	config.Skipper = nil
+	config.Store = nil
+	assert.Panics(t, func() {
+		MiddlewareWithConfig(config)
+	})
+
+	// Core
+	mw = Middleware(store)
+	h = mw(handler)
+	h(c)
+	assert.Contains(t, rec.Header().Get(echo.HeaderSetCookie), "labstack.com")
+}