delete_cmd converted

This commit is contained in:
bigtedde 2023-07-04 16:53:06 -05:00
parent d18525ae49
commit 99bf56cfcd
1 changed files with 79 additions and 75 deletions

View File

@ -1,51 +1,61 @@
import unittest
from ...archive import Archive
from ...constants import * # NOQA
from ...manifest import Manifest
from ...repository import Repository
from . import ArchiverTestCaseBase, RemoteArchiverTestCaseBase, ArchiverTestCaseBinaryBase, RK_ENCRYPTION, BORG_EXES
from . import src_file
from . import cmd, create_regular_file, RK_ENCRYPTION, src_file, create_src_archive
class ArchiverTestCase(ArchiverTestCaseBase):
def test_delete(self):
self.create_regular_file("file1", size=1024 * 80)
self.create_regular_file("dir2/file2", size=1024 * 80)
self.cmd(f"--repo={self.repository_location}", "rcreate", RK_ENCRYPTION)
self.cmd(f"--repo={self.repository_location}", "create", "test", "input")
self.cmd(f"--repo={self.repository_location}", "create", "test.2", "input")
self.cmd(f"--repo={self.repository_location}", "create", "test.3", "input")
self.cmd(f"--repo={self.repository_location}", "create", "another_test.1", "input")
self.cmd(f"--repo={self.repository_location}", "create", "another_test.2", "input")
self.cmd(f"--repo={self.repository_location}", "extract", "test", "--dry-run")
self.cmd(f"--repo={self.repository_location}", "extract", "test.2", "--dry-run")
self.cmd(f"--repo={self.repository_location}", "delete", "--match-archives", "sh:another_*")
self.cmd(f"--repo={self.repository_location}", "delete", "--last", "1")
self.cmd(f"--repo={self.repository_location}", "delete", "-a", "test")
self.cmd(f"--repo={self.repository_location}", "extract", "test.2", "--dry-run")
output = self.cmd(f"--repo={self.repository_location}", "delete", "-a", "test.2", "--stats")
self.assert_in("Original size: -", output) # negative size == deleted data
def pytest_generate_tests(metafunc):
# Generates tests that run on local and remote repos, as well as with a binary base.
if "archivers" in metafunc.fixturenames:
metafunc.parametrize("archivers", ["archiver", "remote_archiver", "binary_archiver"])
def test_delete(archivers, request):
archiver = request.getfixturevalue(archivers)
repo_location, repo_path, input_path = archiver.repository_location, archiver.repository_path, archiver.input_path
create_regular_file(input_path, "file1", size=1024 * 80)
create_regular_file(input_path, "dir2/file2", size=1024 * 80)
cmd(archiver, f"--repo={repo_location}", "rcreate", RK_ENCRYPTION)
cmd(archiver, f"--repo={repo_location}", "create", "test", "input")
cmd(archiver, f"--repo={repo_location}", "create", "test.2", "input")
cmd(archiver, f"--repo={repo_location}", "create", "test.3", "input")
cmd(archiver, f"--repo={repo_location}", "create", "another_test.1", "input")
cmd(archiver, f"--repo={repo_location}", "create", "another_test.2", "input")
cmd(archiver, f"--repo={repo_location}", "extract", "test", "--dry-run")
cmd(archiver, f"--repo={repo_location}", "extract", "test.2", "--dry-run")
cmd(archiver, f"--repo={repo_location}", "delete", "--match-archives", "sh:another_*")
cmd(archiver, f"--repo={repo_location}", "delete", "--last", "1")
cmd(archiver, f"--repo={repo_location}", "delete", "-a", "test")
cmd(archiver, f"--repo={repo_location}", "extract", "test.2", "--dry-run")
output = cmd(archiver, f"--repo={repo_location}", "delete", "-a", "test.2", "--stats")
assert "Original size: -" in output # negative size == deleted data
# Make sure all data except the manifest has been deleted
with Repository(self.repository_path) as repository:
self.assert_equal(len(repository), 1)
with Repository(repo_path) as repository:
assert len(repository) == 1
def test_delete_multiple(self):
self.create_regular_file("file1", size=1024 * 80)
self.cmd(f"--repo={self.repository_location}", "rcreate", RK_ENCRYPTION)
self.cmd(f"--repo={self.repository_location}", "create", "test1", "input")
self.cmd(f"--repo={self.repository_location}", "create", "test2", "input")
self.cmd(f"--repo={self.repository_location}", "create", "test3", "input")
self.cmd(f"--repo={self.repository_location}", "delete", "-a", "test1")
self.cmd(f"--repo={self.repository_location}", "delete", "-a", "test2")
self.cmd(f"--repo={self.repository_location}", "extract", "test3", "--dry-run")
self.cmd(f"--repo={self.repository_location}", "delete", "-a", "test3")
assert not self.cmd(f"--repo={self.repository_location}", "rlist")
def test_delete_force(self):
self.cmd(f"--repo={self.repository_location}", "rcreate", "--encryption=none")
self.create_src_archive("test")
with Repository(self.repository_path, exclusive=True) as repository:
def test_delete_multiple(archivers, request):
archiver = request.getfixturevalue(archivers)
repo_location, input_path = archiver.repository_location, archiver.input_path
create_regular_file(input_path, "file1", size=1024 * 80)
cmd(archiver, f"--repo={repo_location}", "rcreate", RK_ENCRYPTION)
cmd(archiver, f"--repo={repo_location}", "create", "test1", "input")
cmd(archiver, f"--repo={repo_location}", "create", "test2", "input")
cmd(archiver, f"--repo={repo_location}", "create", "test3", "input")
cmd(archiver, f"--repo={repo_location}", "delete", "-a", "test1")
cmd(archiver, f"--repo={repo_location}", "delete", "-a", "test2")
cmd(archiver, f"--repo={repo_location}", "extract", "test3", "--dry-run")
cmd(archiver, f"--repo={repo_location}", "delete", "-a", "test3")
assert not cmd(archiver, f"--repo={repo_location}", "rlist")
def test_delete_force(archivers, request):
archiver = request.getfixturevalue(archivers)
repo_location, repo_path = archiver.repository_location, archiver.repository_path
cmd(archiver, f"--repo={repo_location}", "rcreate", "--encryption=none")
create_src_archive(archiver, "test")
with Repository(repo_path, exclusive=True) as repository:
manifest = Manifest.load(repository, Manifest.NO_OPERATION_CHECK)
archive = Archive(manifest, "test")
for item in archive.iter_items():
@ -55,31 +65,25 @@ class ArchiverTestCase(ArchiverTestCaseBase):
else:
assert False # missed the file
repository.commit(compact=False)
output = self.cmd(f"--repo={self.repository_location}", "delete", "-a", "test", "--force")
self.assert_in("deleted archive was corrupted", output)
self.cmd(f"--repo={self.repository_location}", "check", "--repair")
output = self.cmd(f"--repo={self.repository_location}", "rlist")
self.assert_not_in("test", output)
output = cmd(archiver, f"--repo={repo_location}", "delete", "-a", "test", "--force")
assert "deleted archive was corrupted" in output
cmd(archiver, f"--repo={repo_location}", "check", "--repair")
output = cmd(archiver, f"--repo={repo_location}", "rlist")
assert "test" not in output
def test_delete_double_force(self):
self.cmd(f"--repo={self.repository_location}", "rcreate", "--encryption=none")
self.create_src_archive("test")
with Repository(self.repository_path, exclusive=True) as repository:
def test_delete_double_force(archivers, request):
archiver = request.getfixturevalue(archivers)
repo_location, repo_path = archiver.repository_location, archiver.repository_path
cmd(archiver, f"--repo={repo_location}", "rcreate", "--encryption=none")
create_src_archive(archiver, "test")
with Repository(repo_path, exclusive=True) as repository:
manifest = Manifest.load(repository, Manifest.NO_OPERATION_CHECK)
archive = Archive(manifest, "test")
id = archive.metadata.items[0]
repository.put(id, b"corrupted items metadata stream chunk")
repository.commit(compact=False)
self.cmd(f"--repo={self.repository_location}", "delete", "-a", "test", "--force", "--force")
self.cmd(f"--repo={self.repository_location}", "check", "--repair")
output = self.cmd(f"--repo={self.repository_location}", "rlist")
self.assert_not_in("test", output)
class RemoteArchiverTestCase(RemoteArchiverTestCaseBase, ArchiverTestCase):
"""run the same tests, but with a remote repository"""
@unittest.skipUnless("binary" in BORG_EXES, "no borg.exe available")
class ArchiverTestCaseBinary(ArchiverTestCaseBinaryBase, ArchiverTestCase):
"""runs the same tests, but via the borg binary"""
cmd(archiver, f"--repo={repo_location}", "delete", "-a", "test", "--force", "--force")
cmd(archiver, f"--repo={repo_location}", "check", "--repair")
output = cmd(archiver, f"--repo={repo_location}", "rlist")
assert "test" not in output