def test_create_restore_session_with_device_guid_calls_post_with_device_guid_in_json(
self, connection):
storage_archive_service = StorageArchiveService(connection)
storage_archive_service.create_restore_session(DEVICE_GUID)
json_arg = connection.post.call_args[KWARGS_INDEX][JSON_KEYWORD]
assert json_arg.get(COMPUTER_GUID_KEY) == DEVICE_GUID
def test_create_restore_session_with_encryption_key_calls_post_with_encryption_key_in_json(
self, connection):
storage_archive_service = StorageArchiveService(connection)
storage_archive_service.create_restore_session(
DEVICE_GUID, encryption_key=ENCRYPTION_KEY)
json_arg = connection.post.call_args[KWARGS_INDEX][JSON_KEYWORD]
assert json_arg.get(ENCRYPTION_KEY_KEY) == ENCRYPTION_KEY
def test_create_restore_session_with_private_password_calls_post_with_private_password_in_json(
self, connection):
storage_archive_service = StorageArchiveService(connection)
storage_archive_service.create_restore_session(
DEVICE_GUID, private_password=PRIVATE_PASSWORD)
json_arg = connection.post.call_args[KWARGS_INDEX][JSON_KEYWORD]
assert json_arg.get(PRIVATE_PASSWORD_KEY) == PRIVATE_PASSWORD
def test_create_restore_session_with_data_key_token_calls_post_with_data_key_token_in_json(
self, connection):
storage_archive_service = StorageArchiveService(connection)
storage_archive_service.create_restore_session(
DEVICE_GUID, data_key_token=DATA_KEY_TOKEN)
json_arg = connection.post.call_args[KWARGS_INDEX][JSON_KEYWORD]
assert json_arg.get(DATA_KEY_TOKEN_KEY) == DATA_KEY_TOKEN
def test_create_restore_session_calls_post_with_correct_url(
self, mocker, connection
):
storage_archive_service = StorageArchiveService(connection)
storage_archive_service.create_restore_session(TEST_DEVICE_GUID)
connection.post.assert_called_once_with(
WEB_RESTORE_SESSION_URL, json=mocker.ANY
)
def test_create_restore_session_when_invalid_encryption_key_raises_expected_error(
self, mocker, connection):
def side_effect(*args, **kwargs):
base_err = HTTPError()
base_err.response = mocker.MagicMock(spec=Response)
base_err.response.text = """
[{"name":"CUSTOM_KEY_INVALID","description":"An error has
occurred. See server logs for more information.","objects":[]}]
"""
raise Py42InternalServerError(base_err)
connection.post.side_effect = side_effect
storage_archive_service = StorageArchiveService(connection)
with pytest.raises(Py42InvalidArchiveEncryptionKey) as err:
storage_archive_service.create_restore_session(
TEST_DEVICE_GUID, encryption_key=TEST_ENCRYPTION_KEY)
assert "Invalid archive encryption key." in str(err.value)
def test_create_restore_session_when_invalid_password_raises_expected_error(
self, mocker, connection):
def side_effect(*args, **kwargs):
base_err = HTTPError()
base_err.response = mocker.MagicMock(spec=Response)
base_err.response.text = """
[{"name":"PRIVATE_PASSWORD_INVALID","description":"An error has
occurred. See server logs for more information.","objects":[]}]
"""
raise Py42InternalServerError(base_err)
connection.post.side_effect = side_effect
storage_archive_service = StorageArchiveService(connection)
with pytest.raises(Py42InvalidArchivePassword) as err:
storage_archive_service.create_restore_session(
TEST_DEVICE_GUID, private_password=TEST_PASSWORD)
assert "Invalid archive password." in str(err.value)
