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

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

using System;
using System.Collections.Generic;
using System.Linq;
using Umbraco.Cms.Core.Models;
using Umbraco.Cms.Tests.Common.Builders.Interfaces;

namespace Umbraco.Cms.Tests.Common.Builders
{
    public class DictionaryItemBuilder
        : BuilderBase<DictionaryItem>,
            IWithIdBuilder,
            IWithCreateDateBuilder,
            IWithUpdateDateBuilder,
            IWithDeleteDateBuilder,
            IWithKeyBuilder
    {
        private readonly List<DictionaryTranslationBuilder> _translationBuilders =
            new List<DictionaryTranslationBuilder>();

        private DateTime? _createDate;
        private DateTime? _deleteDate;
        private int? _id;
        private string _itemKey;
        private Guid? _key;
        private Guid? _parentId;
        private DateTime? _updateDate;

        DateTime? IWithCreateDateBuilder.CreateDate
        {
            get => _createDate;
            set => _createDate = value;
        }

        DateTime? IWithDeleteDateBuilder.DeleteDate
        {
            get => _deleteDate;
            set => _deleteDate = value;
        }

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

        Guid? IWithKeyBuilder.Key
        {
            get => _key;
            set => _key = value;
        }

        DateTime? IWithUpdateDateBuilder.UpdateDate
        {
            get => _updateDate;
            set => _updateDate = value;
        }

        public override DictionaryItem Build()
        {
            DateTime createDate = _createDate ?? DateTime.Now;
            DateTime updateDate = _updateDate ?? DateTime.Now;
            DateTime? deleteDate = _deleteDate ?? null;
            var id = _id ?? 1;
            Guid key = _key ?? Guid.NewGuid();
            Guid? parentId = _parentId ?? null;
            var itemKey = _itemKey ?? Guid.NewGuid().ToString();

            var result = new DictionaryItem(itemKey)
            {
                Translations = _translationBuilders.Select(x => x.Build()),
                CreateDate = createDate,
                UpdateDate = updateDate,
                DeleteDate = deleteDate,
                Id = id,
                ParentId = parentId,
                Key = key,
            };
            return result;
        }

        public DictionaryItemBuilder WithParentId(Guid parentId)
        {
            _parentId = parentId;
            return this;
        }

        public DictionaryItemBuilder WithItemKey(string itemKey)
        {
            _itemKey = itemKey;
            return this;
        }

        public DictionaryTranslationBuilder AddTranslation()
        {
            var builder = new DictionaryTranslationBuilder(this);

            _translationBuilders.Add(builder);

            return builder;
        }

        public DictionaryItemBuilder WithRandomTranslations(int count)
        {
            for (var i = 0; i < count; i++)
            {
                AddTranslation().Done();
            }

            return this;
        }
    }
}