Wrong reference of embedded document at after_save cb

test.rb

require 'mongoid'
require 'rspec'

Mongoid.configure.connect_to('mongoid_test')

class Syncer
  def self.perform(address); end
end

class Customer
  include Mongoid::Document
  embeds_one :address

  def address_changed
    Syncer.perform(address)
  end
end

class Address
  include Mongoid::Document
  embedded_in :customer

  after_save { customer.address_changed }
end


describe 'set the address via address=(address)' do
  it 'knows about the new address' do
    customer = Customer.create
    address  = Address.new

    expect(Syncer).to receive(:perform).with(address).once

    customer.address = address
  end
end