def test_input_prompt(self): var=shellutils.get_input("Test Prompt") self.assertEqual(MockRawInput.prompt, "Test Prompt: ") self.assertEqual(var, "Test Raw Input")
def test_input_require(self): MockRawInput.values=["", "Test asdfasd"] var=shellutils.get_input("Test Prompt", require=True) self.assertEqual(MockRawInput.prompt, "Test Prompt: ") self.assertEqual(var, "Test asdfasd") self.assertEqual(MockSys.stderr.getvalue(), "An answer is required\n")
def test_input_restrict_default(self): MockRawInput.values=["baz", "", "bar"] var=shellutils.get_input("Test Prompt", require=True, restrict=["foo", "bar"], default="foo") self.assertEqual(MockRawInput.prompt, "Test Prompt (foo,bar) [foo]: ") self.assertEqual(var, "foo") self.assertEqual(MockSys.stderr.getvalue(), "Answer must be one of foo, bar\n")
def test_input_default(self): MockRawInput.values=[""] var=shellutils.get_input("Test Prompt", default="Hello world") self.assertEqual(MockRawInput.prompt, "Test Prompt [Hello world]: ") self.assertEqual(var, "Hello world")