def test_cancel_error(self):
ex = UploadException('Testing one')
lib = CancelResponse.init_error(self._mock_shared_key(), ex)
assert self._mock_shared_key() == lib.get_result()['sharedKey']
assert CancelResponse.STATUS_FAIL == lib.get_result()['status']
assert 'Testing one' == lib.get_result()['errorMessage']
def test_done_fail(self):
data = self._mock_data()
ex = UploadException('Testing one')
lib = DoneResponse.init_error(self._mock_shared_key(), data, ex)
assert self._mock_shared_key() == lib.get_result()['sharedKey']
assert UploadResponse.STATUS_FAIL == lib.get_result()['status']
assert 'Testing one' == lib.get_result()['errorMessage']
def test_truncate_error(self):
ex = UploadException('Testing one')
lib = TruncateResponse.init_error(self._mock_shared_key(),
self._mock_data(), ex)
assert self._mock_shared_key() == lib.get_result()['sharedKey']
assert TruncateResponse.STATUS_FAIL == lib.get_result()['status']
assert 'Testing one' == lib.get_result()['errorMessage']
def test_init_error(self):
ex = UploadException('Testing one')
lib = InitResponse.init_error(self._mock_data(), ex)
assert '' == lib.get_result()['sharedKey']
assert 'abcdef' == lib.get_result()['name']
assert InitResponse.STATUS_FAIL == lib.get_result()['status']
assert 'Testing one' == lib.get_result()['errorMessage']
def _check_file_base(self):
if not len(self._file_base):
raise UploadException(self._lang.upload_name_is_empty())
def _check_target_dir(self):
if not len(self._target_dir):
raise UploadException(self._lang.target_dir_is_empty())
def _check_remote_name(self):
if not len(self._remote_file_name):
raise UploadException(self._lang.sent_name_is_empty())
def load(self, key: str) -> str:
content = self._data[key]
if not bool(len(content)):
raise UploadException(self._lang.drive_file_cannot_read())
return content