Add some unit tests. These use the python-mock framework