diff --git a/lib/producer/core/env.rb b/lib/producer/core/env.rb index ff29627..b2e3e68 100644 --- a/lib/producer/core/env.rb +++ b/lib/producer/core/env.rb @@ -4,10 +4,11 @@ module Producer attr_reader :input, :output, :registry attr_accessor :target - def initialize(input: $stdin, output: $stdout, registry: {}) + def initialize(input: $stdin, output: $stdout, remote: nil, registry: {}) @input = input @output = output @registry = registry + @remote = remote @target = nil end diff --git a/spec/producer/core/env_spec.rb b/spec/producer/core/env_spec.rb index 74c1422..275f388 100644 --- a/spec/producer/core/env_spec.rb +++ b/spec/producer/core/env_spec.rb @@ -38,6 +38,15 @@ module Producer::Core expect(env.output).to be output end end + + context 'when remote is given as argument' do + let(:remote) { double 'remote' } + subject(:env) { described_class.new(remote: remote) } + + it 'assigns the given remote' do + expect(env.remote).to be remote + end + end end describe '#target' do