def test_find_unused_filename(self):
filesystem = MockFileSystem({"dir/foo.jpg": "", "dir/foo-1.jpg": "", "dir/foo-2.jpg": ""})
workspace = Workspace(filesystem, None)
self.assertEqual(workspace.find_unused_filename("bar", "bar", "bar"), "bar/bar.bar")
self.assertIsNone(workspace.find_unused_filename("dir", "foo", "jpg", search_limit=1))
self.assertIsNone(workspace.find_unused_filename("dir", "foo", "jpg", search_limit=2))
self.assertEqual(workspace.find_unused_filename("dir", "foo", "jpg"), "dir/foo-3.jpg")
def test_find_unused_filename(self):
filesystem = MockFileSystem({
"dir/foo.jpg": "",
"dir/foo-1.jpg": "",
"dir/foo-2.jpg": "",
})
workspace = Workspace(filesystem, None)
self.assertEqual(workspace.find_unused_filename("bar", "bar", "bar"), "bar/bar.bar")
self.assertEqual(workspace.find_unused_filename("dir", "foo", "jpg", search_limit=1), None)
self.assertEqual(workspace.find_unused_filename("dir", "foo", "jpg", search_limit=2), None)
self.assertEqual(workspace.find_unused_filename("dir", "foo", "jpg"), "dir/foo-3.jpg")
def test_create_zip(self):
workspace = Workspace(None, MockExecutive(should_log=True))
class MockZipFile(object):
def __init__(self, path):
self.filename = path
with OutputCapture(level=logging.INFO) as captured:
archive = workspace.create_zip('/zip/path', '/source/path',
MockZipFile)
self.assertEqual(
captured.root.log.getvalue(),
"MOCK run_command: ['zip', '-9', '-r', '/zip/path', '.'], cwd=/source/path\n"
)
self.assertEqual(archive.filename, "/zip/path")
def test_create_zip_exception(self):
workspace = Workspace(None, MockExecutive(should_log=True, should_throw=True))
expected_stderr = "MOCK run_command: ['zip', '-9', '-r', '/zip/path', '.'], cwd=/source/path\n"
class MockZipFile(object):
def __init__(self, path):
self.filename = path
archive = OutputCapture().assert_outputs(self, workspace.create_zip, ["/zip/path", "/source/path", MockZipFile], expected_stderr=expected_stderr)
self.assertEqual(archive, None)
def test_create_zip_exception(self):
workspace = Workspace(None, MockExecutive(should_log=True, should_throw=True))
expected_logs = """MOCK run_command: ['zip', '-9', '-r', '/zip/path', '.'], cwd=/source/path
Workspace.create_zip failed in /source/path:
MOCK ScriptError
MOCK output of child process
"""
class MockZipFile(object):
def __init__(self, path):
self.filename = path
archive = OutputCapture().assert_outputs(self, workspace.create_zip, ["/zip/path", "/source/path", MockZipFile], expected_logs=expected_logs)
self.assertIsNone(archive)
def test_create_zip_exception(self):
workspace = Workspace(
None, MockExecutive(should_log=True, should_throw=True))
class MockZipFile(object):
def __init__(self, path):
self.filename = path
with OutputCapture(level=logging.INFO) as captured:
archive = workspace.create_zip('/zip/path', '/source/path',
MockZipFile)
self.assertEqual(
captured.root.log.getvalue(),
'''MOCK run_command: ['zip', '-9', '-r', '/zip/path', '.'], cwd=/source/path
Workspace.create_zip failed in /source/path:
MOCK ScriptError
MOCK output of child process
''',
)
self.assertIsNone(archive)