94 lines
2.0 KiB
Go
94 lines
2.0 KiB
Go
package driver
|
|
|
|
import (
|
|
"fmt"
|
|
"io/ioutil"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func TestDatastoreAcc(t *testing.T) {
|
|
d := newTestDriver(t)
|
|
ds, err := d.FindDatastore("datastore1", "")
|
|
if err != nil {
|
|
t.Fatalf("Cannot find the default datastore '%v': %v", "datastore1", err)
|
|
}
|
|
info, err := ds.Info("name")
|
|
if err != nil {
|
|
t.Fatalf("Cannot read datastore properties: %v", err)
|
|
}
|
|
if info.Name != "datastore1" {
|
|
t.Errorf("Wrong datastore. expected: 'datastore1', got: '%v'", info.Name)
|
|
}
|
|
}
|
|
|
|
func TestFileUpload(t *testing.T) {
|
|
dsName := "datastore1"
|
|
hostName := "esxi-1.vsphere65.test"
|
|
|
|
fileName := fmt.Sprintf("test-%v", time.Now().Unix())
|
|
tmpFile, err := ioutil.TempFile("", fileName)
|
|
if err != nil {
|
|
t.Fatalf("Error creating temp file")
|
|
}
|
|
err = tmpFile.Close()
|
|
if err != nil {
|
|
t.Fatalf("Error creating temp file")
|
|
}
|
|
|
|
d := newTestDriver(t)
|
|
ds, err := d.FindDatastore(dsName, hostName)
|
|
if err != nil {
|
|
t.Fatalf("Cannot find datastore '%v': %v", dsName, err)
|
|
}
|
|
|
|
err = ds.UploadFile(tmpFile.Name(), fileName, hostName)
|
|
if err != nil {
|
|
t.Fatalf("Cannot upload file: %v", err)
|
|
}
|
|
|
|
if ds.FileExists(fileName) != true {
|
|
t.Fatalf("Cannot find file")
|
|
}
|
|
|
|
err = ds.Delete(fileName)
|
|
if err != nil {
|
|
t.Fatalf("Cannot delete file: %v", err)
|
|
}
|
|
}
|
|
|
|
func TestFileUploadDRS(t *testing.T) {
|
|
dsName := "datastore3"
|
|
hostName := ""
|
|
|
|
fileName := fmt.Sprintf("test-%v", time.Now().Unix())
|
|
tmpFile, err := ioutil.TempFile("", fileName)
|
|
if err != nil {
|
|
t.Fatalf("Error creating temp file")
|
|
}
|
|
err = tmpFile.Close()
|
|
if err != nil {
|
|
t.Fatalf("Error creating temp file")
|
|
}
|
|
|
|
d := newTestDriver(t)
|
|
ds, err := d.FindDatastore(dsName, hostName)
|
|
if err != nil {
|
|
t.Fatalf("Cannot find datastore '%v': %v", dsName, err)
|
|
}
|
|
|
|
err = ds.UploadFile(tmpFile.Name(), fileName, hostName)
|
|
if err != nil {
|
|
t.Fatalf("Cannot upload file: %v", err)
|
|
}
|
|
|
|
if ds.FileExists(fileName) != true {
|
|
t.Fatalf("Cannot find file")
|
|
}
|
|
|
|
err = ds.Delete(fileName)
|
|
if err != nil {
|
|
t.Fatalf("Cannot delete file: %v", err)
|
|
}
|
|
}
|