diff --git a/src/borg/selftest.py b/src/borg/selftest.py index 0acbee19f..3a08d5be8 100644 --- a/src/borg/selftest.py +++ b/src/borg/selftest.py @@ -21,7 +21,7 @@ import sys import time from unittest import TestResult, TestSuite, defaultTestLoader -from .testsuite.crypto_test import CryptoTestCase +from .testsuite.crypto.crypto_test import CryptoTestCase from .testsuite.chunkers.buzhash_self_test import ChunkerTestCase from .testsuite.chunkers.fixed_self_test import ChunkerFixedTestCase diff --git a/src/borg/testsuite/archiver/key_cmds_test.py b/src/borg/testsuite/archiver/key_cmds_test.py index 79d933f22..173d92a1f 100644 --- a/src/borg/testsuite/archiver/key_cmds_test.py +++ b/src/borg/testsuite/archiver/key_cmds_test.py @@ -10,7 +10,7 @@ from ...helpers import CommandError from ...helpers import bin_to_hex, hex_to_bin from ...helpers import msgpack from ...repository import Repository -from ..key_test import TestKey +from ..crypto.key_test import TestKey from . import RK_ENCRYPTION, KF_ENCRYPTION, cmd, _extract_repository_id, _set_repository_id, generate_archiver_tests pytest_generate_tests = lambda metafunc: generate_archiver_tests(metafunc, kinds="local,remote,binary") # NOQA diff --git a/src/borg/testsuite/cache_test.py b/src/borg/testsuite/cache_test.py index e65735459..e6a5092ff 100644 --- a/src/borg/testsuite/cache_test.py +++ b/src/borg/testsuite/cache_test.py @@ -3,7 +3,7 @@ import os import pytest from .hashindex_test import H -from .key_test import TestKey +from .crypto.key_test import TestKey from ..archive import Statistics from ..cache import AdHocWithFilesCache from ..crypto.key import AESOCBRepoKey diff --git a/src/borg/testsuite/crypto/__init__.py b/src/borg/testsuite/crypto/__init__.py new file mode 100644 index 000000000..e69de29bb diff --git a/src/borg/testsuite/crypto_test.py b/src/borg/testsuite/crypto/crypto_test.py similarity index 97% rename from src/borg/testsuite/crypto_test.py rename to src/borg/testsuite/crypto/crypto_test.py index 7c6ba0b22..edaa8b9cf 100644 --- a/src/borg/testsuite/crypto_test.py +++ b/src/borg/testsuite/crypto/crypto_test.py @@ -4,14 +4,14 @@ from unittest.mock import MagicMock import unittest -from ..crypto.low_level import AES256_CTR_HMAC_SHA256, AES256_OCB, CHACHA20_POLY1305, UNENCRYPTED, IntegrityError -from ..crypto.low_level import bytes_to_long, bytes_to_int, long_to_bytes -from ..crypto.low_level import AES, hmac_sha256 +from ...crypto.low_level import AES256_CTR_HMAC_SHA256, AES256_OCB, CHACHA20_POLY1305, UNENCRYPTED, IntegrityError +from ...crypto.low_level import bytes_to_long, bytes_to_int, long_to_bytes +from ...crypto.low_level import AES, hmac_sha256 from hashlib import sha256 -from ..crypto.key import CHPOKeyfileKey, AESOCBRepoKey, FlexiKey, KeyBase, PlaintextKey -from ..helpers import msgpack, bin_to_hex +from ...crypto.key import CHPOKeyfileKey, AESOCBRepoKey, FlexiKey, KeyBase, PlaintextKey +from ...helpers import msgpack, bin_to_hex -from . import BaseTestCase +from .. import BaseTestCase class CryptoTestCase(BaseTestCase): diff --git a/src/borg/testsuite/file_integrity_test.py b/src/borg/testsuite/crypto/file_integrity_test.py similarity index 98% rename from src/borg/testsuite/file_integrity_test.py rename to src/borg/testsuite/crypto/file_integrity_test.py index 95add54ef..052bad81c 100644 --- a/src/borg/testsuite/file_integrity_test.py +++ b/src/borg/testsuite/crypto/file_integrity_test.py @@ -1,6 +1,6 @@ import pytest -from ..crypto.file_integrity import DetachedIntegrityCheckedFile, FileIntegrityError +from ...crypto.file_integrity import DetachedIntegrityCheckedFile, FileIntegrityError class TestReadIntegrityFile: diff --git a/src/borg/testsuite/key_test.py b/src/borg/testsuite/crypto/key_test.py similarity index 93% rename from src/borg/testsuite/key_test.py rename to src/borg/testsuite/crypto/key_test.py index 14c46a0c4..ca2884e29 100644 --- a/src/borg/testsuite/key_test.py +++ b/src/borg/testsuite/crypto/key_test.py @@ -4,20 +4,20 @@ from unittest.mock import MagicMock import pytest -from ..crypto.key import PlaintextKey, AuthenticatedKey, Blake2AuthenticatedKey -from ..crypto.key import RepoKey, KeyfileKey, Blake2RepoKey, Blake2KeyfileKey -from ..crypto.key import AEADKeyBase -from ..crypto.key import AESOCBRepoKey, AESOCBKeyfileKey, CHPORepoKey, CHPOKeyfileKey -from ..crypto.key import Blake2AESOCBRepoKey, Blake2AESOCBKeyfileKey, Blake2CHPORepoKey, Blake2CHPOKeyfileKey -from ..crypto.key import ID_HMAC_SHA_256, ID_BLAKE2b_256 -from ..crypto.key import UnsupportedManifestError, UnsupportedKeyFormatError -from ..crypto.key import identify_key -from ..crypto.low_level import IntegrityError as IntegrityErrorBase -from ..helpers import IntegrityError -from ..helpers import Location -from ..helpers import msgpack -from ..constants import KEY_ALGORITHMS -from ..helpers import hex_to_bin, bin_to_hex +from ...crypto.key import PlaintextKey, AuthenticatedKey, Blake2AuthenticatedKey +from ...crypto.key import RepoKey, KeyfileKey, Blake2RepoKey, Blake2KeyfileKey +from ...crypto.key import AEADKeyBase +from ...crypto.key import AESOCBRepoKey, AESOCBKeyfileKey, CHPORepoKey, CHPOKeyfileKey +from ...crypto.key import Blake2AESOCBRepoKey, Blake2AESOCBKeyfileKey, Blake2CHPORepoKey, Blake2CHPOKeyfileKey +from ...crypto.key import ID_HMAC_SHA_256, ID_BLAKE2b_256 +from ...crypto.key import UnsupportedManifestError, UnsupportedKeyFormatError +from ...crypto.key import identify_key +from ...crypto.low_level import IntegrityError as IntegrityErrorBase +from ...helpers import IntegrityError +from ...helpers import Location +from ...helpers import msgpack +from ...constants import KEY_ALGORITHMS +from ...helpers import hex_to_bin, bin_to_hex class TestKey: diff --git a/src/borg/testsuite/remote_test.py b/src/borg/testsuite/remote_test.py index 26b57e197..aeb0b33c7 100644 --- a/src/borg/testsuite/remote_test.py +++ b/src/borg/testsuite/remote_test.py @@ -14,7 +14,7 @@ from ..helpers import IntegrityError from ..repoobj import RepoObj from .hashindex_test import H from .repository_test import fchunk, pdchunk -from .key_test import TestKey +from .crypto.key_test import TestKey class TestSleepingBandwidthLimiter: