Polish some styles and comments

core/app/models/concerns/spree/user_address_book.rb

@@ -13,10 +13,19 @@ def find_first_by_address_values(address_attrs)
         def mark_default(user_address, address_type: :shipping)
           column_for_default = address_type == :shipping ? :default : :default_billing
           ActiveRecord::Base.transaction do
-            (self - [user_address]).each do |ua| # update_all would be nice, but it bypasses ActiveRecord callbacks
-              ua.persisted? ? ua.update!({ column_for_default => false }) : ua.write_attribute(column_for_default, false)
+            (self - [user_address]).each do |address| # update_all would be nice, but it bypasses ActiveRecord callbacks
+              if address.persisted?
+                address.update!(column_for_default => false)
+              else
+                address.write_attribute(column_for_default, false)
+              end
+            end
+
+            if user_address.persisted?
+              user_address.update!(column_for_default => true, archived: false)
+            else
+              user_address.write_attribute(column_for_default, true)
             end
-            user_address.persisted? ? user_address.update!({ column_for_default => true, :archived => false }) : user_address.write_attribute(column_for_default, true)
           end
         end
       end
@@ -112,7 +121,8 @@ def persist_order_address(order)
     # @param default set whether or not this address will show up from
     # #default_address or not
     def save_in_address_book(address_attributes, default = false, address_type = :shipping)
-      return nil unless address_attributes.present?
+      return nil if address_attributes.blank?
+
       address_attributes = address_attributes.to_h.with_indifferent_access
 
       new_address = Spree::Address.factory(address_attributes)

core/spec/models/spree/concerns/user_address_book_spec.rb

@@ -11,7 +11,7 @@ module Spree
     let!(:user) { create(:user) }
 
     describe "#save_in_address_book" do
-      context "saving a shipping address" do
+      context "saving a default shipping address" do
         let(:user_address) { user.user_addresses.find_first_by_address_values(address.attributes) }
 
         subject do
@@ -45,7 +45,7 @@ module Spree
             end
           end
 
-          it "adds the address to the user's the associated addresses" do
+          it "adds the address to the user's associated addresses" do
             is_expected.to change { user.reload.addresses.count }.by(1)
           end
         end
@@ -404,7 +404,7 @@ module Spree
         end
       end
 
-      context "when one order address is nil" do
+      context "when either ship_address or bill_address is nil" do
         context "when automatic_default_address preference is at a default of true" do
           before do
             stub_spree_preferences(automatic_default_address: true)
@@ -428,20 +428,20 @@ module Spree
         end
 
         context "when automatic_default_address preference is false" do
+          let(:order) { build(:order) }
+
           before do
             stub_spree_preferences(automatic_default_address: false)
           end
 
           it "does not call save_in_address_book on ship address" do
-            order = build(:order)
             order.ship_address = nil
 
             expect(user).to receive(:save_in_address_book).with(kind_of(Hash), false, :billing).once
             user.persist_order_address(order)
           end
 
           it "does not call save_in_address_book on bill address" do
-            order = build(:order)
             order.bill_address = nil
 
             expect(user).to receive(:save_in_address_book).with(kind_of(Hash), false).once
@@ -465,7 +465,7 @@ module Spree
       let!(:address) { create(:address) }
 
       # https://github.com/solidusio/solidus/issues/1241
-      it "resets the association and persists" do
+      it "resets the association and persists the ship_address" do
         # Load (which will cache) the has_one association
         expect(user.ship_address).to be_nil
 
@@ -482,7 +482,7 @@ module Spree
       let!(:address) { create(:address) }
 
       # https://github.com/solidusio/solidus/issues/1241
-      it "resets the association and persists" do
+      it "resets the association and persists the bill_address" do
         # Load (which will cache) the has_one association
         expect(user.bill_address).to be_nil