Add trait to create a poll associated to a budget

spec/features/budget_polls/ballot_sheets_spec.rb

@@ -1,8 +1,7 @@
 require "rails_helper"
 
 describe "Poll budget ballot sheets" do
-  let(:budget) { create(:budget) }
-  let(:poll) { create(:poll, budget: budget, ends_at: 1.day.ago) }
+  let(:poll) { create(:poll, :for_budget, ends_at: 1.day.ago) }
   let(:booth) { create(:poll_booth) }
   let(:poll_officer) { create(:poll_officer) }
 

spec/features/budget_polls/budgets_spec.rb

@@ -60,8 +60,7 @@ describe "Admin Budgets" do
   context "Show" do
 
     scenario "Do not show questions section if the budget have a poll associated" do
-      budget = create(:budget)
-      poll = create(:poll, budget: budget)
+      poll = create(:poll, :for_budget)
 
       visit admin_poll_path(poll)
 

spec/features/budget_polls/polls_spec.rb

@@ -6,7 +6,7 @@ describe "Polls" do
 
     scenario "Budget polls should not be listed" do
       poll = create(:poll)
-      budget_poll = create(:poll, budget: create(:budget))
+      budget_poll = create(:poll, :for_budget)
 
       visit polls_path
 
@@ -22,7 +22,7 @@ describe "Polls" do
       login_as(create(:administrator).user)
 
       poll = create(:poll)
-      budget_poll = create(:poll, budget: create(:budget))
+      budget_poll = create(:poll, :for_budget)
 
       visit admin_polls_path
 

spec/features/budget_polls/questions_spec.rb

@@ -8,10 +8,8 @@ describe "Poll Questions" do
   end
 
   scenario "Do not display polls associated to a budget" do
-    budget = create(:budget)
-
     poll1 = create(:poll, name: "Citizen Proposal Poll")
-    poll2 = create(:poll, budget: budget, name: "Participatory Budget Poll")
+    poll2 = create(:poll, :for_budget, name: "Participatory Budget Poll")
 
     visit admin_questions_path