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
47
48
49
50
51
52
53
54
|
import pytest
import headerparser
from headerparser import HeaderParser
def test_multiname_use_first() -> None:
parser = HeaderParser()
parser.add_field("Foo", "Bar")
msg = parser.parse("Foo: red")
assert dict(msg) == {"Foo": "red"}
assert msg.body is None
def test_multiname_use_second() -> None:
parser = HeaderParser()
parser.add_field("Foo", "Bar")
msg = parser.parse("Bar: red")
assert dict(msg) == {"Foo": "red"}
assert msg.body is None
def test_multiname_multiple() -> None:
parser = HeaderParser()
parser.add_field("Foo", "Bar", multiple=True)
parser.add_field("Baz")
msg = parser.parse("Foo: red\nBar: green\nBaz: blue\n")
assert dict(msg) == {"Foo": ["red", "green"], "Baz": "blue"}
assert msg.body is None
def test_multiname_bad_multiple() -> None:
parser = HeaderParser()
parser.add_field("Foo", "Bar")
parser.add_field("Baz")
with pytest.raises(headerparser.DuplicateFieldError) as excinfo:
parser.parse("Foo: red\nBar: green\nBaz: blue\n")
assert str(excinfo.value) == "Header field 'Foo' occurs more than once"
assert excinfo.value.name == "Foo"
def test_multiname_conflict() -> None:
parser = HeaderParser()
parser.add_field("Foo", "Bar", multiple=True)
with pytest.raises(ValueError) as excinfo:
parser.add_field("Baz", "BAR")
assert "field defined more than once" in str(excinfo.value)
def test_multiname_dest() -> None:
parser = HeaderParser()
parser.add_field("Foo", "Bar", dest="Baz")
msg = parser.parse("Bar: red")
assert dict(msg) == {"Baz": "red"}
assert msg.body is None
|
