More mock unit tests