124 lines
3.2 KiB
Ruby
124 lines
3.2 KiB
Ruby
require File.expand_path("../../../../base", __FILE__)
|
|
|
|
require Vagrant.source_root.join("plugins/commands/ssh_config/command")
|
|
|
|
describe VagrantPlugins::CommandSSHConfig::Command do
|
|
include_context "unit"
|
|
include_context "virtualbox"
|
|
|
|
let(:iso_env) do
|
|
# We have to create a Vagrantfile so there is a root path
|
|
env = isolated_environment
|
|
env.vagrantfile("")
|
|
env.create_vagrant_env
|
|
end
|
|
|
|
let(:guest) { double("guest") }
|
|
let(:host) { double("host") }
|
|
let(:machine) { iso_env.machine(iso_env.machine_names[0], :dummy) }
|
|
|
|
let(:argv) { [] }
|
|
let(:ssh_info) {{
|
|
host: "testhost.vagrant.dev",
|
|
port: 1234,
|
|
username: "testuser",
|
|
private_key_path: [],
|
|
forward_agent: false,
|
|
forward_x11: false
|
|
}}
|
|
|
|
subject { described_class.new(argv, iso_env) }
|
|
|
|
before do
|
|
machine.stub(ssh_info: ssh_info)
|
|
allow(subject).to receive(:with_target_vms) { |&block| block.call machine }
|
|
end
|
|
|
|
describe "execute" do
|
|
it "prints out the ssh config for the given machine" do
|
|
output = ""
|
|
allow(subject).to receive(:safe_puts) do |data|
|
|
output += data if data
|
|
end
|
|
|
|
subject.execute
|
|
|
|
expect(output).to eq(<<-SSHCONFIG)
|
|
Host #{machine.name}
|
|
HostName testhost.vagrant.dev
|
|
User testuser
|
|
Port 1234
|
|
UserKnownHostsFile /dev/null
|
|
StrictHostKeyChecking no
|
|
PasswordAuthentication no
|
|
IdentitiesOnly yes
|
|
LogLevel FATAL
|
|
SSHCONFIG
|
|
end
|
|
|
|
it "turns on agent forwarding when it is configured" do
|
|
allow(machine).to receive(:ssh_info) { ssh_info.merge(forward_agent: true) }
|
|
|
|
output = ""
|
|
allow(subject).to receive(:safe_puts) do |data|
|
|
output += data if data
|
|
end
|
|
|
|
subject.execute
|
|
|
|
expect(output).to include("ForwardAgent yes")
|
|
end
|
|
|
|
it "turns on x11 forwarding when it is configured" do
|
|
allow(machine).to receive(:ssh_info) { ssh_info.merge(forward_x11: true) }
|
|
|
|
output = ""
|
|
allow(subject).to receive(:safe_puts) do |data|
|
|
output += data if data
|
|
end
|
|
|
|
subject.execute
|
|
|
|
expect(output).to include("ForwardX11 yes")
|
|
end
|
|
|
|
it "handles multiple private key paths" do
|
|
allow(machine).to receive(:ssh_info) { ssh_info.merge(private_key_path: ["foo", "bar"]) }
|
|
|
|
output = ""
|
|
allow(subject).to receive(:safe_puts) do |data|
|
|
output += data if data
|
|
end
|
|
|
|
subject.execute
|
|
|
|
expect(output).to include('IdentityFile "foo"')
|
|
expect(output).to include('IdentityFile "bar"')
|
|
end
|
|
|
|
it "puts quotes around an identityfile path if it has a space" do
|
|
allow(machine).to receive(:ssh_info) { ssh_info.merge(private_key_path: ["with a space"]) }
|
|
output = ""
|
|
allow(subject).to receive(:safe_puts) do |data|
|
|
output += data if data
|
|
end
|
|
|
|
subject.execute
|
|
|
|
expect(output).to include('IdentityFile "with a space"')
|
|
end
|
|
|
|
it "escapes special characters" do
|
|
allow(machine).to receive(:ssh_info) { ssh_info.merge(private_key_path: ['/private/tmp/test of "vagrant" plugin/']) }
|
|
output = ""
|
|
allow(subject).to receive(:safe_puts) do |data|
|
|
output += data if data
|
|
end
|
|
|
|
subject.execute
|
|
|
|
expect(output).to include('"/private/tmp/test of \"vagrant\" plugin/"')
|
|
end
|
|
end
|
|
end
|