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)); } } }