Adds additional mocking tests