Add Officer relationship to Poll Voter

app/controllers/officing/voters_controller.rb

@@ -13,7 +13,8 @@ class Officing::VotersController < Officing::BaseController
                              document_number: @user.document_number,
                              user: @user,
                              poll: @poll,
-                             origin: "booth")
+                             origin: "booth",
+                             officer: current_user.poll_officer)
     @voter.save!
   end
 

app/models/poll/voter.rb

@@ -8,6 +8,7 @@ class Poll
     belongs_to :geozone
     belongs_to :booth_assignment
     belongs_to :officer_assignment
+    belongs_to :officer
 
     validates :poll_id, presence: true
     validates :user_id, presence: true

db/migrate/20171003223152_add_officer_to_poll_voter.rb

@@ -0,0 +1,5 @@
+class AddOfficerToPollVoter < ActiveRecord::Migration
+  def change
+    add_column :poll_voters, :officer_id, :integer
+  end
+end

db/schema.rb

@@ -11,7 +11,7 @@
 #
 # It's strongly recommended that you check this file into your version control system.
 
-ActiveRecord::Schema.define(version: 20171003212958) do
+ActiveRecord::Schema.define(version: 20171003223152) do
 
   # These are extensions that must be enabled in order to support this database
   enable_extension "plpgsql"
@@ -752,6 +752,7 @@ ActiveRecord::Schema.define(version: 20171003212958) do
     t.integer  "officer_assignment_id"
     t.integer  "user_id"
     t.string   "origin"
+    t.integer  "officer_id"
   end
 
   add_index "poll_voters", ["booth_assignment_id"], name: "index_poll_voters_on_booth_assignment_id", using: :btree

spec/factories.rb

@@ -531,6 +531,7 @@ FactoryGirl.define do
   factory :poll_voter, class: 'Poll::Voter' do
     poll
     association :user, :level_two
+    association :officer, factory: :poll_officer
     origin "web"
 
     trait :from_booth do