def test_do_prompt_empty_default(self): """Tests Var.do_prompt() with empty default""" resps = [''] self.input_mock.side_effect = lambda x: resps.pop(0) var = Var('foo', default='') self.assertEqual(var.do_prompt(), '') self.assertEqual(self.input_mock.call_count, 1)
def test_do_prompt(self): """Tests Var.do_prompt() """ resps = ['', 'bar'] self.input_mock.side_effect = lambda x: resps.pop(0) var = Var('foo') self.assertEqual(var.do_prompt(), 'bar') self.assertEqual(self.input_mock.call_count, 2)