You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
138 lines
3.4 KiB
Go
138 lines
3.4 KiB
Go
package redis
|
|
|
|
import (
|
|
"context"
|
|
"log"
|
|
"strings"
|
|
"testing"
|
|
"time"
|
|
|
|
red "github.com/go-redis/redis/v8"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestHookProcessCase1(t *testing.T) {
|
|
writer := log.Writer()
|
|
var buf strings.Builder
|
|
log.SetOutput(&buf)
|
|
defer log.SetOutput(writer)
|
|
|
|
ctx, err := durationHook.BeforeProcess(context.Background(), nil)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
assert.Nil(t, durationHook.AfterProcess(ctx, red.NewCmd(context.Background())))
|
|
assert.False(t, strings.Contains(buf.String(), "slow"))
|
|
}
|
|
|
|
func TestHookProcessCase2(t *testing.T) {
|
|
writer := log.Writer()
|
|
var buf strings.Builder
|
|
log.SetOutput(&buf)
|
|
defer log.SetOutput(writer)
|
|
|
|
ctx, err := durationHook.BeforeProcess(context.Background(), nil)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
time.Sleep(slowThreshold.Load() + time.Millisecond)
|
|
|
|
assert.Nil(t, durationHook.AfterProcess(ctx, red.NewCmd(context.Background(), "foo", "bar")))
|
|
assert.True(t, strings.Contains(buf.String(), "slow"))
|
|
}
|
|
|
|
func TestHookProcessCase3(t *testing.T) {
|
|
writer := log.Writer()
|
|
var buf strings.Builder
|
|
log.SetOutput(&buf)
|
|
defer log.SetOutput(writer)
|
|
|
|
assert.Nil(t, durationHook.AfterProcess(context.Background(), red.NewCmd(context.Background())))
|
|
assert.True(t, buf.Len() == 0)
|
|
}
|
|
|
|
func TestHookProcessCase4(t *testing.T) {
|
|
writer := log.Writer()
|
|
var buf strings.Builder
|
|
log.SetOutput(&buf)
|
|
defer log.SetOutput(writer)
|
|
|
|
ctx := context.WithValue(context.Background(), startTimeKey, "foo")
|
|
assert.Nil(t, durationHook.AfterProcess(ctx, red.NewCmd(context.Background())))
|
|
assert.True(t, buf.Len() == 0)
|
|
}
|
|
|
|
func TestHookProcessPipelineCase1(t *testing.T) {
|
|
writer := log.Writer()
|
|
var buf strings.Builder
|
|
log.SetOutput(&buf)
|
|
defer log.SetOutput(writer)
|
|
|
|
ctx, err := durationHook.BeforeProcessPipeline(context.Background(), nil)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
assert.Nil(t, durationHook.AfterProcessPipeline(ctx, []red.Cmder{
|
|
red.NewCmd(context.Background()),
|
|
}))
|
|
assert.False(t, strings.Contains(buf.String(), "slow"))
|
|
}
|
|
|
|
func TestHookProcessPipelineCase2(t *testing.T) {
|
|
writer := log.Writer()
|
|
var buf strings.Builder
|
|
log.SetOutput(&buf)
|
|
defer log.SetOutput(writer)
|
|
|
|
ctx, err := durationHook.BeforeProcessPipeline(context.Background(), nil)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
time.Sleep(slowThreshold.Load() + time.Millisecond)
|
|
|
|
assert.Nil(t, durationHook.AfterProcessPipeline(ctx, []red.Cmder{
|
|
red.NewCmd(context.Background(), "foo", "bar"),
|
|
}))
|
|
assert.True(t, strings.Contains(buf.String(), "slow"))
|
|
}
|
|
|
|
func TestHookProcessPipelineCase3(t *testing.T) {
|
|
writer := log.Writer()
|
|
var buf strings.Builder
|
|
log.SetOutput(&buf)
|
|
defer log.SetOutput(writer)
|
|
|
|
assert.Nil(t, durationHook.AfterProcessPipeline(context.Background(), []red.Cmder{
|
|
red.NewCmd(context.Background()),
|
|
}))
|
|
assert.True(t, buf.Len() == 0)
|
|
}
|
|
|
|
func TestHookProcessPipelineCase4(t *testing.T) {
|
|
writer := log.Writer()
|
|
var buf strings.Builder
|
|
log.SetOutput(&buf)
|
|
defer log.SetOutput(writer)
|
|
|
|
ctx := context.WithValue(context.Background(), startTimeKey, "foo")
|
|
assert.Nil(t, durationHook.AfterProcessPipeline(ctx, []red.Cmder{
|
|
red.NewCmd(context.Background()),
|
|
}))
|
|
assert.True(t, buf.Len() == 0)
|
|
}
|
|
|
|
func TestHookProcessPipelineCase5(t *testing.T) {
|
|
writer := log.Writer()
|
|
var buf strings.Builder
|
|
log.SetOutput(&buf)
|
|
defer log.SetOutput(writer)
|
|
|
|
ctx := context.WithValue(context.Background(), startTimeKey, "foo")
|
|
assert.Nil(t, durationHook.AfterProcessPipeline(ctx, nil))
|
|
assert.True(t, buf.Len() == 0)
|
|
}
|