Time for some unit tests