TimeTables#edit save color from state

Refs #2899

2 files changed, 7 insertions, 1 deletions

diff --git a/app/models/chouette/time_table.rb b/app/models/chouette/time_table.rb
index fcdc525a9..11e7293f1 100644
--- a/app/models/chouette/time_table.rb
+++ b/app/models/chouette/time_table.rb
@@ -90,7 +90,7 @@ class Chouette::TimeTable < Chouette::TridentActiveRecord
   end
 
   def self.state_permited_attributes item
-    item.slice('comment').to_hash
+    item.slice('comment', 'color').to_hash
   end
 
   def presenter
diff --git a/spec/models/chouette/time_table_spec.rb b/spec/models/chouette/time_table_spec.rb
index 579300b74..18d76e805 100644
--- a/spec/models/chouette/time_table_spec.rb
+++ b/spec/models/chouette/time_table_spec.rb
@@ -51,6 +51,12 @@ describe Chouette::TimeTable, :type => :model do
       }.to change {subject.periods.count}.by(-1)
     end
 
+    it 'should update color' do
+      state['color'] = '#FFA070'
+      subject.state_update state
+      expect(subject.reload.color).to eq(state['color'])
+    end
+
     it 'should save new tags' do
       subject.tag_list = "awesome, great"
       subject.save