using System.Linq;
using NUnit.Framework;
using Umbraco.Cms.Core.Trees;
using Umbraco.Cms.Web.BackOffice.Trees;

namespace Umbraco.Cms.Tests.UnitTests.Umbraco.Web.BackOffice.CollectionBuilders
{
    public class TreeCollectionBuilderTests
    {
        [Test]
        public void Adding_Tree_To_Collection_Builder()
        {
            var collectionBuilder = new TreeCollectionBuilder();
            var treeDefinition = new Tree(0, "test", "test", "test", "test", TreeUse.Main, typeof(LanguageTreeController), false);

            collectionBuilder.AddTree(treeDefinition);
            var collection = collectionBuilder.CreateCollection(null);

            Assert.AreEqual(1, collection.Count);
            Assert.AreEqual(treeDefinition, collection.FirstOrDefault());
        }

        [Test]
        public void Remove_Tree_From_Collection_Builder()
        {
            var collectionBuilder = new TreeCollectionBuilder();
            var treeDefinition = new Tree(0, "test", "test", "test", "test", TreeUse.Main, typeof(LanguageTreeController), false);

            collectionBuilder.AddTree(treeDefinition);
            collectionBuilder.RemoveTreeController<LanguageTreeController>();
            var collection = collectionBuilder.CreateCollection(null);

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