From 8eef8b095d8af2299062c572c59445fb4163e0d8 Mon Sep 17 00:00:00 2001
From: Mike Olund <michaelo@openroad.ca>
Date: Thu, 16 Mar 2017 17:11:54 -0700
Subject: [PATCH] DIV-273 - Overview page missing optional steps

---
 edivorce/apps/core/templates/overview.html | 6 +++---
 edivorce/apps/core/views/main.py           | 9 +++++----
 2 files changed, 8 insertions(+), 7 deletions(-)

diff --git a/edivorce/apps/core/templates/overview.html b/edivorce/apps/core/templates/overview.html
index 411e1dcd..2807a947 100644
--- a/edivorce/apps/core/templates/overview.html
+++ b/edivorce/apps/core/templates/overview.html
@@ -46,7 +46,7 @@
         {% elif step_status.your_separation == 'Complete' %}<span class="progress-status"><i class="fa fa-check" aria-hidden="true"></i></span> {% endif %}
     </a>
 
-    {% if 'Spousal support' in want_which_orders|load_json %}
+    {% if 'Spousal support' in which_orders.0.value|load_json %}
     <a href="{% url 'question_steps' 'support' %}" class="progress-question {% if active_page == 'support' %} active {% endif %}">
         <span class="progress-icon"><i class="fa fa-life-ring" aria-hidden="true"></i></span>
         <span class="progress-content"><small>Step 6</small><br />Spousal support</span>
@@ -55,7 +55,7 @@
     </a>
     {% endif %}
 
-    {% if 'Division of property and debts' in want_which_orders|load_json %}
+    {% if 'Division of property and debts' in which_orders.0.value|load_json %}
     <a href="{% url 'question_steps' 'property' %}" class="progress-question {% if step_status.property_and_debt == 'Complete' %} complete {% endif %}">
         <span class="progress-icon"><i class="fa fa-home" aria-hidden="true"></i></span>
         <span class="progress-content"><small>Step {% step_order step="property" %}</small><br />Property and debt</span>
@@ -64,7 +64,7 @@
     </a>
     {% endif %}
 
-    {% if 'Other orders' in want_which_orders|load_json %}
+    {% if 'Other orders' in which_orders.0.value|load_json %}
     <a href="{% url 'question_steps' 'other_orders' %}" class="progress-question {% if step_status.other_orders == 'Complete' %} complete {% endif %}">
         <span class="progress-icon"><i class="fa fa-gavel" aria-hidden="true"></i></span>
         <span class="progress-content"><small>Step {% step_order step="other_orders" %}</small><br />Other orders</span>
diff --git a/edivorce/apps/core/views/main.py b/edivorce/apps/core/views/main.py
index 8cf64ff9..46faab49 100644
--- a/edivorce/apps/core/views/main.py
+++ b/edivorce/apps/core/views/main.py
@@ -141,10 +141,11 @@ def overview(request):
     """
     user = __get_bceid_user(request)
     responses_dict_by_step = get_responses_from_db_grouped_by_steps(user)
-    # step status dictionary
-    status_dict = {'step_status': get_step_status(responses_dict_by_step),
-                   'active_page': 'overview'}
-    return render(request, 'overview.html', context=status_dict)
+    responses_dict_by_step['active_page'] = 'overview'
+
+    # Add step status dictionary
+    responses_dict_by_step['step_status'] = get_step_status(responses_dict_by_step)
+    return render(request, 'overview.html', context=responses_dict_by_step)
 
 
 def __get_bceid_user(request):