Umbraco-CMS/tests/Umbraco.Tests.Common/Builders/PropertyGroupBuilder.cs

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

using System;
using System.Collections.Generic;
using System.Linq;
using Umbraco.Cms.Core.Models;
using Umbraco.Cms.Tests.Common.Builders.Interfaces;

namespace Umbraco.Cms.Tests.Common.Builders
{
    public class PropertyGroupBuilder : PropertyGroupBuilder<NullPropertyGroupBuilderParent>
    {
        public PropertyGroupBuilder()
            : base(null)
        {
        }
    }

    public class NullPropertyGroupBuilderParent : IBuildPropertyGroups
    {
    }

    public class PropertyGroupBuilder<TParent>
        : ChildBuilderBase<TParent, PropertyGroup>,
            IBuildPropertyTypes,
            IWithIdBuilder,
            IWithKeyBuilder,
            IWithCreateDateBuilder,
            IWithUpdateDateBuilder,
            IWithAliasBuilder,
            IWithNameBuilder,
            IWithSortOrderBuilder,
            IWithSupportsPublishing
        where TParent : IBuildPropertyGroups
    {
        private readonly List<PropertyTypeBuilder<PropertyGroupBuilder<TParent>>> _propertyTypeBuilders = new List<PropertyTypeBuilder<PropertyGroupBuilder<TParent>>>();

        private int? _id;
        private Guid? _key;
        private DateTime? _createDate;
        private DateTime? _updateDate;
        private string _alias;
        private string _name;
        private int? _sortOrder;
        private bool? _supportsPublishing;
        private PropertyTypeCollection _propertyTypeCollection;

        public PropertyGroupBuilder(TParent parentBuilder)
            : base(parentBuilder)
        {
        }

        public PropertyGroupBuilder<TParent> WithPropertyTypeCollection(PropertyTypeCollection propertyTypeCollection)
        {
            _propertyTypeCollection = propertyTypeCollection;
            return this;
        }

        public PropertyTypeBuilder<PropertyGroupBuilder<TParent>> AddPropertyType()
        {
            var builder = new PropertyTypeBuilder<PropertyGroupBuilder<TParent>>(this);
            _propertyTypeBuilders.Add(builder);
            return builder;
        }

        public override PropertyGroup Build()
        {
            var id = _id ?? 0;
            Guid key = _key ?? Guid.NewGuid();
            DateTime createDate = _createDate ?? DateTime.Now;
            DateTime updateDate = _updateDate ?? DateTime.Now;
            var alias = _alias ?? Guid.NewGuid().ToString();
            var name = _name ?? Guid.NewGuid().ToString();
            var sortOrder = _sortOrder ?? 0;
            var supportsPublishing = _supportsPublishing ?? false;

            PropertyTypeCollection propertyTypeCollection;
            if (_propertyTypeCollection != null)
            {
                propertyTypeCollection = _propertyTypeCollection;
            }
            else
            {
                propertyTypeCollection = new PropertyTypeCollection(supportsPublishing);
                foreach (PropertyType propertyType in _propertyTypeBuilders.Select(x => x.Build()))
                {
                    propertyTypeCollection.Add(propertyType);
                }
            }

            return new PropertyGroup(propertyTypeCollection)
            {
                Id = id,
                Key = key,
                Alias = alias,
                Name = name,
                SortOrder = sortOrder,
                CreateDate = createDate,
                UpdateDate = updateDate,
            };
        }

        int? IWithIdBuilder.Id
        {
            get => _id;
            set => _id = value;
        }

        Guid? IWithKeyBuilder.Key
        {
            get => _key;
            set => _key = value;
        }

        string IWithAliasBuilder.Alias
        {
            get => _alias;
            set => _alias = value;
        }

        string IWithNameBuilder.Name
        {
            get => _name;
            set => _name = value;
        }

        DateTime? IWithCreateDateBuilder.CreateDate
        {
            get => _createDate;
            set => _createDate = value;
        }

        DateTime? IWithUpdateDateBuilder.UpdateDate
        {
            get => _updateDate;
            set => _updateDate = value;
        }

        int? IWithSortOrderBuilder.SortOrder
        {
            get => _sortOrder;
            set => _sortOrder = value;
        }

        bool? IWithSupportsPublishing.SupportsPublishing
        {
            get => _supportsPublishing;
            set => _supportsPublishing = value;
        }
    }
}