Client Cache tests
This commit is contained in:
parent
496293afa1
commit
67146ecdc2
@ -3,7 +3,9 @@ package wrapper
|
|||||||
import (
|
import (
|
||||||
"context"
|
"context"
|
||||||
"testing"
|
"testing"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/micro/go-micro/v2/client"
|
||||||
"github.com/micro/go-micro/v2/metadata"
|
"github.com/micro/go-micro/v2/metadata"
|
||||||
)
|
)
|
||||||
|
|
||||||
@ -49,5 +51,72 @@ func TestWrapper(t *testing.T) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
type testClient struct {
|
||||||
|
callCount int
|
||||||
|
callRsp interface{}
|
||||||
|
cache *client.Cache
|
||||||
|
client.Client
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *testClient) Call(ctx context.Context, req client.Request, rsp interface{}, opts ...client.CallOption) error {
|
||||||
|
c.callCount++
|
||||||
|
rsp = c.callRsp
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *testClient) Options() client.Options {
|
||||||
|
return client.Options{Cache: c.cache}
|
||||||
|
}
|
||||||
|
func TestCacheWrapper(t *testing.T) {
|
||||||
|
req := client.NewRequest("go.micro.service.foo", "Foo.Bar", nil)
|
||||||
|
|
||||||
|
t.Run("NilCache", func(t *testing.T) {
|
||||||
|
cli := new(testClient)
|
||||||
|
w := CacheClient(cli)
|
||||||
|
|
||||||
|
// perfroming two requests should increment the call count by two indicating the cache wasn't
|
||||||
|
// used even though the WithCache option was passed.
|
||||||
|
w.Call(context.TODO(), req, nil, client.WithCache(time.Minute))
|
||||||
|
w.Call(context.TODO(), req, nil, client.WithCache(time.Minute))
|
||||||
|
|
||||||
|
if cli.callCount != 2 {
|
||||||
|
t.Errorf("Expected the client to have been called twice")
|
||||||
|
}
|
||||||
|
})
|
||||||
|
|
||||||
|
t.Run("OptionNotSet", func(t *testing.T) {
|
||||||
|
cli := new(testClient)
|
||||||
|
w := CacheClient(cli)
|
||||||
|
|
||||||
|
// perfroming two requests should increment the call count by two since we didn't pass the WithCache
|
||||||
|
// option to Call.
|
||||||
|
w.Call(context.TODO(), req, nil)
|
||||||
|
w.Call(context.TODO(), req, nil)
|
||||||
|
|
||||||
|
if cli.callCount != 2 {
|
||||||
|
t.Errorf("Expected the client to have been called twice")
|
||||||
|
}
|
||||||
|
})
|
||||||
|
|
||||||
|
t.Run("OptionSet", func(t *testing.T) {
|
||||||
|
cli := &testClient{callRsp: "foobar", cache: client.NewCache()}
|
||||||
|
w := CacheClient(cli)
|
||||||
|
|
||||||
|
// perfroming two requests should increment the call count by once since the second request should
|
||||||
|
// have used the cache
|
||||||
|
err1 := w.Call(context.TODO(), req, nil, client.WithCache(time.Minute))
|
||||||
|
err2 := w.Call(context.TODO(), req, nil, client.WithCache(time.Minute))
|
||||||
|
|
||||||
|
if err1 != nil {
|
||||||
|
t.Errorf("Expected nil error, got %v", err1)
|
||||||
|
}
|
||||||
|
if err2 != nil {
|
||||||
|
t.Errorf("Expected nil error, got %v", err2)
|
||||||
|
}
|
||||||
|
if cli.callCount != 1 {
|
||||||
|
t.Errorf("Expected the client to be called 1 time, was actually called %v time(s)", cli.callCount)
|
||||||
|
}
|
||||||
|
})
|
||||||
}
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user