From f573181154f27599a31fb9bbd8fefd98ca259ffd Mon Sep 17 00:00:00 2001
From: dandds <dan-ctr@friends.dds.mil>
Date: Tue, 17 Jul 2018 13:56:36 -0400
Subject: [PATCH] login-dev endpoint accepts role query arg

---
 atst/handlers/dev.py     | 21 ++++++++++++++++-----
 templates/header.html.to |  2 +-
 2 files changed, 17 insertions(+), 6 deletions(-)

diff --git a/atst/handlers/dev.py b/atst/handlers/dev.py
index 38c71419..bfb5684e 100644
--- a/atst/handlers/dev.py
+++ b/atst/handlers/dev.py
@@ -2,8 +2,22 @@ import tornado.gen
 
 from atst.handler import BaseHandler
 
+_DEV_USERS = {
+    "ccpo": {
+        "id": "164497f6-c1ea-4f42-a5ef-101da278c012",
+        "first_name": "Sam",
+        "last_name": "CCPO",
+    },
+    "owner": {
+        "id": "cce17030-4109-4719-b958-ed109dbb87c8",
+        "first_name": "Olivia",
+        "last_name": "Owner",
+    },
+}
+
 
 class Dev(BaseHandler):
+
     def initialize(self, action, sessions, authz_client):
         self.action = action
         self.sessions = sessions
@@ -11,9 +25,6 @@ class Dev(BaseHandler):
 
     @tornado.gen.coroutine
     def get(self):
-        user = {
-            "id": "164497f6-c1ea-4f42-a5ef-101da278c012",
-            "first_name": "Test",
-            "last_name": "User",
-        }
+        role = self.get_argument("role", "ccpo")
+        user = _DEV_USERS[role]
         yield self.login(user)
diff --git a/templates/header.html.to b/templates/header.html.to
index 48ec82e8..f14f4337 100644
--- a/templates/header.html.to
+++ b/templates/header.html.to
@@ -5,7 +5,7 @@
     </a>
 
     <a href="/" class="topbar__link topbar__link--secondary">
-      <span>Tech Lead</span>
+      <span>{{ current_user["atat_role"] }}</span>
     </a>
   </nav>
 </header>