diff options
-rw-r--r-- | muck/request.py | 7 | ||||
-rw-r--r-- | muck/request_tests.py | 7 |
2 files changed, 14 insertions, 0 deletions
diff --git a/muck/request.py b/muck/request.py index 2a04d79..4f0e86c 100644 --- a/muck/request.py +++ b/muck/request.py @@ -18,6 +18,13 @@ class Request: def __init__(self, method=None): assert method is not None self._method = method + self._headers = {} def method(self): return self._method + + def add_headers(self, headers): + self._headers.update(headers) + + def get_authorization(self): + return self._headers.get('Authorization') diff --git a/muck/request_tests.py b/muck/request_tests.py index 8bb9194..7151d67 100644 --- a/muck/request_tests.py +++ b/muck/request_tests.py @@ -27,3 +27,10 @@ class RequestTests(unittest.TestCase): def test_has_method(self): r = muck.Request(method='GET') self.assertEqual(r.method(), 'GET') + + def test_returns_authorization_header(self): + r = muck.Request(method='GET') + r.add_headers({ + 'Authorization': 'Bearer XXX', + }) + self.assertEqual(r.get_authorization(), 'Bearer XXX') |