def test_create_restore_session_with_device_guid_calls_post_with_device_guid_in_json(
self, session):
storage_archive_client = StorageArchiveClient(session)
storage_archive_client.create_restore_session(DEVICE_GUID)
json_arg = session.post.call_args[KWARGS_INDEX][JSON_KEYWORD]
assert json_arg.get(COMPUTER_GUID_KEY) == DEVICE_GUID
def test_create_restore_session_with_data_key_token_calls_post_with_data_key_token_in_json(
self, session):
storage_archive_client = StorageArchiveClient(session)
storage_archive_client.create_restore_session(
DEVICE_GUID, data_key_token=DATA_KEY_TOKEN)
json_arg = session.post.call_args[KWARGS_INDEX][JSON_KEYWORD]
assert json_arg.get(DATA_KEY_TOKEN_KEY) == DATA_KEY_TOKEN
def test_create_restore_session_with_encryption_key_calls_post_with_encryption_key_in_json(
self, session):
storage_archive_client = StorageArchiveClient(session)
storage_archive_client.create_restore_session(
DEVICE_GUID, encryption_key=ENCRYPTION_KEY)
json_arg = session.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, session):
storage_archive_client = StorageArchiveClient(session)
storage_archive_client.create_restore_session(
DEVICE_GUID, private_password=PRIVATE_PASSWORD)
json_arg = session.post.call_args[KWARGS_INDEX][JSON_KEYWORD]
assert json_arg.get(PRIVATE_PASSWORD_KEY) == PRIVATE_PASSWORD
def test_create_restore_session_calls_post_with_correct_url(
self, mocker, session):
storage_archive_client = StorageArchiveClient(session)
storage_archive_client.create_restore_session(DEVICE_GUID)
session.post.assert_called_once_with(WEB_RESTORE_SESSION_URL,
json=mocker.ANY)
