summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--muck/request.py7
-rw-r--r--muck/request_tests.py7
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')