makes ballot's back link point to referer

spec/features/budgets/ballots_spec.rb

@@ -350,7 +350,7 @@ feature 'Ballots' do
 
   end
 
-  scenario 'Removing spending proposals from ballot', :js do
+  scenario 'Removing investments from ballot', :js do
     investment = create(:budget_investment, :selected, price: 10, heading: new_york)
     ballot = create(:budget_ballot, user: user, budget: budget)
     ballot.investments << investment
@@ -368,7 +368,7 @@ feature 'Ballots' do
     expect(page).to have_content("You have voted 0 proposals")
   end
 
-  scenario 'Removing spending proposals from ballot (sidebar)', :js do
+  scenario 'Removing investments from ballot (sidebar)', :js do
     investment1 = create(:budget_investment, :selected, price: 10000, heading: new_york)
     investment2 = create(:budget_investment, :selected, price: 20000, heading: new_york)
 
@@ -405,6 +405,28 @@ feature 'Ballots' do
     end
   end
 
+  scenario 'Back link after removing an investment from Ballot', :js do
+    investment = create(:budget_investment, :selected, heading: new_york, price: 10)
+
+    login_as(user)
+    visit budget_investments_path(budget, heading_id: new_york.id)
+    add_to_ballot(investment)
+
+    click_link "Check my ballot"
+
+    expect(page).to have_content("You have voted one proposal")
+
+    within("#budget_investment_#{investment.id}") do
+      find(".remove-investment-project").trigger('click')
+    end
+
+    expect(page).to have_content("You have voted 0 proposals")
+
+    click_link "Go back"
+
+    expect(page).to have_current_path(budget_investments_path(budget, heading_id: new_york.id))
+  end
+
   context 'Permissions' do
 
     scenario 'User not logged in', :js do