From 9d7c5b60924e2c34e9b0ba9ee976d22202572af4 Mon Sep 17 00:00:00 2001
From: Lars Wirzenius <liw@liw.fi>
Date: Sat, 27 Oct 2018 09:59:07 +0300
Subject: Add: Request.get_authorization

---
 muck/request.py       | 7 +++++++
 muck/request_tests.py | 7 +++++++
 2 files changed, 14 insertions(+)

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')
-- 
cgit v1.2.1