131 lines
3.2 KiB
Go
131 lines
3.2 KiB
Go
package vsphere
|
|
|
|
import (
|
|
"fmt"
|
|
"net/url"
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
func getTestConfig() Config {
|
|
return Config{
|
|
Username: "me",
|
|
Password: "notpassword",
|
|
Host: "myhost",
|
|
Datacenter: "mydc",
|
|
Cluster: "mycluster",
|
|
VMName: "my vm",
|
|
Datastore: "my datastore",
|
|
Insecure: true,
|
|
DiskMode: "thin",
|
|
VMFolder: "my folder",
|
|
}
|
|
}
|
|
|
|
func TestArgs(t *testing.T) {
|
|
var p PostProcessor
|
|
|
|
p.config = getTestConfig()
|
|
|
|
source := "something.vmx"
|
|
ovftool_uri := fmt.Sprintf("vi://%s:%s@%s/%s/host/%s",
|
|
url.QueryEscape(p.config.Username),
|
|
url.QueryEscape(p.config.Password),
|
|
p.config.Host,
|
|
p.config.Datacenter,
|
|
p.config.Cluster)
|
|
|
|
if p.config.ResourcePool != "" {
|
|
ovftool_uri += "/Resources/" + p.config.ResourcePool
|
|
}
|
|
|
|
args, err := p.BuildArgs(source, ovftool_uri)
|
|
if err != nil {
|
|
t.Errorf("Error: %s", err)
|
|
}
|
|
|
|
t.Logf("ovftool %s", strings.Join(args, " "))
|
|
}
|
|
|
|
func TestGenerateURI_Basic(t *testing.T) {
|
|
var p PostProcessor
|
|
|
|
p.config = getTestConfig()
|
|
|
|
uri, err := p.generateURI()
|
|
if err != nil {
|
|
t.Fatalf("had error: %s", err)
|
|
}
|
|
expected_uri := "vi://me:notpassword@myhost/mydc/host/mycluster"
|
|
if uri.String() != expected_uri {
|
|
t.Fatalf("URI did not match. Recieved: %s. Expected: %s", uri, expected_uri)
|
|
}
|
|
}
|
|
|
|
func TestGenerateURI_PasswordEscapes(t *testing.T) {
|
|
type escapeCases struct {
|
|
Input string
|
|
Expected string
|
|
}
|
|
|
|
cases := []escapeCases{
|
|
{`this has spaces`, `this%20has%20spaces`},
|
|
{`exclaimation_!`, `exclaimation_%21`},
|
|
{`hash_#_dollar_$`, `hash_%23_dollar_$`},
|
|
{`ampersand_&awesome`, `ampersand_&awesome`},
|
|
{`single_quote_'_and_another_'`, `single_quote_%27_and_another_%27`},
|
|
{`open_paren_(_close_paren_)`, `open_paren_%28_close_paren_%29`},
|
|
{`asterisk_*_plus_+`, `asterisk_%2A_plus_+`},
|
|
{`comma_,slash_/`, `comma_,slash_%2F`},
|
|
{`colon_:semicolon_;`, `colon_%3Asemicolon_;`},
|
|
{`equal_=question_?`, `equal_=question_%3F`},
|
|
{`at_@`, `at_%40`},
|
|
{`open_bracket_[closed_bracket]`, `open_bracket_%5Bclosed_bracket%5D`},
|
|
{`user:password with $paces@host/name.foo`, `user%3Apassword%20with%20$paces%40host%2Fname.foo`},
|
|
}
|
|
|
|
for _, escapeCase := range cases {
|
|
var p PostProcessor
|
|
|
|
p.config = getTestConfig()
|
|
p.config.Password = escapeCase.Input
|
|
|
|
uri, err := p.generateURI()
|
|
if err != nil {
|
|
t.Fatalf("had error: %s", err)
|
|
}
|
|
expected_uri := fmt.Sprintf("vi://me:%s@myhost/mydc/host/mycluster", escapeCase.Expected)
|
|
|
|
if uri.String() != expected_uri {
|
|
t.Fatalf("URI did not match. Recieved: %s. Expected: %s", uri, expected_uri)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestGetEncodedPassword(t *testing.T) {
|
|
|
|
// Password is encoded, and contains a colon
|
|
ovftool_uri := fmt.Sprintf("vi://hostname/Datacenter/host/cluster")
|
|
|
|
u, _ := url.Parse(ovftool_uri)
|
|
u.User = url.UserPassword("us:ername", "P@ssW:rd")
|
|
|
|
encoded, isSet := getEncodedPassword(u)
|
|
expected := "P%40ssW%3Ard"
|
|
if !isSet {
|
|
t.Fatalf("Password is set but test said it is not")
|
|
}
|
|
if encoded != expected {
|
|
t.Fatalf("Should have successfully gotten encoded password. Expected: %s; recieved: %s", expected, encoded)
|
|
}
|
|
|
|
// There is no password
|
|
u.User = url.UserPassword("us:ername", "")
|
|
|
|
_, isSet = getEncodedPassword(u)
|
|
if isSet {
|
|
t.Fatalf("Should have determined that password was not set")
|
|
}
|
|
|
|
}
|