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

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

namespace Umbraco.Cms.Tests.Common.Builders
{
    public class ContentCultureInfosCollectionBuilder : ChildBuilderBase<ContentBuilder, ContentCultureInfosCollection>, IBuildContentCultureInfosCollection
    {
        private readonly List<ContentCultureInfosBuilder> _cultureInfosBuilders;
        public ContentCultureInfosCollectionBuilder(ContentBuilder parentBuilder) : base(parentBuilder) => _cultureInfosBuilders = new List<ContentCultureInfosBuilder>();

        public ContentCultureInfosBuilder AddCultureInfos()
        {
            var builder = new ContentCultureInfosBuilder(this);
            _cultureInfosBuilders.Add(builder);
            return builder;
        }

        public override ContentCultureInfosCollection Build()
        {
            if (_cultureInfosBuilders.Count < 1)
            {
                throw new InvalidOperationException("You must add at least one culture infos to the collection builder");
            }
            var cultureInfosCollection = new ContentCultureInfosCollection();

            foreach (ContentCultureInfosBuilder cultureInfosBuilder in _cultureInfosBuilders)
            {
                cultureInfosCollection.Add(cultureInfosBuilder.Build());
            }

            return cultureInfosCollection;
        }
    }
}