formatting address on user profile page

Author: lilwillifo

app/models/postal_address.rb

@@ -1,8 +1,15 @@
+# frozen_string_literal: true
 class PostalAddress < ApplicationRecord
   belongs_to :user
   validates_presence_of :address_line_1, :city, :postal_code, :country
 
   def formatted
-    "#{address_line_1.split.map(&:capitalize).join(' ')} #{address_line_2} #{city}, #{state_or_province} #{postal_code} #{country}"
+    "#{capitalize(address_line_1)} #{capitalize(address_line_2)} #{capitalize(city)}, #{state_or_province} #{postal_code} #{country}"
+  end
+
+  private
+
+  def capitalize(string)
+    string.split.map(&:capitalize).join(' ')
   end
 end

spec/factories/postal_address.rb

@@ -6,6 +6,6 @@
     city { FFaker::AddressUS.city}
     state_or_province { FFaker::AddressUS.state}
     postal_code { FFaker::AddressUS.zip_code}
-    country "Barbados"
+    country { FFaker::Address.country }
   end
 end

spec/features/users/postal_address_spec.rb

@@ -26,7 +26,7 @@
         expect(page).to have_content address.city
         expect(page).to have_content address.state_or_province
         expect(page).to have_content address.postal_code
-        # expect(page).to have_content address.country
+        expect(page).to have_content user.postal_address.country
       end
 
       it 'does not allow a postal address to be added if required fields are missing' do