More unit tests.

[vaibhavsagar]

We have alright integration tests to check that the system as a whole seems to be doing the right thing, but we have minimal unit tests that check that individual units are working correctly. This would be useful if/when things break so we can isolate the change that is causing the problem, something that the current tests are very bad at.