1 files changed, 109 insertions, 23 deletions
diff --git a/spec/models/public_body_heading_spec.rb b/spec/models/public_body_heading_spec.rb
index add2cac60..be3e7c7d2 100644
--- a/spec/models/public_body_heading_spec.rb
+++ b/spec/models/public_body_heading_spec.rb
@@ -2,9 +2,7 @@
 #
 # Table name: public_body_headings
 #
-#  id            :integer        not null, primary key
-#  locale        :string
-#  name          :text           not null
+#  id            :integer          not null, primary key
 #  display_order :integer
 #
 
@@ -12,26 +10,6 @@ require 'spec_helper'
 
 describe PublicBodyHeading do
 
-    context 'when loading the data' do
-
-        before do
-            PublicBodyCategory.add(:en, [
-                  "Local and regional",
-                      [ "local_council", "Local councils", "a local council" ],
-                  "Miscellaneous",
-                      [ "other", "Miscellaneous", "miscellaneous" ],])
-        end
-
-        it 'should use the display_order field to preserve the original data order' do
-            headings = PublicBodyHeading.all
-            headings[0].name.should eq 'Local and regional'
-            headings[0].display_order.should eq 0
-            headings[1].name.should eq 'Miscellaneous'
-            headings[1].display_order.should eq 1
-        end
-
-    end
-
     context 'when validating' do
 
         it 'should require a name' do
@@ -52,6 +30,13 @@ describe PublicBodyHeading do
             heading.valid?
             heading.display_order.should == PublicBodyHeading.next_display_order
         end
+
+        it 'validates the translations' do
+            heading = FactoryGirl.build(:public_body_heading)
+            translation = heading.translations.build
+            expect(heading).to_not be_valid
+        end
+
     end
 
     context 'when setting a display order' do
@@ -65,4 +50,105 @@ describe PublicBodyHeading do
             PublicBodyHeading.next_display_order.should == 1
         end
     end
+
+    describe :save do
+
+      it 'saves translations' do
+          heading = FactoryGirl.build(:public_body_heading)
+          heading.translations_attributes = { :es => { :locale => 'es',
+                                                       :name => 'El Heading' } }
+
+          heading.save
+          expect(PublicBodyHeading.find(heading.id).translations.size).to eq(2)
+      end
+
+    end
+
+    describe :translations_attributes= do
+
+        context 'translation_attrs is a Hash' do
+
+            it 'does not persist translations' do
+                heading = FactoryGirl.create(:public_body_heading)
+                heading.translations_attributes = { :es => { :locale => 'es',
+                                                             :name => 'El Heading' } }
+
+                expect(PublicBodyHeading.find(heading.id).translations.size).to eq(1)
+            end
+
+            it 'creates a new translation' do
+                heading = FactoryGirl.create(:public_body_heading)
+                heading.translations_attributes = { :es => { :locale => 'es',
+                                                             :name => 'El Heading' } }
+                heading.save
+                heading.reload
+                expect(heading.name(:es)).to eq('El Heading')
+            end
+
+            it 'updates an existing translation' do
+                heading = FactoryGirl.create(:public_body_heading)
+                heading.translations_attributes = { 'es' => { :locale => 'es',
+                                                              :name => 'Name' } }
+                heading.save
+
+                heading.translations_attributes = { 'es' => { :id => heading.translation_for(:es).id,
+                                                              :locale => 'es',
+                                                              :name => 'Renamed' } }
+                heading.save
+                expect(heading.name(:es)).to eq('Renamed')
+            end
+
+            it 'updates an existing translation and creates a new translation' do
+                heading = FactoryGirl.create(:public_body_heading)
+                heading.translations.create(:locale => 'es',
+                                             :name => 'Los Heading')
+
+                expect(heading.translations.size).to eq(2)
+
+                heading.translations_attributes = {
+                    'es' => { :id => heading.translation_for(:es).id,
+                              :locale => 'es',
+                              :name => 'Renamed' },
+                    'fr' => { :locale => 'fr',
+                              :name => 'Le Heading' }
+                }
+
+                expect(heading.translations.size).to eq(3)
+                I18n.with_locale(:es) { expect(heading.name).to eq('Renamed') }
+                I18n.with_locale(:fr) { expect(heading.name).to eq('Le Heading') }
+            end
+
+            it 'skips empty translations' do
+                heading = FactoryGirl.create(:public_body_heading)
+                heading.translations.create(:locale => 'es',
+                                             :name => 'Los Heading')
+
+                expect(heading.translations.size).to eq(2)
+
+                heading.translations_attributes = {
+                    'es' => { :id => heading.translation_for(:es).id,
+                              :locale => 'es',
+                              :name => 'Renamed' },
+                    'fr' => { :locale => 'fr' }
+                }
+
+                expect(heading.translations.size).to eq(2)
+            end
+        end
+    end
+end
+
+describe PublicBodyHeading::Translation do
+
+  it 'requires a locale' do
+    translation = PublicBodyHeading::Translation.new
+    translation.valid?
+    expect(translation.errors[:locale]).to eq(["can't be blank"])
+  end
+
+  it 'is valid if all required attributes are assigned' do
+    translation = PublicBodyHeading::Translation.new(:locale => I18n.default_locale)
+    expect(translation).to be_valid
+  end
+
 end