diff --git a/spec/unit/action/forward_ports_spec.rb b/spec/unit/action/forward_ports_spec.rb index bbaece4..4109891 100644 --- a/spec/unit/action/forward_ports_spec.rb +++ b/spec/unit/action/forward_ports_spec.rb @@ -6,7 +6,7 @@ require 'vagrant-lxc/action/forward_ports' describe Vagrant::LXC::Action::ForwardPorts do let(:app) { double(:app, call: true) } - let(:env) { {machine: machine, ui: double(info: true)} } + let(:env) { {machine: machine, ui: double(info: true, warn: true)} } let(:machine) { double(:machine) } let!(:data_dir) { Pathname.new(Dir.mktmpdir) } let(:provider) { double(Vagrant::LXC::Provider, ssh_info: {host: container_ip}) } @@ -74,4 +74,34 @@ describe Vagrant::LXC::Action::ForwardPorts do subject.stub(system: false) expect { subject.call(env) }.to raise_error(Vagrant::LXC::Errors::RedirNotInstalled) end + + context 'when a privileged port is used' do + let(:host_port) { 80 } + + it 'forwards ports using redir' do + subject.stub(system: true) + subject.call(env) + expect(subject).to have_received(:spawn).with( + "sudo redir --laddr=#{host_ip} --lport=#{host_port} --caddr=#{container_ip} --cport=#{guest_port} 2>/dev/null" + ) + end + + it 'skips --laddr parameter if host_ip is nil' do + forward_conf.delete(:host_ip) + subject.stub(system: true) + subject.call(env) + expect(subject).to have_received(:spawn).with( + "sudo redir --lport=#{host_port} --caddr=#{container_ip} --cport=#{guest_port} 2>/dev/null" + ) + end + + it 'skips --laddr parameter if host_ip is a blank string' do + forward_conf[:host_ip] = ' ' + subject.stub(system: true) + subject.call(env) + expect(subject).to have_received(:spawn).with( + "sudo redir --lport=#{host_port} --caddr=#{container_ip} --cport=#{guest_port} 2>/dev/null" + ) + end + end end diff --git a/spec/unit/action/sudo_forward_ports_spec.rb b/spec/unit/action/sudo_forward_ports_spec.rb deleted file mode 100644 index 77602ae..0000000 --- a/spec/unit/action/sudo_forward_ports_spec.rb +++ /dev/null @@ -1,77 +0,0 @@ -require 'unit_helper' - -require 'tmpdir' -require 'vagrant-lxc/provider' -require 'vagrant-lxc/action/forward_ports' - -describe Vagrant::LXC::Action::ForwardPorts do - let(:app) { double(:app, call: true) } - let(:env) { {machine: machine, ui: double(info: true, warn: true)} } - let(:machine) { double(:machine) } - let!(:data_dir) { Pathname.new(Dir.mktmpdir) } - let(:provider) { double(Vagrant::LXC::Provider, ssh_info: {host: container_ip}) } - let(:host_ip) { '127.0.0.1' } - let(:host_port) { 80 } - let(:guest_port) { 80 } - let(:container_ip) { '10.0.1.234' } - let(:pid) { 'a-pid' } - let(:forward_conf) { {guest: guest_port, host: host_port, host_ip: host_ip} } - let(:networks) { [[:other_config, {}], [:forwarded_port, forward_conf]] } - - subject { described_class.new(app, env) } - - before do - machine.stub_chain(:config, :vm, :networks).and_return(networks) - machine.stub(provider: provider, data_dir: data_dir) - - subject.stub(exec: true) - subject.stub(spawn: pid) - end - - after { FileUtils.rm_rf data_dir.to_s } - - it 'forwards ports using redir' do - subject.stub(system: true) - subject.call(env) - expect(subject).to have_received(:spawn).with( - "sudo redir --laddr=#{host_ip} --lport=#{host_port} --caddr=#{container_ip} --cport=#{guest_port} 2>/dev/null" - ) - end - - it 'skips --laddr parameter if host_ip is nil' do - forward_conf.delete(:host_ip) - subject.stub(system: true) - subject.call(env) - expect(subject).to have_received(:spawn).with( - "sudo redir --lport=#{host_port} --caddr=#{container_ip} --cport=#{guest_port} 2>/dev/null" - ) - end - - it 'skips --laddr parameter if host_ip is a blank string' do - forward_conf[:host_ip] = ' ' - subject.stub(system: true) - subject.call(env) - expect(subject).to have_received(:spawn).with( - "sudo redir --lport=#{host_port} --caddr=#{container_ip} --cport=#{guest_port} 2>/dev/null" - ) - end - - it "stores redir pids on machine's data dir" do - subject.stub(system: true) - subject.call(env) - pid_file = data_dir.join('pids', "redir_#{host_port}.pid").read - expect(pid_file).to eq(pid) - end - - it 'allows disabling a previously forwarded port' do - forward_conf[:disabled] = true - subject.stub(system: true) - subject.call(env) - expect(subject).not_to have_received(:spawn) - end - - it 'raises RedirNotInstalled error if `redir` is not installed' do - subject.stub(system: false) - expect { subject.call(env) }.to raise_error(Vagrant::LXC::Errors::RedirNotInstalled) - end -end