[CHEF-2994][RSPEC] ShellOut#run_command when exiting with invalid code with input data shuld handle exit code gracefully

1 files changed, 13 insertions, 0 deletions

diff --git a/spec/mixlib/shellout/shellout_spec.rb b/spec/mixlib/shellout/shellout_spec.rb
index 7d37a3e..62ac3e5 100644
--- a/spec/mixlib/shellout/shellout_spec.rb
+++ b/spec/mixlib/shellout/shellout_spec.rb
@@ -593,6 +593,19 @@ describe Mixlib::ShellOut do
           it "should set the exit status of the command" do
             exit_status.should eql(exit_code)
           end
+
+          context 'with input data' do
+            let(:options) { { :returns => valid_exit_codes, :input => input } }
+            let(:input) { "Random data #{rand(1000000)}" }
+
+            it "should raise ShellCommandFailed" do
+              lambda { executed_cmd.error! }.should raise_error(Mixlib::ShellOut::ShellCommandFailed)
+            end
+
+            it "should set the exit status of the command" do
+              exit_status.should eql(exit_code)
+            end
+          end
         end
 
         context 'when exiting with invalid code 0' do