Add test for prompt.