micro--go-micro/client/cache_test.go

package client

import (
	"context"
	"testing"
	"time"

	"go-micro.dev/v6/metadata"
	"go-micro.dev/v6/transport/headers"
)

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(), headers.Namespace, "bar")
		key1 := key(mdCtx, &req1)
		key2 := key(ctx, &req1)

		if key1 == key2 {
			t.Errorf("Expected the keys to differ for different metadata")
		}
	})
}