Umbraco-CMS/tests/Umbraco.Tests.Common/Builders/Extensions/BuilderExtensions.cs

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

using System;
using System.Globalization;
using Umbraco.Cms.Core.Models;
using Umbraco.Cms.Tests.Common.Builders.Interfaces;

namespace Umbraco.Cms.Tests.Common.Builders.Extensions
{
    public static class BuilderExtensions
    {
        public static T WithId<T>(this T builder, int id)
            where T : IWithIdBuilder
        {
            builder.Id = id;
            return builder;
        }

        public static T WithId<T, TId>(this T builder, TId id)
            where T : IWithIdBuilder<TId>
        {
            builder.Id = id;
            return builder;
        }

        public static T WithoutIdentity<T>(this T builder)
            where T : IWithIdBuilder
        {
            builder.Id = 0;
            return builder;
        }

        public static T WithCreatorId<T>(this T builder, int creatorId)
            where T : IWithCreatorIdBuilder
        {
            builder.CreatorId = creatorId;
            return builder;
        }

        public static T WithCreateDate<T>(this T builder, DateTime createDate)
            where T : IWithCreateDateBuilder
        {
            builder.CreateDate = createDate;
            return builder;
        }

        public static T WithUpdateDate<T>(this T builder, DateTime updateDate)
            where T : IWithUpdateDateBuilder
        {
            builder.UpdateDate = updateDate;
            return builder;
        }

        public static T WithDeleteDate<T>(this T builder, DateTime deleteDate)
            where T : IWithDeleteDateBuilder
        {
            builder.DeleteDate = deleteDate;
            return builder;
        }

        public static T WithAlias<T>(this T builder, string alias)
            where T : IWithAliasBuilder
        {
            builder.Alias = alias;
            return builder;
        }

        public static T WithName<T>(this T builder, string name)
            where T : IWithNameBuilder
        {
            builder.Name = name;
            return builder;
        }

        public static T WithKey<T>(this T builder, Guid key)
            where T : IWithKeyBuilder
        {
            builder.Key = key;
            return builder;
        }

        public static T WithParentId<T>(this T builder, int parentId)
            where T : IWithParentIdBuilder
        {
            builder.ParentId = parentId;
            return builder;
        }

        public static T WithParentContentType<T>(this T builder, IContentTypeComposition parent)
            where T : IWithParentContentTypeBuilder
        {
            builder.Parent = parent;
            return builder;
        }

        public static T WithTrashed<T>(this T builder, bool trashed)
            where T : IWithTrashedBuilder
        {
            builder.Trashed = trashed;
            return builder;
        }

        public static T WithLevel<T>(this T builder, int level)
            where T : IWithLevelBuilder
        {
            builder.Level = level;
            return builder;
        }

        public static T WithPath<T>(this T builder, string path)
            where T : IWithPathBuilder
        {
            builder.Path = path;
            return builder;
        }

        public static T WithSortOrder<T>(this T builder, int sortOrder)
            where T : IWithSortOrderBuilder
        {
            builder.SortOrder = sortOrder;
            return builder;
        }

        public static T WithDescription<T>(this T builder, string description)
            where T : IWithDescriptionBuilder
        {
            builder.Description = description;
            return builder;
        }

        public static T WithIcon<T>(this T builder, string icon)
            where T : IWithIconBuilder
        {
            builder.Icon = icon;
            return builder;
        }

        public static T WithThumbnail<T>(this T builder, string thumbnail)
            where T : IWithThumbnailBuilder
        {
            builder.Thumbnail = thumbnail;
            return builder;
        }

        public static T WithLogin<T>(this T builder, string username, string rawPasswordValue)
            where T : IWithLoginBuilder
        {
            builder.Username = username;
            builder.RawPasswordValue = rawPasswordValue;
            return builder;
        }

        public static T WithEmail<T>(this T builder, string email)
            where T : IWithEmailBuilder
        {
            builder.Email = email;
            return builder;
        }

        public static T WithFailedPasswordAttempts<T>(this T builder, int failedPasswordAttempts)
            where T : IWithFailedPasswordAttemptsBuilder
        {
            builder.FailedPasswordAttempts = failedPasswordAttempts;
            return builder;
        }

        public static T WithIsApproved<T>(this T builder, bool isApproved)
            where T : IWithIsApprovedBuilder
        {
            builder.IsApproved = isApproved;
            return builder;
        }

        public static T WithIsLockedOut<T>(this T builder, bool isLockedOut, DateTime? lastLockoutDate = null)
            where T : IWithIsLockedOutBuilder
        {
            builder.IsLockedOut = isLockedOut;
            if (lastLockoutDate.HasValue)
            {
                builder.LastLockoutDate = lastLockoutDate.Value;
            }

            return builder;
        }

        public static T WithLastLoginDate<T>(this T builder, DateTime lastLoginDate)
            where T : IWithLastLoginDateBuilder
        {
            builder.LastLoginDate = lastLoginDate;
            return builder;
        }

        public static T WithLastPasswordChangeDate<T>(this T builder, DateTime lastPasswordChangeDate)
            where T : IWithLastPasswordChangeDateBuilder
        {
            builder.LastPasswordChangeDate = lastPasswordChangeDate;
            return builder;
        }

        public static T WithPropertyTypeIdsIncrementingFrom<T>(this T builder, int propertyTypeIdsIncrementingFrom)
            where T : IWithPropertyTypeIdsIncrementingFrom
        {
            builder.PropertyTypeIdsIncrementingFrom = propertyTypeIdsIncrementingFrom;
            return builder;
        }

        public static T WithIsContainer<T>(this T builder, bool isContainer)
            where T : IWithIsContainerBuilder
        {
            builder.IsContainer = isContainer;
            return builder;
        }

        public static T WithCultureInfo<T>(this T builder, string cultureCode)
            where T : IWithCultureInfoBuilder
        {
            builder.CultureInfo = CultureInfo.GetCultureInfo(cultureCode);
            return builder;
        }

        public static T WithSupportsPublishing<T>(this T builder, bool supportsPublishing)
            where T : IWithSupportsPublishing
        {
            builder.SupportsPublishing = supportsPublishing;
            return builder;
        }

        public static T WithPropertyValues<T>(this T builder, object propertyValues, string? culture = null, string? segment = null)
            where T : IWithPropertyValues
        {
            builder.PropertyValues = propertyValues;
            builder.PropertyValuesCulture = culture;
            builder.PropertyValuesSegment = segment;
            return builder;
        }

        public static T WithDate<T>(this T builder, DateTime date) where T : IWithDateBuilder
        {
            builder.Date = date;
            return builder;
        }
    }
}