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

using System;
using Umbraco.Cms.Infrastructure.PublishedCache;
using Umbraco.Cms.Infrastructure.PublishedCache.DataSource;

namespace Umbraco.Cms.Tests.Common.Builders
{
    public class ContentNodeKitBuilder : BuilderBase<ContentNodeKit>
    {
        private int _contentTypeId;
        private ContentNode _contentNode;
        private ContentData _draftData;
        private ContentData _publishedData;

        public ContentNodeKitBuilder WithContentNode(ContentNode contentNode)
        {
            _contentNode = contentNode;
            return this;
        }

        public ContentNodeKitBuilder WithContentNode(int id, Guid uid, int level, string path, int sortOrder, int parentContentId, DateTime createDate, int creatorId)
        {
            _contentNode = new ContentNode(id, uid, level, path, sortOrder, parentContentId, createDate, creatorId);
            return this;
        }

        public ContentNodeKitBuilder WithContentTypeId(int contentTypeId)
        {
            _contentTypeId = contentTypeId;
            return this;
        }

        public ContentNodeKitBuilder WithDraftData(ContentData draftData)
        {
            _draftData = draftData;
            return this;
        }

        public ContentNodeKitBuilder WithPublishedData(ContentData publishedData)
        {
            _publishedData = publishedData;
            return this;
        }

        public override ContentNodeKit Build()
        {
            var data = new ContentNodeKit(_contentNode, _contentTypeId, _draftData, _publishedData);
            return data;
        }

        /// <summary>
        /// Creates a ContentNodeKit 
        /// </summary>
        /// <param name="contentTypeId"></param>
        /// <param name="id"></param>
        /// <param name="path"></param>
        /// <param name="sortOrder"></param>
        /// <param name="level">
        /// Optional. Will get calculated based on the path value if not specified.
        /// </param>
        /// <param name="parentContentId">
        /// Optional. Will get calculated based on the path value if not specified.
        /// </param>
        /// <param name="creatorId"></param>
        /// <param name="uid"></param>
        /// <param name="createDate"></param>
        /// <param name="draftData"></param>
        /// <param name="publishedData"></param>
        /// <returns></returns>
        public static ContentNodeKit CreateWithContent(
            int contentTypeId,
            int id,
            string path,
            int? sortOrder = null,
            int? level = null,
            int? parentContentId = null,
            int creatorId = -1,
            Guid? uid = null,
            DateTime? createDate = null,
            ContentData draftData = null,
            ContentData publishedData = null)
        {
            var pathParts = path.Split(',');
            if (pathParts.Length >= 2)
            {
                parentContentId ??= int.Parse(pathParts[^2]);
            }

            return new ContentNodeKitBuilder()
                       .WithContentTypeId(contentTypeId)
                       .WithContentNode(id, uid ?? Guid.NewGuid(), level ?? pathParts.Length - 1, path, sortOrder ?? 0, parentContentId.Value, createDate ?? DateTime.Now, creatorId)
                       .WithDraftData(draftData)
                       .WithPublishedData(publishedData)
                       .Build();
        }
    }
}