Add Rspec expectations