From 88036f0389c95be81e90e2492a7a7d1e51d26daa Mon Sep 17 00:00:00 2001
From: Thibault Jouan <tj@a13.fr>
Date: Fri, 14 Feb 2014 18:32:35 +0000
Subject: [PATCH] Allow Env constructor to accept a remote instance

---
 lib/producer/core/env.rb       | 3 ++-
 spec/producer/core/env_spec.rb | 9 +++++++++
 2 files changed, 11 insertions(+), 1 deletion(-)

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