diff --git a/test/test_util.py b/test/test_util.py index 25a591d1..d5f156bb 100644 --- a/test/test_util.py +++ b/test/test_util.py @@ -10,6 +10,7 @@ import os import sys import unittest +from unittest.mock import patch import io import re @@ -989,6 +990,16 @@ value = 123 self.assertIs(proxy["abc.def.ghi"], util.NONE) self.assertIs(proxy["os.path2"], util.NONE) + def test_lazy_prompt(self): + prompt = util.LazyPrompt() + + with patch("getpass.getpass") as p: + p.return_value = "***" + result = str(prompt) + + self.assertEqual(result, "***") + p.assert_called_once_with() + def test_null_context(self): with util.NullContext(): pass