def test_get_next_batch(self, from_file_mock, isfile_mock, glob_mock):
"""
Test the FileWalArchiver.get_next_batch method
"""
# WAL batch no errors
glob_mock.return_value = ["000000010000000000000001"]
isfile_mock.return_value = True
# This is an hack, instead of a WalFileInfo we use a simple string to
# ease all the comparisons. The resulting string is the name enclosed
# in colons. e.g. ":000000010000000000000001:"
from_file_mock.side_effect = lambda wal_name: ":%s:" % wal_name
backup_manager = build_backup_manager(name="TestServer")
archiver = FileWalArchiver(backup_manager)
backup_manager.server.archivers = [archiver]
batch = archiver.get_next_batch()
assert [":000000010000000000000001:"] == batch
# WAL batch with errors
wrong_file_name = "test_wrong_wal_file.2"
glob_mock.return_value = ["test_wrong_wal_file.2"]
batch = archiver.get_next_batch()
assert [wrong_file_name] == batch.errors
def test_get_next_batch(self, from_file_mock, isfile_mock, glob_mock):
"""
Test the FileWalArchiver.get_next_batch method
"""
# WAL batch no errors
glob_mock.return_value = ['000000010000000000000001']
isfile_mock.return_value = True
# This is an hack, instead of a WalFileInfo we use a simple string to
# ease all the comparisons. The resulting string is the name enclosed
# in colons. e.g. ":000000010000000000000001:"
from_file_mock.side_effect = lambda wal_name: ':%s:' % wal_name
backup_manager = build_backup_manager(
name='TestServer'
)
archiver = FileWalArchiver(backup_manager)
backup_manager.server.archivers = [archiver]
batch = archiver.get_next_batch()
assert [':000000010000000000000001:'] == batch
# WAL batch with errors
wrong_file_name = 'test_wrong_wal_file.2'
glob_mock.return_value = ['test_wrong_wal_file.2']
batch = archiver.get_next_batch()
assert [wrong_file_name] == batch.errors
