Add unit tests

[JDGrimes]

We really need unit tests here.

[JDGrimes]

I just can't get the acceptance tests to work properly. I guess we'll have to flush them out further later.