diff --git a/tests/Umbraco.Tests.Common/Builders/ContentTypeBuilder.cs b/tests/Umbraco.Tests.Common/Builders/ContentTypeBuilder.cs index f86cb7d5cf..5d10f26ff4 100644 --- a/tests/Umbraco.Tests.Common/Builders/ContentTypeBuilder.cs +++ b/tests/Umbraco.Tests.Common/Builders/ContentTypeBuilder.cs @@ -22,6 +22,7 @@ public class ContentTypeBuilder private readonly List _templateBuilders = new(); private ContentVariation? _contentVariation; private int? _defaultTemplateId; + private bool? _isElement; private PropertyTypeCollection _propertyTypeCollection; private int? _propertyTypeIdsIncrementingFrom; @@ -48,6 +49,12 @@ public class ContentTypeBuilder return this; } + public ContentTypeBuilder WithIsElement(bool isElement) + { + _isElement = isElement; + return this; + } + public ContentTypeBuilder WithContentVariation(ContentVariation contentVariation) { _contentVariation = contentVariation; @@ -117,6 +124,7 @@ public class ContentTypeBuilder contentType.CreatorId = GetCreatorId(); contentType.Trashed = GetTrashed(); contentType.IsContainer = GetIsContainer(); + contentType.IsElement = _isElement ?? false; contentType.HistoryCleanup = new HistoryCleanup(); contentType.Variations = contentVariation;