Spec the failure cases for PreferencesController#update

2 files changed, 27 insertions, 6 deletions

diff --git a/app/controllers/profiles/preferences_controller.rb b/app/controllers/profiles/preferences_controller.rb
index e43a247f725..538b09ca54d 100644
--- a/app/controllers/profiles/preferences_controller.rb
+++ b/app/controllers/profiles/preferences_controller.rb
@@ -5,10 +5,15 @@ class Profiles::PreferencesController < Profiles::ApplicationController
   end
 
   def update
-    if @user.update_attributes(preferences_params)
-      flash[:notice] = 'Preferences saved.'
-    else
-      # TODO (rspeicher): There's no validation on these values, so can it fail?
+    begin
+      if @user.update_attributes(preferences_params)
+        flash[:notice] = 'Preferences saved.'
+      else
+        flash[:alert] = 'Failed to save preferences.'
+      end
+    rescue ArgumentError => e
+      # Raised when `dashboard` is given an invalid value.
+      flash[:alert] = "Failed to save preferences (#{e.message})."
     end
 
     respond_to do |format|
diff --git a/spec/controllers/profiles/preferences_controller_spec.rb b/spec/controllers/profiles/preferences_controller_spec.rb
index 646aa0320b2..1f0943c93d8 100644
--- a/spec/controllers/profiles/preferences_controller_spec.rb
+++ b/spec/controllers/profiles/preferences_controller_spec.rb
@@ -51,8 +51,24 @@ describe Profiles::PreferencesController do
       end
     end
 
-    context 'on unsuccessful update' do
-      # TODO (rspeicher): Can this happen?
+    context 'on failed update' do
+      it 'sets the flash' do
+        expect(user).to receive(:update_attributes).and_return(false)
+
+        go
+
+        expect(flash[:alert]).to eq('Failed to save preferences.')
+      end
+    end
+
+    context 'on invalid dashboard setting' do
+      it 'sets the flash' do
+        prefs = {dashboard: 'invalid'}
+
+        go params: prefs
+
+        expect(flash[:alert]).to match(/\AFailed to save preferences \(.+\)\.\z/)
+      end
     end
 
     context 'as js' do