Add tests for Mock