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)