sonarr-repo-only/NzbDrone.Core.Test/Framework/NBuilderExtensions.cs

using System.Collections.Generic;
using System.Linq;
using FizzWare.NBuilder;
using NzbDrone.Core.Datastore;

namespace NzbDrone.Core.Test.Framework
{
    public static class NBuilderExtensions
    {
        public static T BuildNew<T>(this ISingleObjectBuilder<T> builder) where T : ModelBase, new()
        {
            return builder.With(c => c.Id = 0).Build();
        }

        public static List<T> BuildList<T>(this IListBuilder<T> builder) where T : ModelBase, new()
        {
            return builder.Build().ToList();
        }

        public static List<T> BuildListOfNew<T>(this IListBuilder<T> builder) where T : ModelBase, new()
        {
            return BuildList<T>(builder.All().With(c => c.Id = 0));
        }

    }
}