Adjust officing ballot spec to poll restrictions

consuldemocracy · Sep 5, 2022 · bfe2997 · bfe2997
1 parent 5f0bdff
commit bfe2997
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/spec/system/officing/ballot_sheets_spec.rb b/spec/system/officing/ballot_sheets_spec.rb
@@ -3,12 +3,14 @@
 describe "Officing ballot sheets" do
   let(:budget) { create(:budget, :reviewing_ballots) }
   let(:heading) { create(:budget_heading, budget: budget, price: 300) }
-  let(:poll) { create(:poll, name: "Latest budget poll", budget: budget, ends_at: Date.current) }
+  let(:poll) { create(:poll, :current, name: "Latest budget poll", budget: budget) }
   let(:booth) { create(:poll_booth, name: "The only booth") }
   let(:officer) { create(:poll_officer) }
   let!(:admin) { create(:administrator).user }
 
   scenario "Create a ballot sheet for a budget poll" do
+    poll.update_columns ends_at: Date.current.beginning_of_day
+
     create(:poll_officer_assignment, officer: officer, poll: poll, booth: booth)
     create(:poll_shift, :recount_scrutiny_task, officer: officer, booth: booth, date: Date.current)