48 lines
1.0 KiB
C#
48 lines
1.0 KiB
C#
// 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;
|
|
}
|
|
}
|
|
}
|