24 lines
484 B
Go
24 lines
484 B
Go
|
package middleware
|
||
|
|
||
|
import (
|
||
|
"fmt"
|
||
|
"testing"
|
||
|
|
||
|
"goweb/context"
|
||
|
"goweb/ent"
|
||
|
"goweb/tests"
|
||
|
|
||
|
"github.com/stretchr/testify/assert"
|
||
|
"github.com/stretchr/testify/require"
|
||
|
)
|
||
|
|
||
|
func TestLoadUser(t *testing.T) {
|
||
|
ctx, _ := tests.NewContext(c.Web, "/")
|
||
|
ctx.SetParamNames("user")
|
||
|
ctx.SetParamValues(fmt.Sprintf("%d", usr.ID))
|
||
|
_ = tests.ExecuteMiddleware(ctx, LoadUser(c.ORM))
|
||
|
ctxUsr, ok := ctx.Get(context.UserKey).(*ent.User)
|
||
|
require.True(t, ok)
|
||
|
assert.Equal(t, usr.ID, ctxUsr.ID)
|
||
|
}
|