Added some basic mock unit testing.

This was just to try out the unit testing