Close SSH session before exiting

spec/producer/core/cli_spec.rb

@@ -172,6 +172,11 @@ module Producer::Core
           .with([an_instance_of(Task), an_instance_of(Task)])
         cli.run
       end
+
+      it 'cleans up the env' do
+        expect(cli.env).to receive :cleanup
+        cli.run
+      end
     end
 
     describe '#load_recipe' do

spec/producer/core/env_spec.rb

@@ -157,5 +157,12 @@ module Producer::Core
         expect(env).to be_dry_run
       end
     end
+
+    describe '#cleanup' do
+      it 'cleans up the remote' do
+        expect(env.remote).to receive :cleanup
+        env.cleanup
+      end
+    end
   end
 end

spec/producer/core/remote_spec.rb

@@ -136,5 +136,14 @@ module Producer::Core
         expect(remote.environment['FOO']).to eq 'bar'
       end
     end
+
+    describe '#cleanup' do
+      before { remote.session = double 'session' }
+
+      it 'closes the session' do
+        expect(remote.session).to receive :close
+        remote.cleanup
+      end
+    end
   end
 end