command_port_test.go

package vagrant

import (
	"testing"
)

func init() {
	successfulOutput["port"] = `
1534865103,default,metadata,provider,virtualbox
1534865103,,ui,info,The forwarded ports for the machine are listed below. Please note that\nthese values may differ from values configured in the Vagrantfile if the\nprovider supports automatic port collision detection and resolution.
1534865103,,ui,info,
1534865103,,ui,info,    22 (guest) => 2222 (host)
1534865103,,ui,info,    80 (guest) => 8080 (host)
1534865103,default,forwarded_port,22,2222
1534865103,default,forwarded_port,80,8080
`
}

func TestPortCommand_Run(t *testing.T) {
	client := newMockVagrantClient()
	cmd := client.Port()
	if err := cmd.Run(); err != nil {
		t.Fatalf("Command failed to run: %v", err)
	}
	if cmd.Error != nil {
		t.Fatalf("Command return error: %v", cmd.Error)
	}

	if len(cmd.ForwardedPorts) != 2 {
		t.Fatalf("Expected 2 forwarded port; got %v", len(cmd.ForwardedPorts))
	}
	if cmd.ForwardedPorts[0].Guest != 22 || cmd.ForwardedPorts[0].Host != 2222 {
		t.Errorf("Expected guest port 22 -> host 2222; got %v", cmd.ForwardedPorts[0])
	}
	if cmd.ForwardedPorts[1].Guest != 80 || cmd.ForwardedPorts[1].Host != 8080 {
		t.Errorf("Expected guest port 80 -> host 8080; got %v", cmd.ForwardedPorts[1])
	}
}