Merge pull request #44 from tinh-tinh/enhance/ren/add-unit-test

fix: add unit test session

middleware/session/session.go

@@ -29,8 +29,9 @@ type Options struct {
 
 func New(opt Options) *Config {
 	session := &Config{
-		Secret: opt.Secret,
-		store:  memory.New(opt.StoreOptions),
+		Secret:    opt.Secret,
+		store:     memory.New(opt.StoreOptions),
+		ExpiresIn: opt.ExpiresIn,
 	}
 	if session.ExpiresIn == 0 {
 		session.ExpiresIn = time.Hour

middleware/session/session_test.go

@@ -0,0 +1,30 @@
+package session
+
+import (
+	"testing"
+	"time"
+
+	"github.com/stretchr/testify/require"
+)
+
+func Test_Session(t *testing.T) {
+	s := New(Options{
+		GeneratorID: func() string {
+			return time.Now().String()
+		},
+	})
+
+	cookie := s.Set("abc", "mno")
+	require.Equal(t, "mno", s.Get(cookie.Value))
+}
+
+func Test_Expiration(t *testing.T) {
+	s := New(Options{
+		ExpiresIn: 3 * time.Second,
+	})
+
+	cookie := s.Set("abc", "mno")
+	require.Equal(t, "mno", s.Get(cookie.Value))
+	time.Sleep(3 * time.Second)
+	require.Nil(t, s.Get(cookie.Value))
+}