wehub-resource-sync
6cf6f95f58
CodeQL / Analyze (push) Has been cancelled
Sync to Gitee / Run (push) Has been cancelled
Go / build & test (1.25.x) (push) Has been cancelled
Go / build & test (1.26.x) (push) Has been cancelled
Lint / resolve module (push) Has been cancelled
Lint / lint module (push) Has been cancelled
62 lines
1.3 KiB
Go
62 lines
1.3 KiB
Go
package middleware
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"reflect"
|
|
"testing"
|
|
)
|
|
|
|
var i int
|
|
|
|
func TestChain(t *testing.T) {
|
|
next := func(_ context.Context, req any) (any, error) {
|
|
if req != "hello kratos!" {
|
|
t.Errorf("expect %v, got %v", "hello kratos!", req)
|
|
}
|
|
i += 10
|
|
return "reply", nil
|
|
}
|
|
|
|
got, err := Chain(test1Middleware, test2Middleware, test3Middleware)(next)(context.Background(), "hello kratos!")
|
|
if err != nil {
|
|
t.Errorf("expect %v, got %v", nil, err)
|
|
}
|
|
if !reflect.DeepEqual(got, "reply") {
|
|
t.Errorf("expect %v, got %v", "reply", got)
|
|
}
|
|
if !reflect.DeepEqual(i, 16) {
|
|
t.Errorf("expect %v, got %v", 16, i)
|
|
}
|
|
}
|
|
|
|
func test1Middleware(handler Handler) Handler {
|
|
return func(ctx context.Context, req any) (reply any, err error) {
|
|
fmt.Println("test1 before")
|
|
i++
|
|
reply, err = handler(ctx, req)
|
|
fmt.Println("test1 after")
|
|
return
|
|
}
|
|
}
|
|
|
|
func test2Middleware(handler Handler) Handler {
|
|
return func(ctx context.Context, req any) (reply any, err error) {
|
|
fmt.Println("test2 before")
|
|
i += 2
|
|
reply, err = handler(ctx, req)
|
|
fmt.Println("test2 after")
|
|
return
|
|
}
|
|
}
|
|
|
|
func test3Middleware(handler Handler) Handler {
|
|
return func(ctx context.Context, req any) (reply any, err error) {
|
|
fmt.Println("test3 before")
|
|
i += 3
|
|
reply, err = handler(ctx, req)
|
|
fmt.Println("test3 after")
|
|
return
|
|
}
|
|
}
|