def setup_repositories(self):
self.repo_names = ['Snapshots', 'WAL Files']
self.repositories = {
self.repo_names[0]: get_Snapshot_repository(self.config),
self.repo_names[1]: get_WAL_repository(self.config),
}
self.num_snapshots = 0
self.num_walfiles = 0
self.size_snapshots = 0
self.size_walfiles = 0
def storagestats_main():
options, args = storagestats_handle_args()
conf = get_config_from_filename_and_set_up_logging(options.config_file)
repo_names = ['Snapshots', 'WAL Files']
repositories = {
repo_names[0]: get_Snapshot_repository(conf),
repo_names[1]: get_WAL_repository(conf),
}
reporter = Storage_stats_reporter(repo_names, repositories)
reporter.write_report(stdout)
def archivepgsql_main():
options, args = archivepgsql_handle_args()
conf = get_config_from_filename_and_set_up_logging(options.config_file)
data_dir = get_data_dir(conf)
with TemporaryDirectory(suffix='archivepgsql') as tempdir:
tag = generate_tag()
archive_dst_path = join(tempdir, 'pgsql.snapshot.tar')
repo = get_Snapshot_repository(conf)
if options.dry_run:
print("Dry Run")
return
else:
perform_backup(data_dir, archive_dst_path, tag, repo)
def test_perform_backup(self, mock_pg_start_backup, mock_pg_stop_backup,
mock_create_archive, mock_commit_snapshot_to_repository):
first_WAL = '000000D0'
second_WAL = '000000D1'
mock_pg_start_backup.return_value = first_WAL
mock_pg_stop_backup.return_value = second_WAL
archiveFile = os.path.join(self.archive_dir, 'pgsql.snapshot.tar')
tag = bbpgsql.archive_pgsql.generate_tag()
repo = get_Snapshot_repository(self.config)
bbpgsql.archive_pgsql.perform_backup(self.data_dir,
archiveFile, tag, repo)
mock_pg_start_backup.assert_called_once_with(tag)
mock_create_archive.assert_called_once_with(self.data_dir, archiveFile)
self.assertEqual(mock_pg_stop_backup.called, True)
self.assertEqual(mock_pg_stop_backup.call_count, 1)
mock_commit_snapshot_to_repository.assert_called_once_with(
repo, archiveFile, tag, first_WAL, second_WAL)
