etorres/rbackup
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Test all parts of Repository.cleanup()

Browse Source
This commit is contained in:
Eric Torres 2019-04-12 07:51:08 -07:00
parent aee9848c80
commit 9a2935b4da
1 changed files with 23 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
tests/test_repository.py
View File

@ -227,6 +227,8 @@ class TestRepositoryCleanup(unittest.TestCase):
self.mocked_shutil = self.patched_shutil.start()
self.mocked_snapshot = self.patched_snapshot.start()
self.mocked_shutil.rmtree.avoids_symlink_attacks = True
def test_stops_on_non_symlink_resistant(self):
self.mocked_shutil.rmtree.avoids_symlink_attacks = False
repo = Repository("backup")
@ -236,6 +238,27 @@ class TestRepositoryCleanup(unittest.TestCase):
self.mocked_path.return_value.unlink.assert_not_called()
self.mocked_shutil.rmtree.assert_not_called()
def test_removes_metadata_by_default(self):
repo = Repository('backup')
repo.cleanup()
self.mocked_path.return_value.unlink.assert_called_once()
def test_removes_snapshots(self):
repo = Repository('backup')
repo.cleanup(remove_snapshots=True)
self.mocked_shutil.rmtree.assert_called_once()
def test_removes_repo_dir(self):
repo = Repository('backup')
repo.cleanup(remove_repo_dir=True)
self.mocked_shutil.rmtree.assert_called_once()
def tearDown(self):
self.patched_path.stop()
self.patched_r_metadata.stop()