From 4d516e2caee630a04cd6ea456d32cc2fee86d21a Mon Sep 17 00:00:00 2001 From: sophia Date: Tue, 14 Apr 2020 16:57:02 -0400 Subject: [PATCH] Test provision_winssh --- plugins/provisioners/shell/provisioner.rb | 3 +- .../provisioners/shell/provisioner_test.rb | 100 +++++++++++++++++- 2 files changed, 98 insertions(+), 5 deletions(-) diff --git a/plugins/provisioners/shell/provisioner.rb b/plugins/provisioners/shell/provisioner.rb index 257c68d63..d148969a6 100644 --- a/plugins/provisioners/shell/provisioner.rb +++ b/plugins/provisioners/shell/provisioner.rb @@ -4,7 +4,6 @@ require "tempfile" require "vagrant/util/downloader" require "vagrant/util/retryable" -require "pry-byebug" module VagrantPlugins module Shell class Provisioner < Vagrant.plugin("2", :provisioner) @@ -143,7 +142,7 @@ module VagrantPlugins remote_ext = @machine.config.winssh.shell == "cmd" ? ".bat" : ".ps1" end - remote_path = add_extension(upload_path(), remote_ext) + remote_path = add_extension(upload_path, remote_ext) if remote_ext == "bat" command = "#{env}\n cmd.exe /c \"#{remote_path}\" #{args}" else diff --git a/test/unit/plugins/provisioners/shell/provisioner_test.rb b/test/unit/plugins/provisioners/shell/provisioner_test.rb index 03556fcf9..73dc3ca70 100644 --- a/test/unit/plugins/provisioners/shell/provisioner_test.rb +++ b/test/unit/plugins/provisioners/shell/provisioner_test.rb @@ -4,6 +4,7 @@ require Vagrant.source_root.join("plugins/provisioners/shell/provisioner") describe "Vagrant::Shell::Provisioner" do include_context "unit" + let(:default_win_path) { "C:/tmp/vagrant-shell" } let(:env){ isolated_environment } let(:machine) { double(:machine, env: env, id: "ID").tap { |machine| @@ -490,12 +491,105 @@ describe "Vagrant::Shell::Provisioner" do } it "creates an executable with an extension" do - default_path = "C:/tmp/vagrant-shell" - allow(vsp).to receive(:with_script_file).and_yield(default_path) - allow(communicator).to receive(:upload).with(default_path, /vagrant-shell/) + allow(vsp).to receive(:with_script_file).and_yield(default_win_path) + allow(communicator).to receive(:upload).with(default_win_path, /vagrant-shell/) expect(communicator).to receive(:sudo).with(/vagrant-shell.ps1/, anything) vsp.send(:provision_winrm, "") end end end + + describe "#provision_winssh" do + let(:config) { + double( + :config, + :args => "doesn't matter", + :env => {}, + :upload_path => "arbitrary", + :remote? => false, + :path => nil, + :inline => "something", + :binary => false, + :md5 => nil, + :sha1 => 'EXPECTED_VALUE', + :sha256 => nil, + :sha384 => nil, + :sha512 => nil, + :reset => false, + :reboot => false, + :powershell_args => "", + :name => nil, + :privileged => false, + :powershell_elevated_interactive => false + ) + } + + let(:vsp) { + VagrantPlugins::Shell::Provisioner.new(machine, config) + } + + let(:communicator) { double("communicator") } + let(:guest) { double("guest") } + let(:ui) { double("ui") } + + before { + allow(guest).to receive(:capability?).with(:wait_for_reboot).and_return(false) + allow(ui).to receive(:detail) + allow(communicator).to receive(:sudo) + allow(communicator).to receive_message_chain(:machine_config_ssh, :shell) + allow(machine).to receive(:communicate).and_return(communicator) + allow(machine).to receive(:guest).and_return(guest) + allow(machine).to receive(:ui).and_return(ui) + allow(machine).to receive(:ssh_info).and_return(true) + } + + it "ensures that files are uploaded as .bat when shell is cmd" do + allow(machine).to receive_message_chain(:config, :winssh, :shell).and_return("cmd") + allow(vsp).to receive(:with_script_file).and_yield(default_win_path) + expect(communicator).to receive(:upload).with(default_win_path, /arbitrary.bat/) + expect(communicator).to receive(:execute).with(/arbitrary.bat/, anything) + vsp.send(:provision_winssh, "") + end + + it "ensures that files are uploaded as .ps1 when shell is not cmd" do + allow(machine).to receive_message_chain(:config, :winssh, :shell).and_return("ps") + allow(vsp).to receive(:with_script_file).and_yield(default_win_path) + expect(communicator).to receive(:upload).with(default_win_path, /arbitrary.ps1/) + expect(communicator).to receive(:execute).with(/arbitrary.ps1/, anything) + vsp.send(:provision_winssh, "") + end + + context "ps1 file being uploaded" do + let(:config) { + double( + :config, + :args => "doesn't matter", + :env => {}, + :upload_path => "arbitrary", + :remote? => false, + :path => "script/info.ps1", + :binary => false, + :md5 => nil, + :sha1 => 'EXPECTED_VALUE', + :sha256 => nil, + :sha384 => nil, + :sha512 => nil, + :reset => false, + :reboot => false, + :powershell_args => "", + :name => nil, + :privileged => false, + :powershell_elevated_interactive => false + ) + } + + it "ensures that files are uploaded same extension as provided path.ps1" do + allow(machine).to receive_message_chain(:config, :winssh, :shell).and_return("cmd") + allow(vsp).to receive(:with_script_file).and_yield(config.path) + expect(communicator).to receive(:upload).with(config.path, /arbitrary.ps1/) + expect(communicator).to receive(:execute).with(/arbitrary.ps1/, anything) + vsp.send(:provision_winssh, "") + end + end + end end