def test_update_cold_storage_purge_date_calls_update_cold_storage_with_expected_data(
self, archive_accessor_manager, archive_client, archive_accessor
):
archive = ArchiveModule(archive_accessor_manager, archive_client)
archive.update_cold_storage_purge_date(u"123", u"2020-04-24")
archive_client.update_cold_storage_purge_date.assert_called_once_with(
u"123", u"2020-04-24"
)
def test_get_all_device_restore_history_calls_get_all_restore_history_with_expected_id(
self, archive_accessor_manager, archive_client, archive_accessor
):
archive = ArchiveModule(archive_accessor_manager, archive_client)
archive.get_all_device_restore_history(self._TEST_DAYS, self._TEST_ID)
archive_client.get_all_restore_history.assert_called_once_with(
self._TEST_DAYS, "computerId", self._TEST_ID
)
def test_get_backup_sets_calls_archive_client_get_backup_sets_with_expected_params(
self, archive_accessor_manager, archive_client, archive_accessor
):
archive = ArchiveModule(archive_accessor_manager, archive_client)
archive.get_backup_sets("device_guid", "dest_guid")
archive_client.get_backup_sets.assert_called_once_with(
"device_guid", "dest_guid"
)
def test_stream_from_backup_calls_get_archive_accessor_with_expected_params(
self, archive_accessor_manager, archive_client
):
archive = ArchiveModule(archive_accessor_manager, archive_client)
archive.stream_from_backup(
"path", "device_guid", "dest_guid", "password", "encryption_key"
)
archive._archive_accessor_manager.get_archive_accessor.assert_called_once_with(
"device_guid",
destination_guid="dest_guid",
private_password="******",
encryption_key="encryption_key",
)
def test_get_all_org_cold_storage_archives_calls_client_with_expected_data(
self, archive_accessor_manager, archive_client, archive_accessor
):
archive = ArchiveModule(archive_accessor_manager, archive_client)
archive.get_all_org_cold_storage_archives(
"TEST ORG ID", True, "sort_key", "sort_dir"
)
archive_client.get_all_org_cold_storage_archives.assert_called_once_with(
org_id="TEST ORG ID",
include_child_orgs=True,
sort_key="sort_key",
sort_dir="sort_dir",
)
def test_stream_from_backup_when_given_multiple_paths_calls_archive_accessor_stream_from_backup_with_expected_params(
self, archive_accessor_manager, archive_client, archive_accessor
):
archive_accessor_manager.get_archive_accessor.return_value = archive_accessor
archive = ArchiveModule(archive_accessor_manager, archive_client)
archive.stream_from_backup(
["path/to/first/file", "path/to/second/file"],
"device_guid",
"dest_guid",
"password",
"encryption_key",
)
archive_accessor.stream_from_backup.assert_called_once_with(
["path/to/first/file", "path/to/second/file"], file_size_calc_timeout=10
)
def _get_module(mocker):
client = mocker.MagicMock(spec=ArchiveClient)
accessor = mocker.MagicMock(spec=ArchiveAccessorManager)
return ArchiveModule(accessor, client)