Umbraco-CMS/tests/Umbraco.Tests.UnitTests/Umbraco.Core/Cache/SingleItemsOnlyCachePolicyTests.cs

// Copyright (c) Umbraco.
// See LICENSE for more details.

using System;
using System.Collections.Generic;
using Moq;
using NUnit.Framework;
using Umbraco.Cms.Core.Cache;
using Umbraco.Cms.Core.Models;
using Umbraco.Cms.Core.Scoping;
using Umbraco.Cms.Infrastructure.Scoping;
using IScope = Umbraco.Cms.Infrastructure.Scoping.IScope;

namespace Umbraco.Cms.Tests.UnitTests.Umbraco.Core.Cache
{
    [TestFixture]
    public class SingleItemsOnlyCachePolicyTests
    {
        private IScopeAccessor DefaultAccessor
        {
            get
            {
                var accessor = new Mock<IScopeAccessor>();
                var scope = new Mock<IScope>();
                scope.Setup(x => x.RepositoryCacheMode).Returns(RepositoryCacheMode.Default);
                accessor.Setup(x => x.AmbientScope).Returns(scope.Object);
                return accessor.Object;
            }
        }

        [Test]
        public void Get_All_Doesnt_Cache()
        {
            var cached = new List<string>();
            var cache = new Mock<IAppPolicyCache>();
            cache.Setup(x => x.Insert(It.IsAny<string>(), It.IsAny<Func<object>>(), It.IsAny<TimeSpan?>(), It.IsAny<bool>(), It.IsAny<string[]>()))
                .Callback((string cacheKey, Func<object> o, TimeSpan? t, bool b, string[] s) => cached.Add(cacheKey));
            cache.Setup(x => x.SearchByKey(It.IsAny<string>())).Returns(new AuditItem[] { });

            var defaultPolicy = new SingleItemsOnlyRepositoryCachePolicy<AuditItem, object>(cache.Object, DefaultAccessor, new RepositoryCachePolicyOptions());

            AuditItem[] unused = defaultPolicy.GetAll(new object[] { }, ids => new[]
                    {
                        new AuditItem(1, AuditType.Copy, 123, "test", "blah"),
                        new AuditItem(2, AuditType.Copy, 123, "test", "blah2")
                    });

            Assert.AreEqual(0, cached.Count);
        }

        [Test]
        public void Caches_Single()
        {
            var isCached = false;
            var cache = new Mock<IAppPolicyCache>();
            cache.Setup(x => x.Insert(It.IsAny<string>(), It.IsAny<Func<object>>(), It.IsAny<TimeSpan?>(), It.IsAny<bool>(), It.IsAny<string[]>()))
                .Callback(() => isCached = true);

            var defaultPolicy = new SingleItemsOnlyRepositoryCachePolicy<AuditItem, object>(cache.Object, DefaultAccessor, new RepositoryCachePolicyOptions());

            AuditItem unused = defaultPolicy.Get(1, id => new AuditItem(1, AuditType.Copy, 123, "test", "blah"), ids => null);
            Assert.IsTrue(isCached);
        }
    }
}