77 lines
1.7 KiB
Go
77 lines
1.7 KiB
Go
|
package client
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"testing"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"github.com/micro/go-micro/v2/metadata"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestCache(t *testing.T) {
|
||
|
|
ctx := context.TODO()
|
||
|
|
req := NewRequest("go.micro.service.foo", "Foo.Bar", nil)
|
||
|
|
|
||
|
|
t.Run("CacheMiss", func(t *testing.T) {
|
||
|
|
if _, ok := NewCache().Get(ctx, &req); ok {
|
||
|
|
t.Errorf("Expected to get no result from Get")
|
||
|
|
}
|
||
|
|
})
|
||
|
|
|
||
|
|
t.Run("CacheHit", func(t *testing.T) {
|
||
|
|
c := NewCache()
|
||
|
|
|
||
|
|
rsp := "theresponse"
|
||
|
|
c.Set(ctx, &req, rsp, time.Minute)
|
||
|
|
|
||
|
|
if res, ok := c.Get(ctx, &req); !ok {
|
||
|
|
t.Errorf("Expected a result, got nothing")
|
||
|
|
} else if res != rsp {
|
||
|
|
t.Errorf("Expected '%v' result, got '%v'", rsp, res)
|
||
|
|
}
|
||
|
|
})
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestCacheKey(t *testing.T) {
|
||
|
|
ctx := context.TODO()
|
||
|
|
req1 := NewRequest("go.micro.service.foo", "Foo.Bar", nil)
|
||
|
|
req2 := NewRequest("go.micro.service.foo", "Foo.Baz", nil)
|
||
|
|
req3 := NewRequest("go.micro.service.foo", "Foo.Baz", "customquery")
|
||
|
|
|
||
|
|
t.Run("IdenticalRequests", func(t *testing.T) {
|
||
|
|
key1 := key(ctx, &req1)
|
||
|
|
key2 := key(ctx, &req1)
|
||
|
|
if key1 != key2 {
|
||
|
|
t.Errorf("Expected the keys to match for identical requests and context")
|
||
|
|
}
|
||
|
|
})
|
||
|
|
|
||
|
|
t.Run("DifferentRequestEndpoints", func(t *testing.T) {
|
||
|
|
key1 := key(ctx, &req1)
|
||
|
|
key2 := key(ctx, &req2)
|
||
|
|
|
||
|
|
if key1 == key2 {
|
||
|
|
t.Errorf("Expected the keys to differ for different request endpoints")
|
||
|
|
}
|
||
|
|
})
|
||
|
|
|
||
|
|
t.Run("DifferentRequestBody", func(t *testing.T) {
|
||
|
|
key1 := key(ctx, &req2)
|
||
|
|
key2 := key(ctx, &req3)
|
||
|
|
|
||
|
|
if key1 == key2 {
|
||
|
|
t.Errorf("Expected the keys to differ for different request bodies")
|
||
|
|
}
|
||
|
|
})
|
||
|
|
|
||
|
|
t.Run("DifferentMetadata", func(t *testing.T) {
|
||
|
|
mdCtx := metadata.Set(context.TODO(), "foo", "bar")
|
||
|
|
key1 := key(mdCtx, &req1)
|
||
|
|
key2 := key(ctx, &req1)
|
||
|
|
|
||
|
|
if key1 == key2 {
|
||
|
|
t.Errorf("Expected the keys to differ for different metadata")
|
||
|
|
}
|
||
|
|
})
|
||
|
|
}
|