mirror of https://github.com/flaggo/pydu.git
39 lines
1.3 KiB
Python
39 lines
1.3 KiB
Python
|
from pydu.request import FileName
|
||
|
|
|
||
|
|
|
||
|
|
def test_filename_from_url():
|
||
|
|
url = 'http://www.example.com/test.txt'
|
||
|
|
assert FileName.from_url(url) == 'test.txt'
|
||
|
|
|
||
|
|
url = 'http://www.example.com/'
|
||
|
|
assert FileName.from_url(url) is None
|
||
|
|
|
||
|
|
|
||
|
|
def test_filename_from_headers():
|
||
|
|
headers = {'Content-Disposition': 'attachment; filename=test.txt'}
|
||
|
|
assert FileName.from_headers(headers) == 'test.txt'
|
||
|
|
|
||
|
|
headers = [('Content-Disposition', 'attachment; filename=test.txt')]
|
||
|
|
assert FileName.from_headers(headers) == 'test.txt'
|
||
|
|
|
||
|
|
headers = 'Content-Disposition: attachment; filename=test.txt'
|
||
|
|
assert FileName.from_headers(headers) == 'test.txt'
|
||
|
|
|
||
|
|
headers = 'Content-Disposition: attachment; filename=abc/test.txt'
|
||
|
|
assert FileName.from_headers(headers) == 'test.txt'
|
||
|
|
|
||
|
|
headers = ''
|
||
|
|
assert FileName.from_headers(headers) is None
|
||
|
|
|
||
|
|
headers = 'Content-Disposition: abc'
|
||
|
|
assert FileName.from_headers(headers) is None
|
||
|
|
|
||
|
|
headers = 'Content-Disposition: abc;'
|
||
|
|
assert FileName.from_headers(headers) is None
|
||
|
|
|
||
|
|
headers = 'Content-Disposition: attachment; filename=test.txt; filename=test2.txt'
|
||
|
|
assert FileName.from_headers(headers) is None
|
||
|
|
|
||
|
|
headers = 'Content-Disposition: attachment; filename='
|
||
|
|
assert FileName.from_headers(headers) is None
|