Skip to content

Commit 7aabe82

Browse files
fix binary_archiver tests for modern exit codes
1 parent edfca29 commit 7aabe82

File tree

6 files changed

+27
-12
lines changed

6 files changed

+27
-12
lines changed

src/borg/testsuite/archiver/config_cmd.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -52,7 +52,8 @@ def test_config(archivers, request):
5252

5353
cmd(archiver, "config", "--list", "--delete", exit_code=2)
5454
if archiver.FORK_DEFAULT:
55-
cmd(archiver, "config", exit_code=2)
55+
expected_ec = CommandError().exit_code
56+
cmd(archiver, "config", exit_code=expected_ec)
5657
else:
5758
with pytest.raises(CommandError):
5859
cmd(archiver, "config")

src/borg/testsuite/archiver/create_cmd.py

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -366,7 +366,10 @@ def test_create_content_from_command_with_failed_command(archivers, request):
366366
archiver = request.getfixturevalue(archivers)
367367
cmd(archiver, "rcreate", RK_ENCRYPTION)
368368
if archiver.FORK_DEFAULT:
369-
output = cmd(archiver, "create", "--content-from-command", "test", "--", "sh", "-c", "exit 73;", exit_code=2)
369+
expected_ec = CommandError().exit_code
370+
output = cmd(
371+
archiver, "create", "--content-from-command", "test", "--", "sh", "-c", "exit 73;", exit_code=expected_ec
372+
)
370373
assert output.endswith("Command 'sh' exited with status 73" + os.linesep)
371374
else:
372375
with pytest.raises(CommandError):
@@ -418,7 +421,10 @@ def test_create_paths_from_command_with_failed_command(archivers, request):
418421
archiver = request.getfixturevalue(archivers)
419422
cmd(archiver, "rcreate", RK_ENCRYPTION)
420423
if archiver.FORK_DEFAULT:
421-
output = cmd(archiver, "create", "--paths-from-command", "test", "--", "sh", "-c", "exit 73;", exit_code=2)
424+
expected_ec = CommandError().exit_code
425+
output = cmd(
426+
archiver, "create", "--paths-from-command", "test", "--", "sh", "-c", "exit 73;", exit_code=expected_ec
427+
)
422428
assert output.endswith("Command 'sh' exited with status 73" + os.linesep)
423429
else:
424430
with pytest.raises(CommandError):

src/borg/testsuite/archiver/key_cmds.py

Lines changed: 11 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66
from ...constants import * # NOQA
77
from ...crypto.key import AESOCBRepoKey, AESOCBKeyfileKey, CHPOKeyfileKey, Passphrase
88
from ...crypto.keymanager import RepoIdMismatch, NotABorgKeyFile
9-
from ...helpers import EXIT_ERROR, CommandError
9+
from ...helpers import CommandError
1010
from ...helpers import bin_to_hex, hex_to_bin
1111
from ...helpers import msgpack
1212
from ...repository import Repository
@@ -171,7 +171,8 @@ def test_key_export_directory(archivers, request):
171171
os.mkdir(export_directory)
172172
cmd(archiver, "rcreate", RK_ENCRYPTION)
173173
if archiver.FORK_DEFAULT:
174-
cmd(archiver, "key", "export", export_directory, exit_code=EXIT_ERROR)
174+
expected_ec = CommandError().exit_code
175+
cmd(archiver, "key", "export", export_directory, exit_code=expected_ec)
175176
else:
176177
with pytest.raises(CommandError):
177178
cmd(archiver, "key", "export", export_directory)
@@ -183,7 +184,8 @@ def test_key_export_qr_directory(archivers, request):
183184
os.mkdir(export_directory)
184185
cmd(archiver, "rcreate", RK_ENCRYPTION)
185186
if archiver.FORK_DEFAULT:
186-
cmd(archiver, "key", "export", "--qr-html", export_directory, exit_code=EXIT_ERROR)
187+
expected_ec = CommandError().exit_code
188+
cmd(archiver, "key", "export", "--qr-html", export_directory, exit_code=expected_ec)
187189
else:
188190
with pytest.raises(CommandError):
189191
cmd(archiver, "key", "export", "--qr-html", export_directory)
@@ -194,7 +196,8 @@ def test_key_import_errors(archivers, request):
194196
export_file = archiver.output_path + "/exported"
195197
cmd(archiver, "rcreate", KF_ENCRYPTION)
196198
if archiver.FORK_DEFAULT:
197-
cmd(archiver, "key", "import", export_file, exit_code=EXIT_ERROR)
199+
expected_ec = CommandError().exit_code
200+
cmd(archiver, "key", "import", export_file, exit_code=expected_ec)
198201
else:
199202
with pytest.raises(CommandError):
200203
cmd(archiver, "key", "import", export_file)
@@ -203,7 +206,8 @@ def test_key_import_errors(archivers, request):
203206
fd.write("something not a key\n")
204207

205208
if archiver.FORK_DEFAULT:
206-
cmd(archiver, "key", "import", export_file, exit_code=2)
209+
expected_ec = NotABorgKeyFile().exit_code
210+
cmd(archiver, "key", "import", export_file, exit_code=expected_ec)
207211
else:
208212
with pytest.raises(NotABorgKeyFile):
209213
cmd(archiver, "key", "import", export_file)
@@ -212,7 +216,8 @@ def test_key_import_errors(archivers, request):
212216
fd.write("BORG_KEY a0a0a0\n")
213217

214218
if archiver.FORK_DEFAULT:
215-
cmd(archiver, "key", "import", export_file, exit_code=2)
219+
expected_ec = RepoIdMismatch().exit_code
220+
cmd(archiver, "key", "import", export_file, exit_code=expected_ec)
216221
else:
217222
with pytest.raises(RepoIdMismatch):
218223
cmd(archiver, "key", "import", export_file)

src/borg/testsuite/archiver/rcreate_cmd.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -56,7 +56,8 @@ def test_rcreate_nested_repositories(archivers, request):
5656
cmd(archiver, "rcreate", RK_ENCRYPTION)
5757
archiver.repository_location += "/nested"
5858
if archiver.FORK_DEFAULT:
59-
cmd(archiver, "rcreate", RK_ENCRYPTION, exit_code=2)
59+
expected_ec = Repository.AlreadyExists().exit_code
60+
cmd(archiver, "rcreate", RK_ENCRYPTION, exit_code=expected_ec)
6061
else:
6162
with pytest.raises(Repository.AlreadyExists):
6263
cmd(archiver, "rcreate", RK_ENCRYPTION)

src/borg/testsuite/archiver/rdelete_cmd.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,8 @@ def test_delete_repo(archivers, request):
1818
cmd(archiver, "create", "test.2", "input")
1919
os.environ["BORG_DELETE_I_KNOW_WHAT_I_AM_DOING"] = "no"
2020
if archiver.FORK_DEFAULT:
21-
cmd(archiver, "rdelete", exit_code=2)
21+
expected_ec = CancelledByUser().exit_code
22+
cmd(archiver, "rdelete", exit_code=expected_ec)
2223
else:
2324
with pytest.raises(CancelledByUser):
2425
cmd(archiver, "rdelete")

src/borg/testsuite/archiver/recreate_cmd.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -84,7 +84,8 @@ def test_recreate_target_rc(archivers, request):
8484
archiver = request.getfixturevalue(archivers)
8585
cmd(archiver, "rcreate", RK_ENCRYPTION)
8686
if archiver.FORK_DEFAULT:
87-
output = cmd(archiver, "recreate", "--target=asdf", exit_code=2)
87+
expected_ec = CommandError().exit_code
88+
output = cmd(archiver, "recreate", "--target=asdf", exit_code=expected_ec)
8889
assert "Need to specify single archive" in output
8990
else:
9091
with pytest.raises(CommandError):

0 commit comments

Comments
 (0)