diff --git a/spec/producer/core/task/dsl_spec.rb b/spec/producer/core/task/dsl_spec.rb index afe91a4..9f5be3a 100644 --- a/spec/producer/core/task/dsl_spec.rb +++ b/spec/producer/core/task/dsl_spec.rb @@ -16,7 +16,11 @@ module Producer::Core let(:name) { :some_task } it 'builds a new DSL sandbox with given code' do - expect(Task::DSL).to receive(:new).with(&block).and_call_original + dsl = double('dsl').as_null_object + expect(Task::DSL).to receive(:new).with(no_args) do |&b| + expect(b).to be block + dsl + end Task::DSL.evaluate(name, env, &block) end