Add model to track "votes" for a request

3 files changed, 19 insertions, 0 deletions

diff --git a/app/models/info_request.rb b/app/models/info_request.rb
index d0052603a..0a891df17 100644
--- a/app/models/info_request.rb
+++ b/app/models/info_request.rb
@@ -50,6 +50,7 @@ class InfoRequest < ActiveRecord::Base
     has_many :info_request_events, :order => 'created_at'
     has_many :user_info_request_sent_alerts
     has_many :track_things, :order => 'created_at desc'
+    has_many :widget_votes
     has_many :comments, :order => 'created_at'
     has_many :censor_rules, :order => 'created_at desc'
     has_many :mail_server_logs, :order => 'mail_server_log_done_id'
diff --git a/app/models/widget_vote.rb b/app/models/widget_vote.rb
new file mode 100644
index 000000000..263b8476d
--- /dev/null
+++ b/app/models/widget_vote.rb
@@ -0,0 +1,7 @@
+class WidgetVote < ActiveRecord::Base
+  belongs_to :info_request
+  validates :info_request, :presence => true
+
+  attr_accessible :cookie
+  validates :cookie, :length => { :is => 20 }
+end
diff --git a/db/migrate/20140824191444_create_widget_votes.rb b/db/migrate/20140824191444_create_widget_votes.rb
new file mode 100644
index 000000000..8c406d09f
--- /dev/null
+++ b/db/migrate/20140824191444_create_widget_votes.rb
@@ -0,0 +1,11 @@
+class CreateWidgetVotes < ActiveRecord::Migration
+  def change
+    create_table :widget_votes do |t|
+      t.string :cookie
+      t.belongs_to :info_request
+
+      t.timestamps
+    end
+    add_index :widget_votes, :info_request_id
+  end
+end