diff --git a/lib/producer/core/task/dsl.rb b/lib/producer/core/task/dsl.rb index 218bc54..2d846db 100644 --- a/lib/producer/core/task/dsl.rb +++ b/lib/producer/core/task/dsl.rb @@ -15,7 +15,7 @@ module Producer define_action :file_write, Actions::FileWriter - attr_accessor :env, :actions + attr_accessor :env, :block, :actions, :condition def initialize(env, &block) @env = env diff --git a/spec/producer/core/task/dsl_spec.rb b/spec/producer/core/task/dsl_spec.rb index ac6c09c..d507420 100644 --- a/spec/producer/core/task/dsl_spec.rb +++ b/spec/producer/core/task/dsl_spec.rb @@ -24,12 +24,16 @@ module Producer::Core expect(dsl.env).to be env end + it 'assigns the given block' do + expect(dsl.block).to be block + end + it 'assigns no action' do expect(dsl.actions).to be_empty end it 'assigns true as the condition' do - expect(dsl.instance_eval { @condition }).to be true + expect(dsl.condition).to be true end end