caching_middleware_test.go

package middleware

import (
	"context"
	"testing"
	"time"

	"git.perx.ru/perxis/perxis-go/organizations"
	mocksorgs "git.perx.ru/perxis/perxis-go/organizations/mocks"
	"git.perx.ru/perxis/perxis-go/pkg/cache"
	"git.perx.ru/perxis/perxis-go/pkg/errors"
	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/mock"
	"github.com/stretchr/testify/require"
)

func TestOrganizationsCache(t *testing.T) {

	const (
		orgId = "orgId"
		size  = 5
		ttl   = 20 * time.Millisecond
	)

	errNotFound := errors.NotFound(errors.New("not found"))

	ctx := context.Background()

	t.Run("Get from cache", func(t *testing.T) {
		orgs := &mocksorgs.Organizations{}
		svc := CachingMiddleware(cache.NewMemoryCache(size, ttl))(orgs)

		orgs.On("Get", mock.Anything, orgId).Return(&organizations.Organization{ID: orgId, Name: "Organization"}, nil).Once()

		v1, err := svc.Get(ctx, orgId)
		require.NoError(t, err)

		v2, err := svc.Get(ctx, orgId)
		require.NoError(t, err)
		assert.Equal(t, v1, v2, "Ожидается получение объекта из кэша.")
		assert.NotSame(t, v1, v2)

		orgs.AssertExpectations(t)
	})

	t.Run("Invalidate cache", func(t *testing.T) {
		t.Run("After Update", func(t *testing.T) {
			orgs := &mocksorgs.Organizations{}
			svc := CachingMiddleware(cache.NewMemoryCache(size, ttl))(orgs)

			orgs.On("Get", mock.Anything, orgId).Return(&organizations.Organization{ID: orgId, Name: "Organization"}, nil).Once()

			v1, err := svc.Get(ctx, orgId)
			require.NoError(t, err)

			v2, err := svc.Get(ctx, orgId)
			require.NoError(t, err)
			assert.Equal(t, v1, v2, "Ожидается получение объекта из кэша.")
			assert.NotSame(t, v1, v2)

			orgs.On("Update", mock.Anything, mock.Anything).Return(nil).Once()
			err = svc.Update(ctx, &organizations.Organization{ID: orgId, Name: "OrganizationUPD"})
			require.NoError(t, err)

			orgs.On("Get", mock.Anything, orgId).Return(&organizations.Organization{ID: orgId, Name: "OrganizationUPD"}, nil).Once()

			v3, err := svc.Get(ctx, orgId)
			require.NoError(t, err)
			assert.NotEqual(t, v2, v3, "Ожидается удаление объекта из кэша после обновления и получение заново из сервиса.")

			orgs.AssertExpectations(t)
		})

		t.Run("After Delete", func(t *testing.T) {
			orgs := &mocksorgs.Organizations{}
			svc := CachingMiddleware(cache.NewMemoryCache(size, ttl))(orgs)

			orgs.On("Get", mock.Anything, orgId).Return(&organizations.Organization{ID: orgId, Name: "Organization"}, nil).Once()

			v1, err := svc.Get(ctx, orgId)
			require.NoError(t, err)

			v2, err := svc.Get(ctx, orgId)
			require.NoError(t, err)
			assert.Equal(t, v1, v2, "Ожидается получение объекта из кэша.")
			assert.NotSame(t, v1, v2)

			orgs.On("Delete", mock.Anything, mock.Anything).Return(nil).Once()
			err = svc.Delete(ctx, orgId)
			require.NoError(t, err)

			orgs.On("Get", mock.Anything, orgId).Return(nil, errNotFound).Once()

			_, err = svc.Get(ctx, orgId)
			require.Error(t, err)
			assert.EqualError(t, err, "not found", "Ожидается удаление объекта из кэша после удаления из хранилища и получение ошибки от сервиса.")

		})

		t.Run("After TTL expired", func(t *testing.T) {
			orgs := &mocksorgs.Organizations{}
			svc := CachingMiddleware(cache.NewMemoryCache(size, ttl))(orgs)

			orgs.On("Get", mock.Anything, orgId).Return(&organizations.Organization{ID: orgId, Name: "Organization"}, nil).Once()

			v1, err := svc.Get(ctx, orgId)
			require.NoError(t, err)

			v2, err := svc.Get(ctx, orgId)
			require.NoError(t, err)
			assert.Equal(t, v1, v2, "Ожидается получение объекта из кэша.")
			assert.NotSame(t, v1, v2)

			time.Sleep(2 * ttl)

			orgs.On("Get", mock.Anything, orgId).Return(&organizations.Organization{ID: orgId, Name: "Organization"}, nil).Once()

			v3, err := svc.Get(ctx, orgId)
			require.NoError(t, err)
			assert.NotSame(t, v2, v3, "Ожидается удаление объекта из кэша и получение заново из сервиса.")
			assert.Equal(t, v2, v3)
			assert.NotSame(t, v2, v3)

			orgs.AssertExpectations(t)
		})
	})
}