def test_get_filename_none_left(self):
import solescion.utils.screenshot as screenshot_module
orig = screenshot_module.listdir
screenshot_module.listdir = \
lambda *_: ["A%03d.png" % idx for idx in range(101)]
try:
self.assertNone(_get_filename("A", ".png"), "expected None")
finally:
screenshot_module.listdir = orig
def test_get_filename(self):
import solescion.utils.screenshot as screenshot_module
orig = screenshot_module.listdir
screenshot_module.listdir = lambda *_: ["A000.png"]
try:
self.assertEquals(
_get_filename("A", ".png"),
join(expanduser("~"), "A001.png"),
"bad filename")
finally:
screenshot_module.listdir = orig
