Skip to content

Commit 0e1b4a2

Browse files
committed
Making the alter proxy test/example a bit more realistic
1 parent 43ea66b commit 0e1b4a2

File tree

2 files changed

+12
-4
lines changed

2 files changed

+12
-4
lines changed

NTestDataBuilder.Tests/AsProxyTests.cs

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -21,11 +21,13 @@ public void GivenBuilderIsSetAsProxy_WhenBuilding_AnNSubstituteProxyIsReturned()
2121
[Test]
2222
public void GivenBuilderThatAltersProxyIsSetAsProxy_WhenBuilding_TheProxyIsAltered()
2323
{
24-
var builder = new ProxyAlteringCustomerBuilder().AsProxy();
24+
var builder = new ProxyAlteringCustomerBuilder()
25+
.AsProxy()
26+
.HasBeenMemberForYears(10);
2527

2628
var proxy = builder.Build();
2729

28-
Assert.That(proxy.CustomerForHowManyYears(DateTime.Now), Is.EqualTo(ProxyAlteringCustomerBuilder.MemberFor));
30+
Assert.That(proxy.CustomerForHowManyYears(DateTime.Now), Is.EqualTo(10));
2931
}
3032
}
3133
}

NTestDataBuilder.Tests/Builders/ProxyAlteringCustomerBuilder.cs

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6,11 +6,17 @@ namespace NTestDataBuilder.Tests.Builders
66
{
77
class ProxyAlteringCustomerBuilder : DataBuilder<Customer, ProxyAlteringCustomerBuilder>
88
{
9-
public const int MemberFor = 4;
9+
private int _years;
10+
11+
public ProxyAlteringCustomerBuilder HasBeenMemberForYears(int years)
12+
{
13+
_years = years;
14+
return this;
15+
}
1016

1117
protected override void AlterProxy(Customer proxy)
1218
{
13-
proxy.CustomerForHowManyYears(Arg.Any<DateTime>()).Returns(MemberFor);
19+
proxy.CustomerForHowManyYears(Arg.Any<DateTime>()).Returns(_years);
1420
}
1521

1622
protected override Customer BuildObject()

0 commit comments

Comments
 (0)