使用AutoPoco生成依赖于另一个值的值

时间:2011-07-10 18:47:59

标签: c# .net unit-testing autopoco

我正在使用伟大的.NET库AutoPoco来创建测试和种子数据。

在我的模型中,我有2个日期属性,StartDate和EndDate。

我希望EndDate在开始日期后3小时。

我在下面为autopoco创建了一个自定义数据源,返回最小和最大日期之间的随机日期时间

class DefaultRandomDateSource : DatasourceBase<DateTime>
{
    private DateTime _MaxDate { get; set; }
    private DateTime _MinDate { get; set; }
    private Random _random { get; set; }


    public DefaultRandomDateSource(DateTime MaxDate, DateTime MinDate)
    {
        _MaxDate = MaxDate;
        _MinDate = MinDate;

    }

    public override DateTime Next(IGenerationSession session)
    {
        var tspan = _MaxDate - _MinDate;
        var rndSpan = new TimeSpan(0, _random.Next(0, (int) tspan.TotalMinutes), 0);


        return _MinDate + rndSpan;
    }

}

但是在AutoPoco的配置中,如何在自动生成的开始日期后3小时让我的EndDate说出来?

这是autopoco配置

 IGenerationSessionFactory factory = AutoPocoContainer.Configure(x =>
            {
                x.Conventions(c => { c.UseDefaultConventions(); });
                x.AddFromAssemblyContainingType<Meeting>();
                x.Include<Meeting>()
                    .Setup((c => c.CreatedBy)).Use<FirstNameSource>()
                    .Setup(c => c.StartDate).Use<DefaultRandomDateSource>(DateTime.Parse("21/05/2011"), DateTime.Parse("21/05/2012"));
            });

1 个答案:

答案 0 :(得分:1)

如果我正确理解您需要的问题:EndDate 设置StartDate。我不得不创建一个新的DataSource并获取我们正在构建的当前项目并从中读取值。我没有彻底检查过,但如果在StartDate之后设置了EndDate,它可能会失败(虽然我认为属性是按照它们设置的顺序设置的,但是请阅读AutoPoco的源代码)。此外,我使用的是CodePlex截至今天(2012年2月20日)的最新版本。

public class MeetingsGenerator
{
    public static IList<Meeting> CreateMeeting(int count)
    {
        var factory = AutoPocoContainer.Configure(x =>
        {
            x.Conventions(c => { c.UseDefaultConventions(); });
            x.Include<Meeting>()
                .Setup((c => c.CreatedBy)).Use<FirstNameSource>()
                .Setup(c => c.StartDate).Use<DefaultRandomDateSource>
                                  (DateTime.Parse("21/May/2012"), 
                                    DateTime.Parse("21/May/2011"))
                .Setup(c => c.EndDate).Use<MeetingEndDateSource>(0, 8);
        });
        return factory.CreateSession().List<Meeting>(count).Get();
    }
}

public class Meeting
{
    public string CreatedBy { get; set; }
    public DateTime StartDate { get; set; }
    public DateTime EndDate { get; set; }
}

public class MeetingEndDateSource : DatasourceBase<DateTime>
{
    private int mMin;
    private int mMax;
    private Random mRandom = new Random(1337);

    public MeetingEndDateSource(int min, int max)
    {
        mMin = min;
        mMax = max;
    }

    public override DateTime Next(IGenerationContext context)
    {
        var node = (TypeGenerationContextNode)((context.Node).Parent);
        var item = node.Target) as Meeting;

        if (item == null)
            return DateTime.Now;

        return item.StartDate.AddHours(mRandom.Next(mMin, mMax + 1));
    }
}

class DefaultRandomDateSource : DatasourceBase<DateTime>
{
    private DateTime _MaxDate { get; set; }
    private DateTime _MinDate { get; set; }
    private Random _random = new Random(1337);

    public DefaultRandomDateSource(DateTime MaxDate, DateTime MinDate)
    {
        _MaxDate = MaxDate;
        _MinDate = MinDate;
    }

    public override DateTime Next(IGenerationContext context)
    {
        var tspan = _MaxDate - _MinDate;

        var rndSpan = new TimeSpan(0 
                                  , _random.Next(0, (int)tspan.TotalMinutes)
                                  , 0);

        return _MinDate + rndSpan;
    }
}