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

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

using Umbraco.Cms.Core.Models.Entities;
using Umbraco.Cms.Tests.Common.Builders.Interfaces;

namespace Umbraco.Cms.Tests.Common.Builders;

public class EntitySlimBuilder
    : BuilderBase<EntitySlim>,
        IWithIdBuilder,
        IWithParentIdBuilder
{
    private int? _id;
    private int? _parentId;

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

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

    public override EntitySlim Build()
    {
        var id = _id ?? 1;
        var parentId = _parentId ?? -1;

        return new EntitySlim { Id = id, ParentId = parentId };
    }

    public EntitySlimBuilder WithNoParentId()
    {
        _parentId = 0;
        return this;
    }
}