Mock unit tests

#2