def upload_tmp_txt_file(content=''):
conf.pub_key = 'demopublickey'
tmp_txt_file = NamedTemporaryFile(mode='wb', delete=False)
tmp_txt_file.write(content.encode('utf-8'))
tmp_txt_file.close()
with open(tmp_txt_file.name, 'rb') as fh:
file_ = File.upload(fh, store=False)
return file_
def test_successful_upload_when_file_is_opened_in_binary_mode(self):
with open(self.tmp_txt_file.name, 'rb') as fh:
file_ = File.upload(fh)
self.assertIsInstance(file_, File)
def upload(arg_namespace):
if not _check_upload_args(arg_namespace):
return
with open(arg_namespace.filename, 'rb') as fh:
file_ = File.upload(fh)
_handle_uploaded_file(file_, arg_namespace)
def test_successful_upload_when_file_is_opened_in_binary_mode(self):
with open(self.tmp_txt_file.name, 'rb') as fh:
file_ = File.upload(fh)
self.assertIsInstance(file_, File)
def upload(arg_namespace):
if not _check_upload_args(arg_namespace):
return
with open(arg_namespace.filename, 'rb') as fh:
file_ = File.upload(fh)
_handle_uploaded_file(file_, arg_namespace)
