Introduce role column on User

Author: eliotsykes

app/models/user.rb

@@ -5,6 +5,10 @@ class User < ActiveRecord::Base
          :recoverable, :rememberable, :trackable, :validatable,
          :confirmable, :lockable
 
+  ROLES = ["standard", "premium"].freeze
+
+  validates_inclusion_of :role, in: ROLES
+
   has_many :access_tokens, dependent: :destroy
 
   def issue_access_token

db/migrate/20151206175412_add_role_to_users.rb

@@ -0,0 +1,5 @@
+class AddRoleToUsers < ActiveRecord::Migration
+  def change
+    add_column :users, :role, :string, default: "standard", null: false
+  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: 20151024100339) do
+ActiveRecord::Schema.define(version: 20151206175412) do
 
   create_table "access_tokens", force: :cascade do |t|
     t.string   "locator",          null: false
@@ -47,12 +47,12 @@
   add_index "subscriptions", ["email"], name: "index_subscriptions_on_email", unique: true
 
   create_table "users", force: :cascade do |t|
-    t.string   "email",                  default: "", null: false
-    t.string   "encrypted_password",     default: "", null: false
+    t.string   "email",                  default: "",         null: false
+    t.string   "encrypted_password",     default: "",         null: false
     t.string   "reset_password_token"
     t.datetime "reset_password_sent_at"
     t.datetime "remember_created_at"
-    t.integer  "sign_in_count",          default: 0,  null: false
+    t.integer  "sign_in_count",          default: 0,          null: false
     t.datetime "current_sign_in_at"
     t.datetime "last_sign_in_at"
     t.string   "current_sign_in_ip"
@@ -61,11 +61,12 @@
     t.datetime "confirmed_at"
     t.datetime "confirmation_sent_at"
     t.string   "unconfirmed_email"
-    t.integer  "failed_attempts",        default: 0,  null: false
+    t.integer  "failed_attempts",        default: 0,          null: false
     t.string   "unlock_token"
     t.datetime "locked_at"
     t.datetime "created_at"
     t.datetime "updated_at"
+    t.string   "role",                   default: "standard", null: false
   end
 
   add_index "users", ["confirmation_token"], name: "index_users_on_confirmation_token", unique: true

spec/models/user_spec.rb

@@ -2,18 +2,19 @@
 
 describe User do
 
-  context 'validations' do
+  describe 'validations' do
     it { should validate_presence_of :email }
     it { should validate_presence_of :password }
     it { should validate_confirmation_of :password }
     it { should validate_uniqueness_of(:email).case_insensitive }
+    it { should validate_inclusion_of(:role).in_array(["standard", "premium"]) }
   end
 
-  context 'associations' do
+  describe 'associations' do
     it { should have_many(:access_tokens).dependent(:destroy) }
   end
 
-  context '#issue_access_token' do
+  describe '#issue_access_token' do
     it 'creates access token belonging to user' do
       user = create(:user)
 
@@ -26,4 +27,9 @@
     end
   end
 
+  describe '#role' do
+    it 'defaults to standard' do
+      expect(User.new.role).to eq "standard"
+    end
+  end
 end