diff --git a/lib/producer/core/recipe/dsl.rb b/lib/producer/core/recipe/dsl.rb index 0a8e712..74c3864 100644 --- a/lib/producer/core/recipe/dsl.rb +++ b/lib/producer/core/recipe/dsl.rb @@ -2,7 +2,7 @@ module Producer module Core class Recipe class DSL - attr_reader :env, :tasks + attr_reader :env, :code, :block, :tasks def initialize(env, code = nil, &block) @env = env diff --git a/spec/producer/core/recipe/dsl_spec.rb b/spec/producer/core/recipe/dsl_spec.rb index 6ae2a1e..52dcc7d 100644 --- a/spec/producer/core/recipe/dsl_spec.rb +++ b/spec/producer/core/recipe/dsl_spec.rb @@ -22,13 +22,13 @@ module Producer::Core subject(:dsl) { described_class.new(env, code) } it 'assigns the string of code' do - expect(dsl.instance_eval { @code }).to eq code + expect(dsl.code).to eq code end end context 'when a code block is given as argument' do it 'assigns the code block' do - expect(dsl.instance_eval { @block }).to be code + expect(dsl.block).to be code end end end