From b8642439cded4aa53c34dc4710ce202ce964fe3a Mon Sep 17 00:00:00 2001 From: Dan Helfman Date: Mon, 22 Dec 2025 15:46:28 -0800 Subject: [PATCH 1/4] Fix for the database "restore" action restoring more databases than the "--database" flag specifies (#1208). --- NEWS | 4 ++++ borgmatic/actions/restore.py | 4 ++++ borgmatic/hooks/data_source/bootstrap.py | 4 ++++ borgmatic/hooks/data_source/btrfs.py | 4 ++++ borgmatic/hooks/data_source/dump.py | 2 +- borgmatic/hooks/data_source/lvm.py | 4 ++++ borgmatic/hooks/data_source/mariadb.py | 18 +++++++++++++++--- borgmatic/hooks/data_source/mongodb.py | 18 +++++++++++++++--- borgmatic/hooks/data_source/mysql.py | 18 +++++++++++++++--- borgmatic/hooks/data_source/postgresql.py | 18 +++++++++++++++--- borgmatic/hooks/data_source/sqlite.py | 18 +++++++++++++++--- borgmatic/hooks/data_source/zfs.py | 4 ++++ pyproject.toml | 2 +- tests/unit/actions/test_restore.py | 16 ++++++++++++++++ tests/unit/hooks/data_source/test_dump.py | 2 +- 15 files changed, 118 insertions(+), 18 deletions(-) diff --git a/NEWS b/NEWS index 185b4483..4b26ba95 100644 --- a/NEWS +++ b/NEWS @@ -1,3 +1,7 @@ +2.0.14.dev0 + * #1208: Fix for the database "restore" action restoring more databases than the "--database" flag + specifies. + 2.0.13 * #1054: Allow the Btrfs hook to create and delete snapshots even when running as a non-root user. See the documentation for more information: diff --git a/borgmatic/actions/restore.py b/borgmatic/actions/restore.py index 40ba95ca..8d0e3c7c 100644 --- a/borgmatic/actions/restore.py +++ b/borgmatic/actions/restore.py @@ -200,6 +200,10 @@ def restore_single_dump( borgmatic.hooks.dispatch.Hook_type.DATA_SOURCE, borgmatic_runtime_directory, data_source['name'], + data_source.get('hostname'), + data_source.get('port'), + data_source.get('container'), + data_source.get('label'), )[hook_name.split('_databases', 1)[0]] destination_path = ( diff --git a/borgmatic/hooks/data_source/bootstrap.py b/borgmatic/hooks/data_source/bootstrap.py index c1add4fe..6765d33c 100644 --- a/borgmatic/hooks/data_source/bootstrap.py +++ b/borgmatic/hooks/data_source/bootstrap.py @@ -116,6 +116,10 @@ def make_data_source_dump_patterns( config, borgmatic_runtime_directory, name=None, + hostname=None, + port=None, + container=None, + label=None, ): # pragma: no cover ''' Restores are implemented via the separate, purpose-specific "bootstrap" action rather than the diff --git a/borgmatic/hooks/data_source/btrfs.py b/borgmatic/hooks/data_source/btrfs.py index cab8beb9..9f938c75 100644 --- a/borgmatic/hooks/data_source/btrfs.py +++ b/borgmatic/hooks/data_source/btrfs.py @@ -412,6 +412,10 @@ def make_data_source_dump_patterns( config, borgmatic_runtime_directory, name=None, + hostname=None, + port=None, + container=None, + label=None, ): # pragma: no cover ''' Restores aren't implemented, because stored files can be extracted directly with "extract". diff --git a/borgmatic/hooks/data_source/dump.py b/borgmatic/hooks/data_source/dump.py index a585548b..8eaae4f2 100644 --- a/borgmatic/hooks/data_source/dump.py +++ b/borgmatic/hooks/data_source/dump.py @@ -124,6 +124,6 @@ def convert_glob_patterns_to_borg_pattern(patterns): # longer than the pattern. E.g., a pattern of "borgmatic/*/foo_databases/test" should also match # paths like "borgmatic/*/foo_databases/test/toc.dat" return 're:' + '|'.join( - fnmatch.translate(pattern.lstrip('/')).replace('\\z', '').replace('\\Z', '') + fnmatch.translate(pattern.lstrip('/')).replace('\\z', '').replace('\\Z', '') + '$' for pattern in patterns ) diff --git a/borgmatic/hooks/data_source/lvm.py b/borgmatic/hooks/data_source/lvm.py index 3cdc2cdc..a51834af 100644 --- a/borgmatic/hooks/data_source/lvm.py +++ b/borgmatic/hooks/data_source/lvm.py @@ -452,6 +452,10 @@ def make_data_source_dump_patterns( config, borgmatic_runtime_directory, name=None, + hostname=None, + port=None, + container=None, + label=None, ): # pragma: no cover ''' Restores aren't implemented, because stored files can be extracted directly with "extract". diff --git a/borgmatic/hooks/data_source/mariadb.py b/borgmatic/hooks/data_source/mariadb.py index 625acb6d..190a0d7c 100644 --- a/borgmatic/hooks/data_source/mariadb.py +++ b/borgmatic/hooks/data_source/mariadb.py @@ -410,6 +410,10 @@ def make_data_source_dump_patterns( config, borgmatic_runtime_directory, name=None, + hostname=None, + port=None, + container=None, + label=None, ): # pragma: no cover ''' Given a sequence of configurations dicts, a configuration dict, the borgmatic runtime directory, @@ -419,16 +423,24 @@ def make_data_source_dump_patterns( borgmatic_source_directory = borgmatic.config.paths.get_borgmatic_source_directory(config) return ( - dump.make_data_source_dump_filename(make_dump_path('borgmatic'), name, label='*'), + dump.make_data_source_dump_filename( + make_dump_path('borgmatic'), name, hostname, port, container, label + ), dump.make_data_source_dump_filename( make_dump_path(borgmatic_runtime_directory), name, - label='*', + hostname, + port, + container, + label, ), dump.make_data_source_dump_filename( make_dump_path(borgmatic_source_directory), name, - label='*', + hostname, + port, + container, + label, ), ) diff --git a/borgmatic/hooks/data_source/mongodb.py b/borgmatic/hooks/data_source/mongodb.py index c4dceed5..63311ac4 100644 --- a/borgmatic/hooks/data_source/mongodb.py +++ b/borgmatic/hooks/data_source/mongodb.py @@ -212,6 +212,10 @@ def make_data_source_dump_patterns( config, borgmatic_runtime_directory, name=None, + hostname=None, + port=None, + container=None, + label=None, ): # pragma: no cover ''' Given a sequence of configurations dicts, a configuration dict, the borgmatic runtime directory, @@ -221,16 +225,24 @@ def make_data_source_dump_patterns( borgmatic_source_directory = borgmatic.config.paths.get_borgmatic_source_directory(config) return ( - dump.make_data_source_dump_filename(make_dump_path('borgmatic'), name, label='*'), + dump.make_data_source_dump_filename( + make_dump_path('borgmatic'), name, hostname, port, container, label + ), dump.make_data_source_dump_filename( make_dump_path(borgmatic_runtime_directory), name, - label='*', + hostname, + port, + container, + label, ), dump.make_data_source_dump_filename( make_dump_path(borgmatic_source_directory), name, - label='*', + hostname, + port, + container, + label, ), ) diff --git a/borgmatic/hooks/data_source/mysql.py b/borgmatic/hooks/data_source/mysql.py index a7932495..ddabd41b 100644 --- a/borgmatic/hooks/data_source/mysql.py +++ b/borgmatic/hooks/data_source/mysql.py @@ -341,6 +341,10 @@ def make_data_source_dump_patterns( config, borgmatic_runtime_directory, name=None, + hostname=None, + port=None, + container=None, + label=None, ): # pragma: no cover ''' Given a sequence of configurations dicts, a configuration dict, the borgmatic runtime directory, @@ -350,16 +354,24 @@ def make_data_source_dump_patterns( borgmatic_source_directory = borgmatic.config.paths.get_borgmatic_source_directory(config) return ( - dump.make_data_source_dump_filename(make_dump_path('borgmatic'), name, label='*'), + dump.make_data_source_dump_filename( + make_dump_path('borgmatic'), name, name, hostname, port, container, label + ), dump.make_data_source_dump_filename( make_dump_path(borgmatic_runtime_directory), name, - label='*', + hostname, + port, + container, + label, ), dump.make_data_source_dump_filename( make_dump_path(borgmatic_source_directory), name, - label='*', + hostname, + port, + container, + label, ), ) diff --git a/borgmatic/hooks/data_source/postgresql.py b/borgmatic/hooks/data_source/postgresql.py index 6c746417..af2cc987 100644 --- a/borgmatic/hooks/data_source/postgresql.py +++ b/borgmatic/hooks/data_source/postgresql.py @@ -302,6 +302,10 @@ def make_data_source_dump_patterns( config, borgmatic_runtime_directory, name=None, + hostname=None, + port=None, + container=None, + label=None, ): # pragma: no cover ''' Given a sequence of configurations dicts, a configuration dict, the borgmatic runtime directory, @@ -311,16 +315,24 @@ def make_data_source_dump_patterns( borgmatic_source_directory = borgmatic.config.paths.get_borgmatic_source_directory(config) return ( - dump.make_data_source_dump_filename(make_dump_path('borgmatic'), name, label='*'), + dump.make_data_source_dump_filename( + make_dump_path('borgmatic'), name, hostname, port, container, label + ), dump.make_data_source_dump_filename( make_dump_path(borgmatic_runtime_directory), name, - label='*', + hostname, + port, + container, + label, ), dump.make_data_source_dump_filename( make_dump_path(borgmatic_source_directory), name, - label='*', + hostname, + port, + container, + label, ), ) diff --git a/borgmatic/hooks/data_source/sqlite.py b/borgmatic/hooks/data_source/sqlite.py index a31a8abf..43c9bd07 100644 --- a/borgmatic/hooks/data_source/sqlite.py +++ b/borgmatic/hooks/data_source/sqlite.py @@ -144,6 +144,10 @@ def make_data_source_dump_patterns( config, borgmatic_runtime_directory, name=None, + hostname=None, + port=None, + container=None, + label=None, ): # pragma: no cover ''' Given a sequence of configurations dicts, a configuration dict, the borgmatic runtime directory, @@ -153,16 +157,24 @@ def make_data_source_dump_patterns( borgmatic_source_directory = borgmatic.config.paths.get_borgmatic_source_directory(config) return ( - dump.make_data_source_dump_filename(make_dump_path('borgmatic'), name, label='*'), + dump.make_data_source_dump_filename( + make_dump_path('borgmatic'), name, hostname, port, container, label + ), dump.make_data_source_dump_filename( make_dump_path(borgmatic_runtime_directory), name, - label='*', + hostname, + port, + container, + label, ), dump.make_data_source_dump_filename( make_dump_path(borgmatic_source_directory), name, - label='*', + hostname, + port, + container, + label, ), ) diff --git a/borgmatic/hooks/data_source/zfs.py b/borgmatic/hooks/data_source/zfs.py index 77b3ac9d..7b285849 100644 --- a/borgmatic/hooks/data_source/zfs.py +++ b/borgmatic/hooks/data_source/zfs.py @@ -453,6 +453,10 @@ def make_data_source_dump_patterns( config, borgmatic_runtime_directory, name=None, + hostname=None, + port=None, + container=None, + label=None, ): # pragma: no cover ''' Restores aren't implemented, because stored files can be extracted directly with "extract". diff --git a/pyproject.toml b/pyproject.toml index 8a7eb600..2a0a0ff9 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,6 +1,6 @@ [project] name = "borgmatic" -version = "2.0.13" +version = "2.0.14.dev0" authors = [ { name="Dan Helfman", email="witten@torsion.org" }, ] diff --git a/tests/unit/actions/test_restore.py b/tests/unit/actions/test_restore.py index d167b254..00fc474d 100644 --- a/tests/unit/actions/test_restore.py +++ b/tests/unit/actions/test_restore.py @@ -315,6 +315,10 @@ def test_restore_single_dump_extracts_and_restores_single_file_dump(): object, object, object, + object, + object, + object, + object, ).and_return({'postgresql': flexmock()}) flexmock(module.tempfile).should_receive('mkdtemp').never() flexmock(module.borgmatic.hooks.data_source.dump).should_receive( @@ -359,6 +363,10 @@ def test_restore_single_dump_extracts_and_restores_directory_dump(): object, object, object, + object, + object, + object, + object, ).and_return({'postgresql': flexmock()}) flexmock(module.tempfile).should_receive('mkdtemp').once().and_return( '/run/user/0/borgmatic/tmp1234', @@ -405,6 +413,10 @@ def test_restore_single_dump_with_directory_dump_error_cleans_up_temporary_direc object, object, object, + object, + object, + object, + object, ).and_return({'postgresql': flexmock()}) flexmock(module.tempfile).should_receive('mkdtemp').once().and_return( '/run/user/0/borgmatic/tmp1234', @@ -452,6 +464,10 @@ def test_restore_single_dump_with_directory_dump_and_dry_run_skips_directory_mov object, object, object, + object, + object, + object, + object, ).and_return({'postgresql': flexmock()}) flexmock(module.tempfile).should_receive('mkdtemp').once().and_return('/run/borgmatic/tmp1234') flexmock(module.borgmatic.hooks.data_source.dump).should_receive( diff --git a/tests/unit/hooks/data_source/test_dump.py b/tests/unit/hooks/data_source/test_dump.py index 2ea45b63..0fb30506 100644 --- a/tests/unit/hooks/data_source/test_dump.py +++ b/tests/unit/hooks/data_source/test_dump.py @@ -158,5 +158,5 @@ def test_remove_data_source_dumps_without_dump_path_present_skips_removal(): def test_convert_glob_patterns_to_borg_pattern_makes_multipart_regular_expression(): assert ( module.convert_glob_patterns_to_borg_pattern(('/etc/foo/bar', '/bar/*/baz')) - == 're:(?s:etc/foo/bar)|(?s:bar/.*/baz)' + == 're:(?s:etc/foo/bar)$|(?s:bar/.*/baz)$' ) From ea25d969cb733929616aedb4364f5a39a40c1086 Mon Sep 17 00:00:00 2001 From: Dan Helfman Date: Mon, 22 Dec 2025 16:02:53 -0800 Subject: [PATCH 2/4] Fix typo in MySQL hook (#1208). --- borgmatic/hooks/data_source/mysql.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/borgmatic/hooks/data_source/mysql.py b/borgmatic/hooks/data_source/mysql.py index ddabd41b..ab412312 100644 --- a/borgmatic/hooks/data_source/mysql.py +++ b/borgmatic/hooks/data_source/mysql.py @@ -355,7 +355,7 @@ def make_data_source_dump_patterns( return ( dump.make_data_source_dump_filename( - make_dump_path('borgmatic'), name, name, hostname, port, container, label + make_dump_path('borgmatic'), name, hostname, port, container, label ), dump.make_data_source_dump_filename( make_dump_path(borgmatic_runtime_directory), From 00e22846623268ac8f6277d9e06e7ef430e5e81f Mon Sep 17 00:00:00 2001 From: Dan Helfman Date: Wed, 24 Dec 2025 12:56:52 -0800 Subject: [PATCH 3/4] When creating patterns for a "borg extract" during a restore, match directory format dumps too. Also make dump restore ordering to be deterministic (#1208). --- NEWS | 4 +- borgmatic/actions/restore.py | 66 ++++++++++++++--------------- borgmatic/hooks/data_source/dump.py | 28 +++++++++--- 3 files changed, 57 insertions(+), 41 deletions(-) diff --git a/NEWS b/NEWS index 994c97b4..7262e1bd 100644 --- a/NEWS +++ b/NEWS @@ -1,6 +1,6 @@ 2.0.14.dev0 - * #1208: Fix for the database "restore" action restoring more databases than the "--database" flag - specifies. + * #1208: Fix for the "restore" action incorrectly extracting more database dumps than the + "--database" flag specifies. * #1212: Fix an error when restoring multiple directory-format database dumps at once. 2.0.13 diff --git a/borgmatic/actions/restore.py b/borgmatic/actions/restore.py index 92d4612a..086a7531 100644 --- a/borgmatic/actions/restore.py +++ b/borgmatic/actions/restore.py @@ -273,7 +273,7 @@ def collect_dumps_from_archive( borgmatic runtime directory, query the archive for the names of data sources dumps it contains and return them as a set of Dump instances. ''' - dumps_from_archive = set() + dumps_from_archive = {} # Use a dict as an ordered set. # There is (at most) one dump metadata file per data source hook. Load each. for dumps_metadata_path in borgmatic.borg.list.capture_archive_listing( @@ -302,33 +302,30 @@ def collect_dumps_from_archive( if not dumps_metadata_path: continue - dumps_from_archive.update( - set( - borgmatic.hooks.data_source.dump.parse_data_source_dumps_metadata( - borgmatic.borg.extract.extract_archive( - global_arguments.dry_run, - repository, - archive, - [dumps_metadata_path], - config, - local_borg_version, - global_arguments, - local_path=local_path, - remote_path=remote_path, - extract_to_stdout=True, - ) - .stdout.read() - .decode(), - dumps_metadata_path, - ) + for dump in borgmatic.hooks.data_source.dump.parse_data_source_dumps_metadata( + borgmatic.borg.extract.extract_archive( + global_arguments.dry_run, + repository, + archive, + [dumps_metadata_path], + config, + local_borg_version, + global_arguments, + local_path=local_path, + remote_path=remote_path, + extract_to_stdout=True, ) - ) + .stdout.read() + .decode(), + dumps_metadata_path, + ): + dumps_from_archive[dump] = None # If we've successfully loaded any dumps metadata, we're done. if dumps_from_archive: logger.debug('Collecting database dumps from archive data source dumps metadata files') - return dumps_from_archive + return tuple(dumps_from_archive.keys()) # No dumps metadata files were found, so for backwards compatibility, fall back to parsing the # paths of dumps found in the archive to get their respective dump metadata. @@ -392,11 +389,11 @@ def collect_dumps_from_archive( except (ValueError, TypeError): port = None - dumps_from_archive.add( + dumps_from_archive[ Dump( hook_name, data_source_name, None if hostname == 'localhost' else hostname, port ) - ) + ] = None # We've successfully parsed the dump path, so need to probe any further. break @@ -405,13 +402,13 @@ def collect_dumps_from_archive( f'Ignoring invalid data source dump path "{dump_path}" in archive {archive}', ) - return dumps_from_archive + return tuple(dumps_from_archive.keys()) def get_dumps_to_restore(restore_arguments, dumps_from_archive): ''' Given restore arguments as an argparse.Namespace instance indicating which dumps to restore and - a set of Dump instances representing the dumps found in an archive, return a set of specific + a tuple of Dump instances representing the dumps found in an archive, return a tuple of specific Dump instances from the archive to restore. As part of this, replace any Dump having a data source name of "all" with multiple named Dump instances as appropriate. @@ -456,11 +453,12 @@ def get_dumps_to_restore(restore_arguments, dumps_from_archive): } ) missing_dumps = set() - dumps_to_restore = set() + dumps_to_restore = {} # Use a dict as an ordered set. # If there's a requested "all" dump, add every dump from the archive to the dumps to restore. if any(dump for dump in requested_dumps if dump.data_source_name == 'all'): - dumps_to_restore.update(dumps_from_archive) + for dump in dumps_from_archive: + dumps_to_restore[dump] = None # If any archive dump matches a requested dump, add the archive dump to the dumps to restore. for requested_dump in requested_dumps: @@ -476,7 +474,7 @@ def get_dumps_to_restore(restore_arguments, dumps_from_archive): if len(matching_dumps) == 0: missing_dumps.add(requested_dump) elif len(matching_dumps) == 1: - dumps_to_restore.add(matching_dumps[0]) + dumps_to_restore[matching_dumps[0]] = None else: raise ValueError( f'Cannot restore data source {render_dump_metadata(requested_dump)} because there are multiple matching dumps in the archive. Try adding flags to disambiguate.', @@ -491,20 +489,20 @@ def get_dumps_to_restore(restore_arguments, dumps_from_archive): f"Cannot restore data source dump{'s' if len(missing_dumps) > 1 else ''} {rendered_dumps} missing from archive", ) - return dumps_to_restore + return tuple(dumps_to_restore.keys()) def ensure_requested_dumps_restored(dumps_to_restore, dumps_actually_restored): ''' - Given a set of requested dumps to restore and a set of dumps actually restored, raise ValueError - if any requested dumps to restore weren't restored, indicating that they were missing from the - configuration. + Given a tuple of requested dumps to restore and a set of dumps actually restored, raise + ValueError if any requested dumps to restore weren't restored, indicating that they were missing + from the configuration. ''' if not dumps_actually_restored: raise ValueError('No data source dumps were found to restore') missing_dumps = sorted( - dumps_to_restore - dumps_actually_restored, + set(dumps_to_restore) - dumps_actually_restored, key=lambda dump: dump.data_source_name, ) diff --git a/borgmatic/hooks/data_source/dump.py b/borgmatic/hooks/data_source/dump.py index 8eaae4f2..e26ca5a7 100644 --- a/borgmatic/hooks/data_source/dump.py +++ b/borgmatic/hooks/data_source/dump.py @@ -2,6 +2,7 @@ import fnmatch import json import logging import os +import re import shutil import borgmatic.actions.restore @@ -114,16 +115,33 @@ def remove_data_source_dumps(dump_path, data_source_type_name, dry_run): shutil.rmtree(dump_path) +END_OF_STRING_PATTTERN = re.compile(r'\\z', flags=re.IGNORECASE) + + def convert_glob_patterns_to_borg_pattern(patterns): ''' Convert a sequence of shell glob patterns like "/etc/*", "/tmp/*" to the corresponding Borg regular expression archive pattern as a single string like "re:etc/.*|tmp/.*". ''' - # Remove the "\z" or "\Z" generated by fnmatch.translate() because we don't want the pattern to - # match only at the end of a path, as directory format dumps require extracting files with paths - # longer than the pattern. E.g., a pattern of "borgmatic/*/foo_databases/test" should also match - # paths like "borgmatic/*/foo_databases/test/toc.dat" + # This deserves some explanation. Here's what this code is doing for each shell glob pattern: + # + # * Strip off the leading slash, since Borg doesn't store leading slashes in archives. + # * Use fnmatch.translate() to convert the pattern to a Borg pattern. + # * Remove the "\z" or \"Z" end-of-string special character generated by fnmatch.translate(), + # because Borg doesn't like it. Replace it with a "$" end-of-string special character + # instead. And yes, this is using a regular expression to modify a regular expression. + # * Do the above for each of: + # * The plain pattern. This supports the use case of a standard database where its name is a + # filename in the path. Example: borgmatic/foo_databases/hostname/test + # * The pattern with "/*" tacked onto the end of it. This supports the use case of a + # directory-format database where its name is a directory name in the path. Example: + # borgmatic/foo_databases/hostname/test/toc.dat + # + # Join the resulting transformed patterns together with "|" and return them as a string. return 're:' + '|'.join( - fnmatch.translate(pattern.lstrip('/')).replace('\\z', '').replace('\\Z', '') + '$' + re.sub(END_OF_STRING_PATTTERN, '$', fnmatch.translate(stripped)) + + '|' + + re.sub(END_OF_STRING_PATTTERN, '$', fnmatch.translate(stripped + '/*')) for pattern in patterns + for stripped in (pattern.lstrip('/'),) ) From 047a374514603c4f0c6edf4c737387d11f4cf5ab Mon Sep 17 00:00:00 2001 From: Dan Helfman Date: Thu, 25 Dec 2025 13:56:01 -0800 Subject: [PATCH 4/4] Fix existing automated tests (#1208). --- borgmatic/actions/restore.py | 13 ++-- tests/unit/actions/test_restore.py | 84 ++++++++++------------- tests/unit/hooks/data_source/test_dump.py | 4 +- 3 files changed, 45 insertions(+), 56 deletions(-) diff --git a/borgmatic/actions/restore.py b/borgmatic/actions/restore.py index 086a7531..856ae766 100644 --- a/borgmatic/actions/restore.py +++ b/borgmatic/actions/restore.py @@ -271,7 +271,7 @@ def collect_dumps_from_archive( Given a local or remote repository path, a resolved archive name, a configuration dict, the local Borg version, global arguments an argparse.Namespace, local and remote Borg paths, and the borgmatic runtime directory, query the archive for the names of data sources dumps it contains - and return them as a set of Dump instances. + and return them as a tuple of Dump instances. ''' dumps_from_archive = {} # Use a dict as an ordered set. @@ -415,7 +415,8 @@ def get_dumps_to_restore(restore_arguments, dumps_from_archive): Raise ValueError if any of the requested data source names cannot be found in the archive or if there are multiple archive dump matches for a given requested dump. ''' - requested_dumps = ( + requested_dumps = tuple( + # Use a dict comprehension as an ordered set. { Dump( hook_name=( @@ -432,16 +433,16 @@ def get_dumps_to_restore(restore_arguments, dumps_from_archive): port=restore_arguments.original_port, label=restore_arguments.original_label or UNSPECIFIED, container=restore_arguments.original_container or UNSPECIFIED, - ) + ): None for name in restore_arguments.data_sources or (UNSPECIFIED,) - } + }.keys() if restore_arguments.hook or restore_arguments.data_sources or restore_arguments.original_hostname or restore_arguments.original_port or restore_arguments.original_label or restore_arguments.original_container - else { + else ( Dump( hook_name=UNSPECIFIED, data_source_name='all', @@ -450,7 +451,7 @@ def get_dumps_to_restore(restore_arguments, dumps_from_archive): label=UNSPECIFIED, container=UNSPECIFIED, ), - } + ) ) missing_dumps = set() dumps_to_restore = {} # Use a dict as an ordered set. diff --git a/tests/unit/actions/test_restore.py b/tests/unit/actions/test_restore.py index d15e9b87..93f91797 100644 --- a/tests/unit/actions/test_restore.py +++ b/tests/unit/actions/test_restore.py @@ -542,7 +542,7 @@ def test_collect_dumps_from_archive_with_dumps_metadata_parses_it(): borgmatic_runtime_directory='/run/borgmatic', ) - assert archive_dumps == set(dumps_metadata) + assert archive_dumps == tuple(dumps_metadata) def test_collect_dumps_from_archive_with_empty_dumps_metadata_path_falls_back_to_parsing_archive_paths(): @@ -580,11 +580,11 @@ def test_collect_dumps_from_archive_with_empty_dumps_metadata_path_falls_back_to borgmatic_runtime_directory='/run/borgmatic', ) - assert archive_dumps == { + assert archive_dumps == ( module.Dump('postgresql_databases', 'foo'), module.Dump('postgresql_databases', 'bar', 'host', 1234), module.Dump('mysql_databases', 'quux'), - } + ) def test_collect_dumps_from_archive_without_dumps_metadata_falls_back_to_parsing_archive_paths(): @@ -622,11 +622,11 @@ def test_collect_dumps_from_archive_without_dumps_metadata_falls_back_to_parsing borgmatic_runtime_directory='/run/borgmatic', ) - assert archive_dumps == { + assert archive_dumps == ( module.Dump('postgresql_databases', 'foo'), module.Dump('postgresql_databases', 'bar', 'host', 1234), module.Dump('mysql_databases', 'quux'), - } + ) def test_collect_dumps_from_archive_parses_archive_paths_with_different_base_directories(): @@ -665,12 +665,12 @@ def test_collect_dumps_from_archive_parses_archive_paths_with_different_base_dir borgmatic_runtime_directory='/run/borgmatic', ) - assert archive_dumps == { + assert archive_dumps == ( module.Dump('postgresql_databases', 'foo'), module.Dump('postgresql_databases', 'bar'), module.Dump('postgresql_databases', 'baz'), module.Dump('mysql_databases', 'quux'), - } + ) def test_collect_dumps_from_archive_parses_directory_format_archive_paths(): @@ -707,9 +707,7 @@ def test_collect_dumps_from_archive_parses_directory_format_archive_paths(): borgmatic_runtime_directory='/run/borgmatic', ) - assert archive_dumps == { - module.Dump('postgresql_databases', 'foo'), - } + assert archive_dumps == (module.Dump('postgresql_databases', 'foo'),) def test_collect_dumps_from_archive_skips_bad_archive_paths_or_bad_path_components(): @@ -749,18 +747,18 @@ def test_collect_dumps_from_archive_skips_bad_archive_paths_or_bad_path_componen borgmatic_runtime_directory='/run/borgmatic', ) - assert archive_dumps == { + assert archive_dumps == ( module.Dump('postgresql_databases', 'foo'), module.Dump('postgresql_databases', 'bar'), - } + ) def test_get_dumps_to_restore_gets_requested_dumps_found_in_archive(): - dumps_from_archive = { + dumps_from_archive = ( module.Dump('postgresql_databases', 'foo'), module.Dump('postgresql_databases', 'bar'), module.Dump('postgresql_databases', 'baz'), - } + ) flexmock(module).should_receive('dumps_match').and_return(False) flexmock(module).should_receive('dumps_match').with_args( module.Dump( @@ -793,16 +791,14 @@ def test_get_dumps_to_restore_gets_requested_dumps_found_in_archive(): original_container=None, ), dumps_from_archive=dumps_from_archive, - ) == { + ) == ( module.Dump('postgresql_databases', 'foo'), module.Dump('postgresql_databases', 'bar'), - } + ) def test_get_dumps_to_restore_raises_for_requested_dumps_missing_from_archive(): - dumps_from_archive = { - module.Dump('postgresql_databases', 'foo'), - } + dumps_from_archive = (module.Dump('postgresql_databases', 'foo'),) flexmock(module).should_receive('dumps_match').and_return(False) flexmock(module).should_receive('dumps_match').with_args( module.Dump( @@ -832,10 +828,10 @@ def test_get_dumps_to_restore_raises_for_requested_dumps_missing_from_archive(): def test_get_dumps_to_restore_without_requested_dumps_finds_all_archive_dumps(): - dumps_from_archive = { + dumps_from_archive = ( module.Dump('postgresql_databases', 'foo'), module.Dump('postgresql_databases', 'bar'), - } + ) flexmock(module).should_receive('dumps_match').and_return(False) assert ( @@ -855,10 +851,10 @@ def test_get_dumps_to_restore_without_requested_dumps_finds_all_archive_dumps(): def test_get_dumps_to_restore_with_all_in_requested_dumps_finds_all_archive_dumps(): - dumps_from_archive = { + dumps_from_archive = ( module.Dump('postgresql_databases', 'foo'), module.Dump('postgresql_databases', 'bar'), - } + ) flexmock(module).should_receive('dumps_match').and_return(False) flexmock(module).should_receive('dumps_match').with_args( module.Dump( @@ -898,10 +894,10 @@ def test_get_dumps_to_restore_with_all_in_requested_dumps_finds_all_archive_dump def test_get_dumps_to_restore_with_all_in_requested_dumps_plus_additional_requested_dumps_omits_duplicates(): - dumps_from_archive = { + dumps_from_archive = ( module.Dump('postgresql_databases', 'foo'), module.Dump('postgresql_databases', 'bar'), - } + ) flexmock(module).should_receive('dumps_match').and_return(False) flexmock(module).should_receive('dumps_match').with_args( module.Dump( @@ -974,10 +970,10 @@ def test_get_dumps_to_restore_raises_for_multiple_matching_dumps_in_archive(): original_label=None, original_container=None, ), - dumps_from_archive={ + dumps_from_archive=( module.Dump('postgresql_databases', 'foo'), module.Dump('mariadb_databases', 'foo'), - }, + ), ) assert 'Try adding flags to disambiguate.' in str(exc_info.value) @@ -1006,17 +1002,17 @@ def test_get_dumps_to_restore_raises_for_all_in_requested_dumps_and_requested_du original_label=None, original_container=None, ), - dumps_from_archive={module.Dump('postgresql_databases', 'foo')}, + dumps_from_archive=(module.Dump('postgresql_databases', 'foo'),), ) assert 'dump test missing from archive' in str(exc_info.value) def test_get_dumps_to_restore_with_requested_hook_name_filters_dumps_found_in_archive(): - dumps_from_archive = { + dumps_from_archive = ( module.Dump('mariadb_databases', 'foo'), module.Dump('postgresql_databases', 'foo'), module.Dump('sqlite_databases', 'bar'), - } + ) flexmock(module).should_receive('dumps_match').and_return(False) flexmock(module).should_receive('dumps_match').with_args( module.Dump( @@ -1039,17 +1035,15 @@ def test_get_dumps_to_restore_with_requested_hook_name_filters_dumps_found_in_ar original_container=None, ), dumps_from_archive=dumps_from_archive, - ) == { - module.Dump('postgresql_databases', 'foo'), - } + ) == (module.Dump('postgresql_databases', 'foo'),) def test_get_dumps_to_restore_with_requested_shortened_hook_name_filters_dumps_found_in_archive(): - dumps_from_archive = { + dumps_from_archive = ( module.Dump('mariadb_databases', 'foo'), module.Dump('postgresql_databases', 'foo'), module.Dump('sqlite_databases', 'bar'), - } + ) flexmock(module).should_receive('dumps_match').and_return(False) flexmock(module).should_receive('dumps_match').with_args( module.Dump( @@ -1072,17 +1066,15 @@ def test_get_dumps_to_restore_with_requested_shortened_hook_name_filters_dumps_f original_container=None, ), dumps_from_archive=dumps_from_archive, - ) == { - module.Dump('postgresql_databases', 'foo'), - } + ) == (module.Dump('postgresql_databases', 'foo'),) def test_get_dumps_to_restore_with_requested_hostname_filters_dumps_found_in_archive(): - dumps_from_archive = { + dumps_from_archive = ( module.Dump('postgresql_databases', 'foo'), module.Dump('postgresql_databases', 'foo', 'host'), module.Dump('postgresql_databases', 'bar'), - } + ) flexmock(module).should_receive('dumps_match').and_return(False) flexmock(module).should_receive('dumps_match').with_args( module.Dump( @@ -1105,17 +1097,15 @@ def test_get_dumps_to_restore_with_requested_hostname_filters_dumps_found_in_arc original_container=None, ), dumps_from_archive=dumps_from_archive, - ) == { - module.Dump('postgresql_databases', 'foo', 'host'), - } + ) == (module.Dump('postgresql_databases', 'foo', 'host'),) def test_get_dumps_to_restore_with_requested_port_filters_dumps_found_in_archive(): - dumps_from_archive = { + dumps_from_archive = ( module.Dump('postgresql_databases', 'foo', 'host'), module.Dump('postgresql_databases', 'foo', 'host', 1234), module.Dump('postgresql_databases', 'bar'), - } + ) flexmock(module).should_receive('dumps_match').and_return(False) flexmock(module).should_receive('dumps_match').with_args( module.Dump( @@ -1139,9 +1129,7 @@ def test_get_dumps_to_restore_with_requested_port_filters_dumps_found_in_archive original_container=None, ), dumps_from_archive=dumps_from_archive, - ) == { - module.Dump('postgresql_databases', 'foo', 'host', 1234), - } + ) == (module.Dump('postgresql_databases', 'foo', 'host', 1234),) def test_ensure_requested_dumps_restored_with_all_dumps_restored_does_not_raise(): diff --git a/tests/unit/hooks/data_source/test_dump.py b/tests/unit/hooks/data_source/test_dump.py index 0fb30506..a00606e3 100644 --- a/tests/unit/hooks/data_source/test_dump.py +++ b/tests/unit/hooks/data_source/test_dump.py @@ -157,6 +157,6 @@ def test_remove_data_source_dumps_without_dump_path_present_skips_removal(): def test_convert_glob_patterns_to_borg_pattern_makes_multipart_regular_expression(): assert ( - module.convert_glob_patterns_to_borg_pattern(('/etc/foo/bar', '/bar/*/baz')) - == 're:(?s:etc/foo/bar)$|(?s:bar/.*/baz)$' + module.convert_glob_patterns_to_borg_pattern(('/etc/foo/bar', '/bar/baz/quux')) + == 're:(?s:etc/foo/bar)$|(?s:etc/foo/bar/.*)$|(?s:bar/baz/quux)$|(?s:bar/baz/quux/.*)$' )