Add: Request.get_authorization

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')