1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
|
# -*- coding: utf-8 -*-
import functools
import unittest
import multipart
class TestHeaderParser(unittest.TestCase):
def test_token_unquote(self):
unquote = multipart.header_unquote
self.assertEqual('foo', unquote('"foo"'))
self.assertEqual('foo"bar', unquote('"foo\\"bar"'))
self.assertEqual('ie.exe', unquote('"\\\\network\\ie.exe"', True))
self.assertEqual('ie.exe', unquote('"c:\\wondows\\ie.exe"', True))
unquote = multipart.content_disposition_unquote
self.assertEqual('foo', unquote('"foo"'))
self.assertEqual('foo"bar', unquote('foo%22bar'))
self.assertEqual('foo"bar', unquote('"foo%22bar"'))
self.assertEqual('foo"bar', unquote('"foo\\"bar"'))
self.assertEqual('ie.exe', unquote('"\\\\network\\ie.exe"', True))
self.assertEqual('ie.exe', unquote('"c:\\wondows\\ie.exe"', True))
def test_token_quote(self):
quote = multipart.header_quote
self.assertEqual(quote('foo'), 'foo')
self.assertEqual(quote('foo"bar'), '"foo\\"bar"')
quote = multipart.content_disposition_quote
self.assertEqual(quote('foo'), '"foo"')
self.assertEqual(quote('foo"bar'), '"foo%22bar"')
def test_options_parser(self):
parse = multipart.parse_options_header
head = 'form-data; name="Test"; '
self.assertEqual(parse(head+'filename="Test.txt"')[0], 'form-data')
self.assertEqual(parse(head+'filename="Test.txt"')[1]['name'], 'Test')
self.assertEqual(parse(head+'filename="Test.txt"')[1]['filename'], 'Test.txt')
self.assertEqual(parse(head+'FileName="Te\\"s\\\\t.txt"')[1]['filename'], 'Te"s\\t.txt')
self.assertEqual(parse(head+'filename="C:\\test\\bla.txt"')[1]['filename'], 'bla.txt')
self.assertEqual(parse(head+'filename="\\\\test\\bla.txt"')[1]['filename'], 'bla.txt')
self.assertEqual(parse(head+'filename="täst.txt"')[1]['filename'], 'täst.txt')
parse = functools.partial(multipart.parse_options_header, unquote=multipart.content_disposition_unquote)
self.assertEqual(parse(head+'FileName="Te%22s\\\\t.txt"')[1]['filename'], 'Te"s\\\\t.txt')
|
