def testRemove(self):
gs_path = 'gs://bucket/some/path'
local_path = '/some/local/path'
http_path = 'http://host.domain/some/path'
self.mox.StubOutWithMock(gslib, 'Remove')
self.mox.StubOutWithMock(filelib, 'Remove')
# Set up the test replay script.
# Run 1, two local files.
filelib.Remove(local_path, local_path + '.1')
# Run 2, local and GS.
gslib.Remove(local_path, gs_path, ignore_no_match=True)
# Run 4, GS and GS
gslib.Remove(gs_path,
gs_path + '.1',
ignore_no_match=True,
recurse=True)
# Run 7, local and HTTP
self.mox.ReplayAll()
# Run the test verification.
urilib.Remove(local_path, local_path + '.1')
urilib.Remove(local_path, gs_path, ignore_no_match=True)
urilib.Remove(gs_path,
gs_path + '.1',
ignore_no_match=True,
recurse=True)
self.assertRaises(urilib.NotSupportedForTypes, urilib.Remove,
local_path, http_path)
self.mox.VerifyAll()
def Remove(*args, **kwargs):
"""Delete the file(s) at uris, or directory(s) with recurse set.
Args:
args: One or more URIs.
ignore_no_match: If True, then do not complain if anything was not
removed because no URI match was found. Like rm -f. Defaults to False.
recurse: Remove recursively starting at path. Same as rm -R. Defaults
to False.
"""
uri_types = set([GetUriType(u) for u in args])
if TYPE_GS in uri_types:
# GS support only allows local files among list.
if len(uri_types) == 1 or (TYPE_LOCAL in uri_types and len(uri_types) == 2):
return gslib.Remove(*args, **kwargs)
if TYPE_LOCAL in uri_types and len(uri_types) == 1:
return filelib.Remove(*args, **kwargs)
raise NotSupportedForTypes(*list(uri_types))
def testRemove(self):
# pylint: disable=E1101
path1 = os.path.join(self.tempdir, 'file1')
path2 = os.path.join(self.tempdir, 'file2')
missing_path = os.path.join(self.tempdir, 'missing')
subdir = os.path.join(self.tempdir, 'subdir')
subpath1 = os.path.join(subdir, 'file3')
subpath2 = os.path.join(subdir, 'file4')
# Test remove on path that does not exist.
self.assertRaises(filelib.MissingFileError, filelib.Remove, path1)
self.assertFalse(filelib.Remove(path1, ignore_no_match=True))
# Test remove on simple file.
self._CreateSimpleFile(path1)
self.assertTrue(filelib.Remove(path1))
self.assertRaises(filelib.MissingFileError, filelib.Remove, path1)
self.assertFalse(filelib.Remove(path1, ignore_no_match=True))
# Test remove on more than one file.
self._CreateSimpleFile(path1, path2)
self.assertTrue(filelib.Remove(path1, path2))
# Test remove on multiple files, with one missing.
self._CreateSimpleFile(path1, path2)
self.assertRaises(filelib.MissingFileError, filelib.Remove,
path1, missing_path, path2)
# First path1 removed, but path2 not because it was after missing.
self.assertFalse(filelib.Exists(path1))
self.assertTrue(filelib.Exists(path2))
# Test remove multiple files, one missing, with ignore_no_match True.
self._CreateSimpleFile(path1, path2)
self.assertFalse(filelib.Remove(path1, missing_path, path2,
ignore_no_match=True))
self.assertFalse(filelib.Exists(path1))
self.assertFalse(filelib.Exists(path2))
# Test recursive Remove.
os.makedirs(subdir)
self._CreateSimpleFile(path1, path2, subpath1, subpath2)
self.assertTrue(filelib.Remove(path1, path2, subdir, recurse=True))
self.assertFalse(filelib.Exists(path1))
self.assertFalse(filelib.Exists(subpath1))
# Test recursive Remove with one missing path.
os.makedirs(subdir)
self._CreateSimpleFile(path1, path2, subpath1, subpath2)
self.assertRaises(filelib.MissingFileError, filelib.Remove,
path1, subdir, missing_path, path2, recurse=True)
self.assertFalse(filelib.Exists(path1))
self.assertTrue(filelib.Exists(path2))
self.assertFalse(filelib.Exists(subpath1))
# Test recursive Remove with one missing path and ignore_no_match True.
os.makedirs(subdir)
self._CreateSimpleFile(path1, path2, subpath1, subpath2)
self.assertFalse(filelib.Remove(path1, subdir, missing_path, path2,
recurse=True, ignore_no_match=True))
self.assertFalse(filelib.Exists(path1))
self.assertFalse(filelib.Exists(path2))
self.assertFalse(filelib.Exists(subpath1))