diff options
| -rw-r--r-- | libpathod/language.py | 2 | ||||
| -rw-r--r-- | test/test_language.py | 5 |
2 files changed, 4 insertions, 3 deletions
diff --git a/libpathod/language.py b/libpathod/language.py index 30788569..6aae7dc7 100644 --- a/libpathod/language.py +++ b/libpathod/language.py @@ -846,7 +846,6 @@ class Response(_Message): InjectAt, ShortcutContentType, ShortcutLocation, - ShortcutUserAgent, Raw, Reason ) @@ -893,6 +892,7 @@ class Request(_Message): DisconnectAt, InjectAt, ShortcutContentType, + ShortcutUserAgent, Raw ) logattrs = ["method", "path", "body"] diff --git a/test/test_language.py b/test/test_language.py index d7f7b4cc..409b0eb6 100644 --- a/test/test_language.py +++ b/test/test_language.py @@ -300,8 +300,9 @@ class TestHeaders: def test_shortcuts(self): assert language.parse_response({}, "400:c'foo'").headers[0].key.val == "Content-Type" assert language.parse_response({}, "400:l'foo'").headers[0].key.val == "Location" - assert 'Android' in language.parse_response({}, "400:ua").headers[0].value.val - assert language.parse_response({}, "400:ua").headers[0].key.val == "User-Agent" + + assert 'Android' in language.parse_request({}, "get:/:ua").headers[0].value.val + assert language.parse_request({}, "get:/:ua").headers[0].key.val == "User-Agent" class TestShortcutUserAgent: |