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;

        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;
        }

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

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