groups and headings navigation

app/views/budgets/groups/show.html.erb

@@ -1,9 +1,9 @@
 <div class="row investment-projects-scope margin-bottom no-margin-top">
   <div id="select-district" class="small-12 medium-7 column select-district">
-    <%# link_to participatory_budget_path, class: "back" do %>
+    <%= link_to budget_path(@budget), class: "back" do %>
       <span class="icon-angle-left"></span>
       <%= t('shared.back') %>
-    <%# end %>
+    <% end %>
 
     <h2><%# t("spending_proposals.welcome.districts") %></h2>
 
@@ -13,8 +13,9 @@
           <% slice.each do |heading| %>
             <span id="<%= dom_id(heading) %>"
                   class="<%= css_for_ballot_heading(heading) %>">
-              <%= link_to heading.name, budget_investments_path(heading_id: heading.id),
-                        data: { no_turbolink: true } %><br>
+              <%= link_to heading.name,
+                          budget_investments_path(heading_id: heading.id),
+                          data: { no_turbolink: true } %><br>
             </span>
           <% end %>
         </div>

app/views/budgets/investments/_header.html.erb

@@ -4,7 +4,7 @@
 
       <div class="row">
         <div class="small-12 column">
-          <%= link_to @budget, class: "back" do %>
+          <%= link_to budget_group_path(@budget, @heading.group), class: "back" do %>
             <i class="icon-angle-left"></i>
             <%= t("shared.back") %>
           <% end %>

app/views/budgets/show.html.erb

@@ -22,11 +22,18 @@
         <% @budget.groups.each do |group| %>
           <tr>
             <td>
-              <%= link_to group.name, budget_group_path(@budget, group) %>
+              <% if group.headings.count == 1 %>
+                <%= link_to group.name,
+                            budget_investments_path(@budget, heading_id: group.headings.first.id),
+                            data: { no_turbolink: true } %>
+              <% else %>
+                <%= link_to group.name, budget_group_path(@budget, group) %>
+              <% end %>
+              <br>
             </td>
           </tr>
         <% end %>
       </tbody>
     </table>
   </div>
-</div>
+</div>