Unit Testing Time

Date Published: 07 July 2008

Unit Testing Time

Oren has a post showing how he deals with time sensitive code in his unit tests. One thing that’s interesting is that, like my previous post, deals with the System.Func construct introduced in .NET 3.5. I see this convention more and more and it’s really growing on me. I’ve dealt with timing issues in my own code using a convention similar to the one Ayende demonstrates, which is to create a singleton or static property for the system time, and consistently reference this time everywhere rather than DateTime.Now(). References to DateTime.Now() are dependencies that should be abstracted away through some form of Dependency Injection or similar technique. Writing unit tests that look like this is icky (trust me, I know, this is from some real code of mine that wraps HttpRuntime.Cache):

[TestMethod]
public void TimeDependency()
{
    myCache.Insert(myKey, myValue, cacheSeconds);
    Assert.AreEqual(myValue, myCache[myKey], "Value removed immediately - should still be there.");
    System.Threading.Thread.Sleep(new TimeSpan(0, 0, 0, cacheSeconds).Add(TimeSpan.FromMilliseconds(500)));
    Assert.IsNull(myCache[myKey], "Value should be expired from cache at this point.");
}

Any time you’re having to put Thread.Sleep() in your unit test, that’s a big sign something’s wrong.

To correct this, what I have done in the past is define a Context.ServerTime property of type DateTime which I can set, but which defaults to DateTime.Now(). However, Oren’s technique of actually using a delegate for the time, which can be replaced with either a static value or a new method for computing the current time, seems like a definite improvement (and looks like this):

public static class SystemTime 
{
    public static Func<DateTime> Now = () => DateTime.Now; 
}

With this code in place, changing the definition of “now” in a unit test involves simply one line of code:

SystemTime.Now = () =&gt; new DateTime(2008,1,1);

The () => syntax is simply a parameterless delegate. I’m really loving these.

Steve Smith

About Ardalis

Software Architect

Steve is an experienced software architect and trainer, focusing on code quality and Domain-Driven Design with .NET.