From 64d028d9d423a5883652dffe39ef2cbf2a892b01 Mon Sep 17 00:00:00 2001
From: Zach Smith <zach@driver.xyz>
Date: Sat, 25 Mar 2017 19:29:20 -0700
Subject: Add withAuthToken convenience method to ServiceRequestContext

---
 src/main/scala/xyz/driver/core/rest.scala | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/src/main/scala/xyz/driver/core/rest.scala b/src/main/scala/xyz/driver/core/rest.scala
index d3a06d2..aa2a75e 100644
--- a/src/main/scala/xyz/driver/core/rest.scala
+++ b/src/main/scala/xyz/driver/core/rest.scala
@@ -98,6 +98,9 @@ package rest {
 
     def authToken: Option[AuthToken] =
       contextHeaders.get(AuthProvider.AuthenticationTokenHeader).map(AuthToken.apply)
+
+    def withAuthToken(authToken: AuthToken): ServiceRequestContext =
+      copy(contextHeaders = contextHeaders.updated(AuthProvider.AuthenticationTokenHeader, authToken.value))
   }
 
   object ContextHeaders {
-- 
cgit v1.2.3