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

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

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

namespace Umbraco.Cms.Tests.Common.Builders;

public class ContentItemSaveBuilder : BuilderBase<ContentItemSave>,
    IWithIdBuilder,
    IWithParentIdBuilder
{
    private ContentSaveAction? _action;
    private string _contentTypeAlias;

    private int? _id;
    private int? _parentId;
    private readonly List<ContentVariantSaveBuilder<ContentItemSaveBuilder>> _variantBuilders = new();

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

    int? IWithParentIdBuilder.ParentId
    {
        get => _parentId;
        set => _parentId = value;
    }

    public ContentVariantSaveBuilder<ContentItemSaveBuilder> AddVariant()
    {
        var builder = new ContentVariantSaveBuilder<ContentItemSaveBuilder>(this);
        _variantBuilders.Add(builder);
        return builder;
    }

    public override ContentItemSave Build()
    {
        var id = _id ?? 0;
        var parentId = _parentId ?? -1;
        var contentTypeAlias = _contentTypeAlias;
        var action = _action ?? ContentSaveAction.Save;
        var variants = _variantBuilders.Select(x => x.Build());

        return new TestContentItemSave
        {
            Id = id,
            ParentId = parentId,
            ContentTypeAlias = contentTypeAlias,
            Action = action,
            Variants = variants
        };
    }

    public ContentItemSaveBuilder WithAction(ContentSaveAction action)
    {
        _action = action;
        return this;
    }

    public ContentItemSaveBuilder WithContentTypeAlias(string contentTypeAlias)
    {
        _contentTypeAlias = contentTypeAlias;
        return this;
    }
}

public class TestContentItemSave : ContentItemSave
{
}