package client import ( "context" "testing" "time" "go-micro.dev/v4/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(), "Micro-Namespace", "bar") key1 := key(mdCtx, &req1) key2 := key(ctx, &req1) if key1 == key2 { t.Errorf("Expected the keys to differ for different metadata") } }) }