115 lines
3.1 KiB
Go
115 lines
3.1 KiB
Go
package iso
|
|
|
|
import (
|
|
"fmt"
|
|
"net"
|
|
"testing"
|
|
|
|
"github.com/mitchellh/multistep"
|
|
vmwcommon "github.com/mitchellh/packer/builder/vmware/common"
|
|
)
|
|
|
|
func TestESX5Driver_implDriver(t *testing.T) {
|
|
var _ vmwcommon.Driver = new(ESX5Driver)
|
|
}
|
|
|
|
func TestESX5Driver_UpdateVMX(t *testing.T) {
|
|
var driver ESX5Driver
|
|
data := make(map[string]string)
|
|
driver.UpdateVMX("0.0.0.0", "", 5900, data)
|
|
if _, ok := data["remotedisplay.vnc.ip"]; ok {
|
|
// Do not add the remotedisplay.vnc.ip on ESXi
|
|
t.Fatal("invalid VMX data key: remotedisplay.vnc.ip")
|
|
}
|
|
if enabled := data["remotedisplay.vnc.enabled"]; enabled != "TRUE" {
|
|
t.Errorf("bad VMX data for key remotedisplay.vnc.enabled: %v", enabled)
|
|
}
|
|
if port := data["remotedisplay.vnc.port"]; port != fmt.Sprint(port) {
|
|
t.Errorf("bad VMX data for key remotedisplay.vnc.port: %v", port)
|
|
}
|
|
}
|
|
|
|
func TestESX5Driver_implOutputDir(t *testing.T) {
|
|
var _ vmwcommon.OutputDir = new(ESX5Driver)
|
|
}
|
|
|
|
func TestESX5Driver_implVNCAddressFinder(t *testing.T) {
|
|
var _ vmwcommon.VNCAddressFinder = new(ESX5Driver)
|
|
}
|
|
|
|
func TestESX5Driver_implRemoteDriver(t *testing.T) {
|
|
var _ RemoteDriver = new(ESX5Driver)
|
|
}
|
|
|
|
func TestESX5Driver_HostIP(t *testing.T) {
|
|
expected_host := "127.0.0.1"
|
|
|
|
//create mock SSH server
|
|
listen, _ := net.Listen("tcp", fmt.Sprintf("%s:0", expected_host))
|
|
port := listen.Addr().(*net.TCPAddr).Port
|
|
defer listen.Close()
|
|
|
|
driver := ESX5Driver{Host: "localhost", Port: uint(port)}
|
|
|
|
if host, _ := driver.HostIP(); host != expected_host {
|
|
t.Error(fmt.Sprintf("Expected string, %s but got %s", expected_host, host))
|
|
}
|
|
}
|
|
|
|
func TestESX5Driver_CommHost(t *testing.T) {
|
|
const expected_host = "127.0.0.1"
|
|
|
|
config := testConfig()
|
|
config["communicator"] = "winrm"
|
|
config["winrm_username"] = "username"
|
|
config["winrm_password"] = "password"
|
|
config["winrm_host"] = expected_host
|
|
|
|
var b Builder
|
|
warns, err := b.Prepare(config)
|
|
if len(warns) > 0 {
|
|
t.Fatalf("bad: %#v", warns)
|
|
}
|
|
if err != nil {
|
|
t.Fatalf("should not have error: %s", err)
|
|
}
|
|
if host := b.config.CommConfig.Host(); host != expected_host {
|
|
t.Fatalf("setup failed, bad host name: %s", host)
|
|
}
|
|
|
|
state := new(multistep.BasicStateBag)
|
|
state.Put("config", &b.config)
|
|
|
|
var driver ESX5Driver
|
|
host, err := driver.CommHost(state)
|
|
if err != nil {
|
|
t.Fatalf("should not have error: %s", err)
|
|
}
|
|
if host != expected_host {
|
|
t.Errorf("bad host name: %s", host)
|
|
}
|
|
address, ok := state.GetOk("vm_address")
|
|
if !ok {
|
|
t.Error("state not updated with vm_address")
|
|
}
|
|
if address.(string) != expected_host {
|
|
t.Errorf("bad vm_address: %s", address.(string))
|
|
}
|
|
}
|
|
|
|
func TestESX5Driver_UpdateVMX(t *testing.T) {
|
|
var driver ESX5Driver
|
|
data := make(map[string]string)
|
|
driver.UpdateVMX("0.0.0.0", 5900, data)
|
|
if _, ok := data["remotedisplay.vnc.ip"]; ok {
|
|
// Do not add the remotedisplay.vnc.ip on ESXi
|
|
t.Fatal("invalid VMX data key: remotedisplay.vnc.ip")
|
|
}
|
|
if enabled := data["remotedisplay.vnc.enabled"]; enabled != "TRUE" {
|
|
t.Errorf("bad VMX data for key remotedisplay.vnc.enabled: %v", enabled)
|
|
}
|
|
if port := data["remotedisplay.vnc.port"]; port != fmt.Sprint(port) {
|
|
t.Errorf("bad VMX data for key remotedisplay.vnc.port: %v", port)
|
|
}
|
|
}
|