Use rspec expect / allow syntax