Add tests for mockPrompt