spec/vagrant-windows/winrmcommunicator_spec.rb in vagrant-windows-1.2.1 vs spec/vagrant-windows/winrmcommunicator_spec.rb in vagrant-windows-1.2.2

@@ -9,18 +9,20 @@
     @communicator = VagrantWindows::Communication::WinRMCommunicator.new({})
     @communicator.set_winrmshell(@shell)
   end
   
   describe "execute" do
-    it "should return 1" do
-      expect(@communicator.execute("exit 1")).to eq(1)
+    it "should return 1 when error_check is false" do
+      expect(@communicator.execute("exit 1", { :error_check => false })).to eq(1)
     end
     
-    it "should return 1 with block" do
-      exit_code = @communicator.sudo("exit 1", {}) do |type, line|
-        puts line
-      end
-      expect(exit_code).to eq(1)
+    it "should raise WinRMExecutionError when error_check is true" do
+      expect { @communicator.execute("exit 1") }.to raise_error(VagrantWindows::Errors::WinRMExecutionError)
+    end
+    
+    it "should raise specified error type when specified and error_check is true" do
+      opts = { :error_class => VagrantWindows::Errors::WinRMInvalidShell }
+      expect { @communicator.execute("exit 1", opts) }.to raise_error(VagrantWindows::Errors::WinRMInvalidShell)
     end
   end
 
 end