Umbraco-CMS/tests/Umbraco.Tests.UnitTests/Umbraco.Core/Models/ContentScheduleTests.cs

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

using System;
using System.Collections.Generic;
using System.Linq;
using NUnit.Framework;
using Umbraco.Cms.Core.Models;

namespace Umbraco.Cms.Tests.UnitTests.Umbraco.Core.Models
{
    [TestFixture]
    public class ContentScheduleTests
    {
        [Test]
        public void Release_Date_Less_Than_Expire_Date()
        {
            DateTime now = DateTime.Now;
            var schedule = new ContentScheduleCollection();
            Assert.IsFalse(schedule.Add(now, now));
        }

        [Test]
        public void Cannot_Add_Duplicate_Dates_Invariant()
        {
            DateTime now = DateTime.Now;
            var schedule = new ContentScheduleCollection();
            schedule.Add(now, null);
            Assert.Throws<ArgumentException>(() => schedule.Add(null, now));
        }

        [Test]
        public void Cannot_Add_Duplicate_Dates_Variant()
        {
            DateTime now = DateTime.Now;
            var schedule = new ContentScheduleCollection();
            schedule.Add(now, null);
            schedule.Add("en-US", now, null);
            Assert.Throws<ArgumentException>(() => schedule.Add("en-US", null, now));
            Assert.Throws<ArgumentException>(() => schedule.Add(null, now));
        }

        [Test]
        public void Can_Remove_Invariant()
        {
            DateTime now = DateTime.Now;
            var schedule = new ContentScheduleCollection();
            schedule.Add(now, null);
            IEnumerable<ContentSchedule> invariantSched = schedule.GetSchedule(string.Empty);
            schedule.Remove(invariantSched.First());
            Assert.AreEqual(0, schedule.FullSchedule.Count());
        }

        [Test]
        public void Can_Remove_Variant()
        {
            DateTime now = DateTime.Now;
            var schedule = new ContentScheduleCollection();
            schedule.Add(now, null);
            schedule.Add("en-US", now, null);
            IEnumerable<ContentSchedule> invariantSched = schedule.GetSchedule(string.Empty);
            schedule.Remove(invariantSched.First());
            Assert.AreEqual(0, schedule.GetSchedule(string.Empty).Count());
            Assert.AreEqual(1, schedule.FullSchedule.Count());
            IEnumerable<ContentSchedule> variantSched = schedule.GetSchedule("en-US");
            schedule.Remove(variantSched.First());
            Assert.AreEqual(0, schedule.GetSchedule("en-US").Count());
            Assert.AreEqual(0, schedule.FullSchedule.Count());
        }

        [Test]
        public void Can_Clear_Start_Invariant()
        {
            DateTime now = DateTime.Now;
            var schedule = new ContentScheduleCollection();
            schedule.Add(now, now.AddDays(1));

            schedule.Clear(ContentScheduleAction.Release);

            Assert.AreEqual(0, schedule.GetSchedule(ContentScheduleAction.Release).Count());
            Assert.AreEqual(1, schedule.GetSchedule(ContentScheduleAction.Expire).Count());
            Assert.AreEqual(1, schedule.FullSchedule.Count());
        }

        [Test]
        public void Can_Clear_End_Variant()
        {
            DateTime now = DateTime.Now;
            var schedule = new ContentScheduleCollection();
            schedule.Add(now, now.AddDays(1));
            schedule.Add("en-US", now, now.AddDays(1));

            schedule.Clear(ContentScheduleAction.Expire);

            Assert.AreEqual(0, schedule.GetSchedule(ContentScheduleAction.Expire).Count());
            Assert.AreEqual(1, schedule.GetSchedule(ContentScheduleAction.Release).Count());
            Assert.AreEqual(1, schedule.GetSchedule("en-US", ContentScheduleAction.Expire).Count());
            Assert.AreEqual(1, schedule.GetSchedule("en-US", ContentScheduleAction.Release).Count());
            Assert.AreEqual(3, schedule.FullSchedule.Count());

            schedule.Clear("en-US", ContentScheduleAction.Expire);

            Assert.AreEqual(0, schedule.GetSchedule(ContentScheduleAction.Expire).Count());
            Assert.AreEqual(1, schedule.GetSchedule(ContentScheduleAction.Release).Count());
            Assert.AreEqual(0, schedule.GetSchedule("en-US", ContentScheduleAction.Expire).Count());
            Assert.AreEqual(1, schedule.GetSchedule("en-US", ContentScheduleAction.Release).Count());
            Assert.AreEqual(2, schedule.FullSchedule.Count());
        }
    }
}