Add additional mocking tests