Client Cache tests

This commit is contained in:
Ben Toogood 2020-05-24 18:05:23 +01:00
parent 496293afa1
commit 67146ecdc2

View File

@ -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)
}
})
} }