Skip to content
Projects
Groups
Snippets
Help
This project
Loading...
Sign in / Register
Toggle navigation
E
edx-platform
Overview
Overview
Details
Activity
Cycle Analytics
Repository
Repository
Files
Commits
Branches
Tags
Contributors
Graph
Compare
Charts
Issues
0
Issues
0
List
Board
Labels
Milestones
Merge Requests
0
Merge Requests
0
CI / CD
CI / CD
Pipelines
Jobs
Schedules
Charts
Wiki
Wiki
Snippets
Snippets
Members
Members
Collapse sidebar
Close sidebar
Activity
Graph
Charts
Create a new issue
Jobs
Commits
Issue Boards
Open sidebar
edx
edx-platform
Commits
07b036f3
Commit
07b036f3
authored
Dec 10, 2014
by
cahrens
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
Return the filename as returned from default storage's save.
It might be different from the requested filename.
parent
e4729378
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
32 additions
and
11 deletions
+32
-11
common/djangoapps/util/file.py
+2
-1
common/djangoapps/util/tests/test_file.py
+30
-10
No files found.
common/djangoapps/util/file.py
View file @
07b036f3
...
...
@@ -70,7 +70,8 @@ def store_uploaded_file(
stored_file_name
=
base_storage_filename
+
file_extension
file_storage
=
DefaultStorage
()
file_storage
.
save
(
stored_file_name
,
uploaded_file
)
# If a file already exists with the supplied name, file_storage will make the filename unique.
stored_file_name
=
file_storage
.
save
(
stored_file_name
,
uploaded_file
)
if
validator
:
try
:
...
...
common/djangoapps/util/tests/test_file.py
View file @
07b036f3
...
...
@@ -151,7 +151,7 @@ class StoreUploadedFileTestCase(TestCase):
def
success_validator
(
storage
,
filename
):
""" Validation test function that is a no-op """
self
.
assert
Equal
(
"success_file.csv"
,
os
.
path
.
basename
(
filename
))
self
.
assert
True
(
"success_file"
in
os
.
path
.
basename
(
filename
))
store_file_data
(
storage
,
filename
)
with
self
.
assertRaises
(
FileValidationException
)
as
error
:
...
...
@@ -173,26 +173,46 @@ class StoreUploadedFileTestCase(TestCase):
"""
Tests uploading a file with lower case extension. Verifies that the stored file contents are correct.
"""
self
.
file_storage
,
self
.
stored_file_name
=
store_uploaded_file
(
file_storage
,
stored_file_name
=
store_uploaded_file
(
self
.
request
,
"uploaded_file"
,
[
".csv"
],
"stored_file"
,
self
.
default_max_size
)
self
.
_verify_successful_upload
()
self
.
_verify_successful_upload
(
file_storage
,
stored_file_name
,
self
.
file_content
)
def
test_file_upload_upper_case_extension
(
self
):
"""
Tests uploading a file with upper case extension. Verifies that the stored file contents are correct.
"""
self
.
request
.
FILES
=
{
"uploaded_file"
:
SimpleUploadedFile
(
"tempfile.CSV"
,
self
.
file_content
)}
self
.
file_storage
,
self
.
stored_file_name
=
store_uploaded_file
(
file_content
=
"uppercase"
self
.
request
.
FILES
=
{
"uploaded_file"
:
SimpleUploadedFile
(
"tempfile.CSV"
,
file_content
)}
file_storage
,
stored_file_name
=
store_uploaded_file
(
self
.
request
,
"uploaded_file"
,
[
".gif"
,
".csv"
],
"second_stored_file"
,
self
.
default_max_size
)
self
.
_verify_successful_upload
()
self
.
_verify_successful_upload
(
file_storage
,
stored_file_name
,
file_content
)
def
_verify_successful_upload
(
self
):
def
test_unique_filenames
(
self
):
"""
Test that the file storage method will create a unique filename if the file already exists.
"""
requested_file_name
=
"nonunique_store"
file_content
=
"copy"
self
.
request
.
FILES
=
{
"nonunique_file"
:
SimpleUploadedFile
(
"nonunique.txt"
,
file_content
)}
_
,
first_stored_file_name
=
store_uploaded_file
(
self
.
request
,
"nonunique_file"
,
[
".txt"
],
requested_file_name
,
self
.
default_max_size
)
file_storage
,
second_stored_file_name
=
store_uploaded_file
(
self
.
request
,
"nonunique_file"
,
[
".txt"
],
requested_file_name
,
self
.
default_max_size
)
self
.
assertNotEqual
(
first_stored_file_name
,
second_stored_file_name
)
self
.
assertTrue
(
requested_file_name
in
second_stored_file_name
)
self
.
_verify_successful_upload
(
file_storage
,
second_stored_file_name
,
file_content
)
def
_verify_successful_upload
(
self
,
storage
,
file_name
,
expected_content
):
""" Helper method that checks that the stored version of the uploaded file has the correct content """
self
.
assertTrue
(
s
elf
.
file_storage
.
exists
(
self
.
stored_
file_name
))
with
s
elf
.
file_storage
.
open
(
self
.
stored_
file_name
,
'r'
)
as
f
:
self
.
assertEqual
(
self
.
file
_content
,
f
.
read
())
self
.
assertTrue
(
s
torage
.
exists
(
file_name
))
with
s
torage
.
open
(
file_name
,
'r'
)
as
f
:
self
.
assertEqual
(
expected
_content
,
f
.
read
())
@ddt.ddt
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment