From 867ef7ad992e5e5160d6a3072a6c3e3e7d7b4044 Mon Sep 17 00:00:00 2001
From: Rebecca Law <rebecca.law@digital.cabinet-office.gov.uk>
Date: Fri, 29 Jan 2016 10:40:55 +0000
Subject: [PATCH] Default service name if service name is not set on the
 session. Pop service name off the session on sign out.

---
 app/main/views/sign_out.py        | 2 ++
 app/templates/admin_template.html | 6 +++---
 2 files changed, 5 insertions(+), 3 deletions(-)

diff --git a/app/main/views/sign_out.py b/app/main/views/sign_out.py
index fe1e47329..d4add09ea 100644
--- a/app/main/views/sign_out.py
+++ b/app/main/views/sign_out.py
@@ -8,5 +8,7 @@ from app.main import main
 @main.route('/sign-out', methods=(['GET']))
 @login_required
 def sign_out():
+    if session.get('service_name', None):
+        session.pop('service_name')
     logout_user()
     return redirect(url_for('main.index'))
diff --git a/app/templates/admin_template.html b/app/templates/admin_template.html
index bfaf44ffa..84687acff 100644
--- a/app/templates/admin_template.html
+++ b/app/templates/admin_template.html
@@ -48,17 +48,17 @@
 
 {% block content %}
 
-  {% if current_user.is_authenticated() %}
+  {% if current_user.is_authenticated() and session['service_name'] != None %}
     <nav class="management-navigation">
       <div class="grid-row">
         <div class="column-half">
           <details class="dropdown">
             <summary class="dropdown-toggle">
-              {{ session['service_name'] }}
+              {{ session.get('service_name', 'Choose service') }}
             </summary>
             <div>
               <a href="{{ url_for('main.choose_service') }}">Switch service</a>
-              <a href="{{ url_for('.add_service') }}">Add a new service to GOV.UK Notify</a>
+              <a href="{{ url_for('main.add_service') }}">Add a new service to GOV.UK Notify</a>
             </div>
           </details>
         </div>