mirror of
https://github.com/kata-containers/kata-containers.git
synced 2025-06-26 07:22:20 +00:00
clh: basic/unit tests for clh driver
- added clh unit tests - removed some inconsistencies in the cli builder to enable unit tests - suppressed version check for in startSandbox to enable unit tests - added clh related constants and methods to virtcontainer test - small corrections after review applied Fixes: #2205 Signed-off-by: Johan Kuijpers <johan.kuijpers@ericsson.com>
This commit is contained in:
Johan Kuijpers
parent
3ef8f6cf1b
commit
e8cc87b378
@ -52,6 +52,7 @@ const (
|
||||
supportedMinorVersion = 3
|
||||
defaultClhPath = "/usr/local/bin/cloud-hypervisor"
|
||||
virtioFsCacheAlways = "always"
|
||||
maxClhVcpus = uint32(64)
|
||||
)
|
||||
|
||||
type CloudHypervisorVersion struct {
|
||||
@ -126,6 +127,12 @@ func (clh *cloudHypervisor) createSandbox(ctx context.Context, id string, networ
|
||||
clh.config = *hypervisorConfig
|
||||
clh.state.state = clhNotReady
|
||||
|
||||
// version check only applicable to 'cloud-hypervisor' executable
|
||||
clhPath, perr := clh.clhPath()
|
||||
if perr != nil {
|
||||
return perr
|
||||
}
|
||||
if strings.HasSuffix(clhPath, "cloud-hypervisor") {
|
||||
err = clh.getAvailableVersion()
|
||||
if err != nil {
|
||||
return err
|
||||
@ -139,7 +146,7 @@ func (clh *cloudHypervisor) createSandbox(ctx context.Context, id string, networ
|
||||
supportedMinorVersion)
|
||||
return errors.New(errorMessage)
|
||||
}
|
||||
|
||||
}
|
||||
clh.cliBuilder = &DefaultCLIBuilder{}
|
||||
|
||||
socketPath, err := clh.vsockSocketPath(id)
|
||||
@ -411,12 +418,6 @@ func (clh *cloudHypervisor) getPids() []int {
|
||||
return pids
|
||||
}
|
||||
|
||||
//###########################################################################
|
||||
//
|
||||
// Local helper methods related to the hypervisor interface implementation
|
||||
//
|
||||
//###########################################################################
|
||||
|
||||
func (clh *cloudHypervisor) addDevice(devInfo interface{}, devType deviceType) error {
|
||||
span, _ := clh.trace("addDevice")
|
||||
defer span.Finish()
|
||||
@ -449,6 +450,12 @@ func (clh *cloudHypervisor) addDevice(devInfo interface{}, devType deviceType) e
|
||||
return err
|
||||
}
|
||||
|
||||
//###########################################################################
|
||||
//
|
||||
// Local helper methods related to the hypervisor interface implementation
|
||||
//
|
||||
//###########################################################################
|
||||
|
||||
func (clh *cloudHypervisor) Logger() *log.Entry {
|
||||
return virtLog.WithField("subsystem", "cloudHypervisor")
|
||||
}
|
||||
@ -856,6 +863,11 @@ func (clh *cloudHypervisor) LaunchClh() (string, int, error) {
|
||||
return errStr, cmd.Process.Pid, nil
|
||||
}
|
||||
|
||||
// MaxClhVCPUs returns the maximum number of vCPUs supported
|
||||
func MaxClhVCPUs() uint32 {
|
||||
return maxClhVcpus
|
||||
}
|
||||
|
||||
//###########################################################################
|
||||
//
|
||||
// Cloud-hypervisor CLI builder
|
||||
@ -865,6 +877,8 @@ func (clh *cloudHypervisor) LaunchClh() (string, int, error) {
|
||||
const (
|
||||
cctOFF string = "off"
|
||||
cctFILE string = "file"
|
||||
cctNULL string = "null"
|
||||
cctTTY string = "tty"
|
||||
)
|
||||
|
||||
const (
|
||||
@ -1083,6 +1097,7 @@ type CLIFs struct {
|
||||
queues uint32
|
||||
queueSize uint32
|
||||
dax bool
|
||||
cacheSize string
|
||||
}
|
||||
|
||||
func (o *CLIFs) Build(cmdline *CommandLine) {
|
||||
@ -1092,6 +1107,9 @@ func (o *CLIFs) Build(cmdline *CommandLine) {
|
||||
fsarg := "tag=" + o.tag + ",sock=" + o.socketPath
|
||||
if o.dax {
|
||||
fsarg += ",dax=on"
|
||||
if o.cacheSize != "" {
|
||||
fsarg += ",cache_size=" + o.cacheSize
|
||||
}
|
||||
} else {
|
||||
fsarg += ",num_queues=" + strconv.FormatUint(uint64(o.queues), 10) + ",queue_size=" + strconv.FormatUint(uint64(o.queueSize), 10)
|
||||
}
|
||||
@ -1103,6 +1121,8 @@ func (o *CLIFs) Build(cmdline *CommandLine) {
|
||||
//****************************************
|
||||
type CLINet struct {
|
||||
device string
|
||||
ip string
|
||||
mask string
|
||||
mac string
|
||||
iommu bool
|
||||
}
|
||||
@ -1118,13 +1138,23 @@ func (o *CLINets) Build(cmdline *CommandLine) {
|
||||
networks := ""
|
||||
netIndex := 1
|
||||
for _, net := range o.networks {
|
||||
tapName := "tap" + strconv.FormatUint(uint64(netIndex), 10)
|
||||
netIndex++
|
||||
if net.iommu {
|
||||
networks += "tap=" + tapName + ",mac=" + net.mac + ",iommu=on"
|
||||
} else {
|
||||
networks += "tap=" + tapName + ",mac=" + net.mac
|
||||
cnet := "tap=tap" + strconv.FormatUint(uint64(netIndex), 10)
|
||||
if net.ip != "" && net.mask != "" {
|
||||
cnet += ",ip=" + net.ip + ",mask=" + net.mask
|
||||
}
|
||||
if net.mac != "" {
|
||||
cnet += ",mac=" + net.mac
|
||||
}
|
||||
|
||||
if net.iommu {
|
||||
cnet += ",iommu=on"
|
||||
}
|
||||
if netIndex > 1 {
|
||||
networks += "," + cnet
|
||||
} else {
|
||||
networks += cnet
|
||||
}
|
||||
netIndex++
|
||||
}
|
||||
cmdline.args = append(cmdline.args, networks)
|
||||
}
|
||||
@ -1259,6 +1289,10 @@ func (d *DefaultCLIBuilder) GetCommandLine() (*CommandLine, error) {
|
||||
d.console.Build(cmdLine)
|
||||
}
|
||||
|
||||
if d.apiSocket != nil {
|
||||
d.apiSocket.Build(cmdLine)
|
||||
}
|
||||
|
||||
if d.logFile != nil {
|
||||
d.logFile.Build(cmdLine)
|
||||
}
|
||||
|
||||
691
virtcontainers/clh_test.go
Normal file
691
virtcontainers/clh_test.go
Normal file
@ -0,0 +1,691 @@
|
||||
// Copyright (c) 2019 Ericsson Eurolab Deutschland G.m.b.H.
|
||||
//
|
||||
// SPDX-License-Identifier: Apache-2.0
|
||||
//
|
||||
|
||||
package virtcontainers
|
||||
|
||||
import (
|
||||
"context"
|
||||
"errors"
|
||||
"os"
|
||||
"path/filepath"
|
||||
"strconv"
|
||||
"strings"
|
||||
"testing"
|
||||
|
||||
"github.com/kata-containers/runtime/virtcontainers/device/config"
|
||||
"github.com/kata-containers/runtime/virtcontainers/store"
|
||||
"github.com/kata-containers/runtime/virtcontainers/types"
|
||||
"github.com/stretchr/testify/assert"
|
||||
)
|
||||
|
||||
//
|
||||
// Cli helper functions
|
||||
//
|
||||
|
||||
func getCliOption(args []string, key string) (string, error) {
|
||||
for i, p := range args {
|
||||
if p == key && i < (len(args)-1) {
|
||||
return args[i+1], nil
|
||||
}
|
||||
}
|
||||
return "", errors.New("Key not found")
|
||||
}
|
||||
|
||||
func newClhConfig() HypervisorConfig {
|
||||
return HypervisorConfig{
|
||||
KernelPath: testClhKernelPath,
|
||||
ImagePath: testClhImagePath,
|
||||
HypervisorPath: testClhPath,
|
||||
NumVCPUs: defaultVCPUs,
|
||||
BlockDeviceDriver: config.VirtioBlock,
|
||||
MemorySize: defaultMemSzMiB,
|
||||
DefaultBridges: defaultBridges,
|
||||
DefaultMaxVCPUs: MaxClhVCPUs(),
|
||||
// Adding this here, as hypervisorconfig.valid()
|
||||
// forcefully adds it even when 9pfs is not supported
|
||||
Msize9p: defaultMsize9p,
|
||||
VirtioFSCache: virtioFsCacheAlways,
|
||||
}
|
||||
}
|
||||
|
||||
//
|
||||
// --cmdline <cmdline> Kernel command line
|
||||
//
|
||||
func TestClhCliKernelParameters(t *testing.T) {
|
||||
assert := assert.New(t)
|
||||
|
||||
expectedOut := "--cmdline foo=foo bar=bar"
|
||||
params := []Param{
|
||||
{
|
||||
Key: "foo",
|
||||
Value: "foo",
|
||||
},
|
||||
{
|
||||
Key: "bar",
|
||||
Value: "bar",
|
||||
},
|
||||
}
|
||||
|
||||
builder := &DefaultCLIBuilder{}
|
||||
builder.AddKernelParameters(params)
|
||||
director := &CommandLineDirector{}
|
||||
|
||||
cli, err := director.Build(builder)
|
||||
assert.NotNil(cli.args)
|
||||
assert.Nil(err)
|
||||
|
||||
assert.Equal(strings.TrimSpace(strings.Join(cli.args, " ")), expectedOut)
|
||||
}
|
||||
|
||||
//
|
||||
// Control (virtio) console: <console> off|null|tty|file=/path/to/a/file,iommu=on|off [default: tty]
|
||||
//
|
||||
func TestClhCliConsole(t *testing.T) {
|
||||
assert := assert.New(t)
|
||||
builder := &DefaultCLIBuilder{}
|
||||
director := &CommandLineDirector{}
|
||||
defaultFilePath := "/a/b/c"
|
||||
|
||||
builder.SetConsole(&CLIConsole{
|
||||
consoleType: cctOFF,
|
||||
})
|
||||
cli, err := director.Build(builder)
|
||||
assert.NotNil(cli.args)
|
||||
assert.Nil(err)
|
||||
assert.Equal(strings.TrimSpace(strings.Join(cli.args, " ")), "--console off")
|
||||
|
||||
builder.SetConsole(&CLIConsole{
|
||||
consoleType: cctNULL,
|
||||
})
|
||||
cli, err = director.Build(builder)
|
||||
assert.NotNil(cli.args)
|
||||
assert.Nil(err)
|
||||
assert.Equal(strings.TrimSpace(strings.Join(cli.args, " ")), "--console null")
|
||||
|
||||
builder.SetConsole(&CLIConsole{
|
||||
consoleType: cctTTY,
|
||||
})
|
||||
cli, err = director.Build(builder)
|
||||
assert.NotNil(cli.args)
|
||||
assert.Nil(err)
|
||||
assert.Equal(strings.TrimSpace(strings.Join(cli.args, " ")), "--console tty")
|
||||
|
||||
builder.SetConsole(&CLIConsole{
|
||||
consoleType: cctFILE,
|
||||
filePath: defaultFilePath,
|
||||
})
|
||||
cli, err = director.Build(builder)
|
||||
assert.NotNil(cli.args)
|
||||
assert.Nil(err)
|
||||
assert.Equal(strings.TrimSpace(strings.Join(cli.args, " ")), "--console file="+defaultFilePath+",iommu=off")
|
||||
|
||||
builder.SetConsole(&CLIConsole{
|
||||
consoleType: cctFILE,
|
||||
filePath: defaultFilePath,
|
||||
iommu: true,
|
||||
})
|
||||
cli, err = director.Build(builder)
|
||||
assert.NotNil(cli.args)
|
||||
assert.Nil(err)
|
||||
assert.Equal(strings.TrimSpace(strings.Join(cli.args, " ")), "--console file="+defaultFilePath+",iommu=on")
|
||||
|
||||
}
|
||||
|
||||
//
|
||||
// Control serial port: off|null|tty|file=/path/to/a/file [default: tty]
|
||||
//
|
||||
func TestClhCliSerial(t *testing.T) {
|
||||
assert := assert.New(t)
|
||||
builder := &DefaultCLIBuilder{}
|
||||
director := &CommandLineDirector{}
|
||||
defaultFilePath := "/a/b/c"
|
||||
|
||||
builder.SetSerial(&CLISerialConsole{
|
||||
consoleType: cctOFF,
|
||||
})
|
||||
cli, err := director.Build(builder)
|
||||
assert.NotNil(cli.args)
|
||||
assert.Nil(err)
|
||||
assert.Equal(strings.TrimSpace(strings.Join(cli.args, " ")), "--serial off")
|
||||
|
||||
builder.SetSerial(&CLISerialConsole{
|
||||
consoleType: cctNULL,
|
||||
})
|
||||
cli, err = director.Build(builder)
|
||||
assert.NotNil(cli.args)
|
||||
assert.Nil(err)
|
||||
assert.Equal(strings.TrimSpace(strings.Join(cli.args, " ")), "--serial null")
|
||||
|
||||
builder.SetSerial(&CLISerialConsole{
|
||||
consoleType: cctTTY,
|
||||
})
|
||||
cli, err = director.Build(builder)
|
||||
assert.NotNil(cli.args)
|
||||
assert.Nil(err)
|
||||
assert.Equal(strings.TrimSpace(strings.Join(cli.args, " ")), "--serial tty")
|
||||
|
||||
builder.SetSerial(&CLISerialConsole{
|
||||
consoleType: cctFILE,
|
||||
filePath: defaultFilePath,
|
||||
})
|
||||
cli, err = director.Build(builder)
|
||||
assert.NotNil(cli.args)
|
||||
assert.Nil(err)
|
||||
assert.Equal(strings.TrimSpace(strings.Join(cli.args, " ")), "--serial file="+defaultFilePath)
|
||||
|
||||
}
|
||||
|
||||
//
|
||||
// --api-socket <api-socket> HTTP API socket path (UNIX domain socket). [default: /run/cloud-hypervisor.23605
|
||||
//
|
||||
func TestClhCliApiSocket(t *testing.T) {
|
||||
assert := assert.New(t)
|
||||
builder := &DefaultCLIBuilder{}
|
||||
director := &CommandLineDirector{}
|
||||
defaultFilePath := "/a/b/c"
|
||||
|
||||
builder.SetAPISocket(&CLIAPISocket{
|
||||
socketPath: defaultFilePath,
|
||||
})
|
||||
cli, err := director.Build(builder)
|
||||
assert.NotNil(cli.args)
|
||||
assert.Nil(err)
|
||||
assert.Equal(strings.TrimSpace(strings.Join(cli.args, " ")), "--api-socket "+defaultFilePath)
|
||||
}
|
||||
|
||||
//
|
||||
// --cpus <cpus> Number of virtual CPUs [default: 1]
|
||||
//
|
||||
func TestClhCliCpus(t *testing.T) {
|
||||
assert := assert.New(t)
|
||||
builder := &DefaultCLIBuilder{}
|
||||
director := &CommandLineDirector{}
|
||||
defaultCPUs := "4"
|
||||
|
||||
builder.SetCpus(&CLICpus{
|
||||
cpus: 4,
|
||||
})
|
||||
cli, err := director.Build(builder)
|
||||
assert.NotNil(cli.args)
|
||||
assert.Nil(err)
|
||||
assert.Equal(strings.TrimSpace(strings.Join(cli.args, " ")), "--cpus "+defaultCPUs)
|
||||
}
|
||||
|
||||
//
|
||||
// --disk <disk> Disk parameters "path=<disk_image_path>,iommu=on|off"
|
||||
//
|
||||
func TestClhCliDisk(t *testing.T) {
|
||||
assert := assert.New(t)
|
||||
builder := &DefaultCLIBuilder{}
|
||||
director := &CommandLineDirector{}
|
||||
defaultDiskPath := "/a/b/c.img"
|
||||
|
||||
builder.SetDisk(&CLIDisk{
|
||||
path: defaultDiskPath,
|
||||
})
|
||||
cli, err := director.Build(builder)
|
||||
assert.NotNil(cli.args)
|
||||
assert.Nil(err)
|
||||
assert.Equal(strings.TrimSpace(strings.Join(cli.args, " ")), "--disk path="+defaultDiskPath+",iommu=off")
|
||||
|
||||
builder.SetDisk(&CLIDisk{
|
||||
path: defaultDiskPath,
|
||||
iommu: true,
|
||||
})
|
||||
cli, err = director.Build(builder)
|
||||
assert.NotNil(cli.args)
|
||||
assert.Nil(err)
|
||||
assert.Equal(strings.TrimSpace(strings.Join(cli.args, " ")), "--disk path="+defaultDiskPath+",iommu=on")
|
||||
}
|
||||
|
||||
//
|
||||
// --fs <fs> virtio-fs parameters
|
||||
// "tag=<tag_name>,sock=<socket_path>,num_queues=<number_of_queues>,queue_size=<size_of_each_queue>,dax=on|off,cache_size=<DAX
|
||||
// cache size: default 8Gib>"
|
||||
//
|
||||
func TestClhCliFs(t *testing.T) {
|
||||
assert := assert.New(t)
|
||||
builder := &DefaultCLIBuilder{}
|
||||
director := &CommandLineDirector{}
|
||||
defaultFsPath := "/a/b/c"
|
||||
defaultFsQs := uint32(2)
|
||||
defaultFsQss := uint32(1024)
|
||||
defaultFsTag := "myTag"
|
||||
defaultFsCacheSize := "1Gib"
|
||||
|
||||
builder.SetFs(&CLIFs{
|
||||
tag: defaultFsTag,
|
||||
socketPath: defaultFsPath,
|
||||
queues: defaultFsQs,
|
||||
queueSize: defaultFsQss,
|
||||
dax: false,
|
||||
})
|
||||
cli, err := director.Build(builder)
|
||||
assert.NotNil(cli.args)
|
||||
assert.Nil(err)
|
||||
assert.Equal(strings.TrimSpace(strings.Join(cli.args, " ")),
|
||||
"--fs tag="+defaultFsTag+
|
||||
",sock="+defaultFsPath+
|
||||
",num_queues="+strconv.FormatUint(uint64(defaultFsQs), 10)+
|
||||
",queue_size="+strconv.FormatUint(uint64(defaultFsQss), 10))
|
||||
|
||||
builder.SetFs(&CLIFs{
|
||||
tag: defaultFsTag,
|
||||
socketPath: defaultFsPath,
|
||||
dax: true,
|
||||
})
|
||||
cli, err = director.Build(builder)
|
||||
assert.NotNil(cli.args)
|
||||
assert.Nil(err)
|
||||
assert.Equal(strings.TrimSpace(strings.Join(cli.args, " ")),
|
||||
"--fs tag="+defaultFsTag+
|
||||
",sock="+defaultFsPath+
|
||||
",dax=on")
|
||||
|
||||
builder.SetFs(&CLIFs{
|
||||
tag: defaultFsTag,
|
||||
socketPath: defaultFsPath,
|
||||
dax: true,
|
||||
cacheSize: defaultFsCacheSize,
|
||||
})
|
||||
cli, err = director.Build(builder)
|
||||
assert.NotNil(cli.args)
|
||||
assert.Nil(err)
|
||||
assert.Equal(strings.TrimSpace(strings.Join(cli.args, " ")),
|
||||
"--fs tag="+defaultFsTag+
|
||||
",sock="+defaultFsPath+
|
||||
",dax=on,cache_size="+defaultFsCacheSize)
|
||||
|
||||
}
|
||||
|
||||
//
|
||||
// --disk <disk> Disk parameters "path=<disk_image_path>,iommu=on|off"
|
||||
//
|
||||
func TestClhCliKernel(t *testing.T) {
|
||||
assert := assert.New(t)
|
||||
builder := &DefaultCLIBuilder{}
|
||||
director := &CommandLineDirector{}
|
||||
defaultKernel := "/a/b/vmlinuz"
|
||||
|
||||
builder.SetKernel(&CLIKernel{
|
||||
path: defaultKernel,
|
||||
})
|
||||
cli, err := director.Build(builder)
|
||||
assert.NotNil(cli.args)
|
||||
assert.Nil(err)
|
||||
assert.Equal(strings.TrimSpace(strings.Join(cli.args, " ")), "--kernel "+defaultKernel)
|
||||
|
||||
}
|
||||
|
||||
//
|
||||
// --log-file <log-file> Log file. Standard error is used if not specified
|
||||
//
|
||||
func TestClhCliLogFile(t *testing.T) {
|
||||
assert := assert.New(t)
|
||||
builder := &DefaultCLIBuilder{}
|
||||
director := &CommandLineDirector{}
|
||||
defaultPath := "/a/b/clh.log"
|
||||
|
||||
builder.SetLogFile(&CLILogFile{
|
||||
path: defaultPath,
|
||||
})
|
||||
cli, err := director.Build(builder)
|
||||
assert.NotNil(cli.args)
|
||||
assert.Nil(err)
|
||||
assert.Equal(strings.TrimSpace(strings.Join(cli.args, " ")), "--log-file "+defaultPath)
|
||||
|
||||
}
|
||||
|
||||
//
|
||||
// --memory <memory> Memory parameters "size=<guest_memory_size>,file=<backing_file_path>"
|
||||
//
|
||||
func TestClhCliMemory(t *testing.T) {
|
||||
assert := assert.New(t)
|
||||
builder := &DefaultCLIBuilder{}
|
||||
director := &CommandLineDirector{}
|
||||
defaultMemory := uint32(1024)
|
||||
defaultFile := "/a/b.shm"
|
||||
|
||||
builder.SetMemory(&CLIMemory{
|
||||
memorySize: defaultMemory,
|
||||
})
|
||||
cli, err := director.Build(builder)
|
||||
assert.NotNil(cli.args)
|
||||
assert.Nil(err)
|
||||
assert.Equal(strings.TrimSpace(strings.Join(cli.args, " ")), "--memory size="+strconv.FormatUint(uint64(defaultMemory), 10)+"M")
|
||||
|
||||
builder.SetMemory(&CLIMemory{
|
||||
memorySize: defaultMemory,
|
||||
backingFile: defaultFile,
|
||||
})
|
||||
cli, err = director.Build(builder)
|
||||
assert.NotNil(cli.args)
|
||||
assert.Nil(err)
|
||||
assert.Equal(strings.TrimSpace(strings.Join(cli.args, " ")),
|
||||
"--memory size="+strconv.FormatUint(uint64(defaultMemory), 10)+"M"+
|
||||
",file="+defaultFile)
|
||||
|
||||
}
|
||||
|
||||
//
|
||||
// --net <net> Network parameters
|
||||
// "tap=<if_name>,ip=<ip_addr>,mask=<net_mask>,mac=<mac_addr>,iommu=on|off"
|
||||
//
|
||||
func TestClhCliNetwork(t *testing.T) {
|
||||
assert := assert.New(t)
|
||||
builder := &DefaultCLIBuilder{}
|
||||
director := &CommandLineDirector{}
|
||||
defaultIfName := "tap1"
|
||||
defaultIP := "1.2.3.4"
|
||||
defaultNetMask := "255.255.255.0"
|
||||
defaultMac := "00:11:22:33:44:55"
|
||||
|
||||
builder.AddNet(CLINet{
|
||||
mac: defaultMac,
|
||||
})
|
||||
cli, err := director.Build(builder)
|
||||
assert.NotNil(cli.args)
|
||||
assert.Nil(err)
|
||||
assert.Equal(strings.TrimSpace(strings.Join(cli.args, " ")), "--net tap=tap1,mac="+defaultMac)
|
||||
|
||||
builder = &DefaultCLIBuilder{}
|
||||
builder.AddNet(CLINet{
|
||||
device: defaultIfName,
|
||||
mac: defaultMac,
|
||||
})
|
||||
|
||||
cli, err = director.Build(builder)
|
||||
assert.NotNil(cli.args)
|
||||
assert.Nil(err)
|
||||
assert.Equal(strings.TrimSpace(strings.Join(cli.args, " ")), "--net tap="+defaultIfName+",mac="+defaultMac)
|
||||
|
||||
builder = &DefaultCLIBuilder{}
|
||||
builder.AddNet(CLINet{
|
||||
device: defaultIfName,
|
||||
mac: defaultMac,
|
||||
ip: defaultIP,
|
||||
mask: defaultNetMask,
|
||||
})
|
||||
|
||||
cli, err = director.Build(builder)
|
||||
assert.NotNil(cli.args)
|
||||
assert.Nil(err)
|
||||
assert.Equal(strings.TrimSpace(strings.Join(cli.args, " ")), "--net tap="+defaultIfName+
|
||||
",ip="+defaultIP+
|
||||
",mask="+defaultNetMask+
|
||||
",mac="+defaultMac)
|
||||
|
||||
builder.AddNet(CLINet{
|
||||
mac: defaultMac,
|
||||
ip: defaultIP,
|
||||
mask: defaultNetMask,
|
||||
})
|
||||
|
||||
cli, err = director.Build(builder)
|
||||
assert.NotNil(cli.args)
|
||||
assert.Nil(err)
|
||||
assert.Equal(strings.TrimSpace(strings.Join(cli.args, " ")),
|
||||
"--net tap="+defaultIfName+
|
||||
",ip="+defaultIP+
|
||||
",mask="+defaultNetMask+
|
||||
",mac="+defaultMac+
|
||||
",tap=tap2"+
|
||||
",ip="+defaultIP+
|
||||
",mask="+defaultNetMask+
|
||||
",mac="+defaultMac)
|
||||
}
|
||||
|
||||
//
|
||||
// --rng <rng> Random number generator parameters "src=<entropy_source_path>,iommu=on|off"
|
||||
//
|
||||
func TestClhCliRng(t *testing.T) {
|
||||
assert := assert.New(t)
|
||||
builder := &DefaultCLIBuilder{}
|
||||
director := &CommandLineDirector{}
|
||||
defaultSource := "/a/b/c.random"
|
||||
|
||||
builder.SetRng(&CLIRng{
|
||||
src: defaultSource,
|
||||
})
|
||||
cli, err := director.Build(builder)
|
||||
assert.NotNil(cli.args)
|
||||
assert.Nil(err)
|
||||
assert.Equal(strings.TrimSpace(strings.Join(cli.args, " ")), "--rng src="+defaultSource+",iommu=off")
|
||||
|
||||
builder.SetRng(&CLIRng{
|
||||
src: defaultSource,
|
||||
iommu: true,
|
||||
})
|
||||
cli, err = director.Build(builder)
|
||||
assert.NotNil(cli.args)
|
||||
assert.Nil(err)
|
||||
assert.Equal(strings.TrimSpace(strings.Join(cli.args, " ")), "--rng src="+defaultSource+",iommu=on")
|
||||
}
|
||||
|
||||
//
|
||||
// --vsock <vsock> Virtio VSOCK parameters "cid=<context_id>,sock=<socket_path>,iommu=on|off"
|
||||
//
|
||||
func TestClhCliVsock(t *testing.T) {
|
||||
assert := assert.New(t)
|
||||
builder := &DefaultCLIBuilder{}
|
||||
director := &CommandLineDirector{}
|
||||
defaultSocket := "/a/b/c.sock"
|
||||
defaultCid := uint32(12345)
|
||||
|
||||
builder.SetVsock(&CLIVsock{
|
||||
cid: defaultCid,
|
||||
socketPath: defaultSocket,
|
||||
})
|
||||
cli, err := director.Build(builder)
|
||||
assert.NotNil(cli.args)
|
||||
assert.Nil(err)
|
||||
assert.Equal(strings.TrimSpace(strings.Join(cli.args, " ")),
|
||||
"--vsock cid="+strconv.FormatUint(uint64(defaultCid), 10)+
|
||||
",sock="+defaultSocket+
|
||||
",iommu=off")
|
||||
|
||||
builder.SetVsock(&CLIVsock{
|
||||
cid: defaultCid,
|
||||
socketPath: defaultSocket,
|
||||
iommu: true,
|
||||
})
|
||||
cli, err = director.Build(builder)
|
||||
assert.NotNil(cli.args)
|
||||
assert.Nil(err)
|
||||
assert.Equal(strings.TrimSpace(strings.Join(cli.args, " ")),
|
||||
"--vsock cid="+strconv.FormatUint(uint64(defaultCid), 10)+
|
||||
",sock="+defaultSocket+
|
||||
",iommu=on")
|
||||
|
||||
}
|
||||
|
||||
func TestClhCreateSandbox(t *testing.T) {
|
||||
clhConfig := newClhConfig()
|
||||
clh := &cloudHypervisor{
|
||||
config: clhConfig,
|
||||
}
|
||||
assert := assert.New(t)
|
||||
|
||||
sandbox := &Sandbox{
|
||||
ctx: context.Background(),
|
||||
id: "testSandbox",
|
||||
config: &SandboxConfig{
|
||||
HypervisorConfig: clhConfig,
|
||||
},
|
||||
}
|
||||
|
||||
vcStore, err := store.NewVCSandboxStore(sandbox.ctx, sandbox.id)
|
||||
assert.NoError(err)
|
||||
|
||||
sandbox.store = vcStore
|
||||
|
||||
// Create the hypervisor fake binary
|
||||
testClhPath := filepath.Join(testDir, testHypervisor)
|
||||
_, err = os.Create(testClhPath)
|
||||
assert.NoError(err)
|
||||
|
||||
// Create parent dir path for hypervisor.json
|
||||
parentDir := store.SandboxConfigurationRootPath(sandbox.id)
|
||||
assert.NoError(os.MkdirAll(parentDir, store.DirMode))
|
||||
|
||||
err = clh.createSandbox(context.Background(), sandbox.id, NetworkNamespace{}, &sandbox.config.HypervisorConfig, sandbox.store, false)
|
||||
assert.NoError(err)
|
||||
assert.NoError(os.RemoveAll(parentDir))
|
||||
assert.Exactly(clhConfig, clh.config)
|
||||
}
|
||||
|
||||
func TestClhAddDeviceNet(t *testing.T) {
|
||||
defaultEndpointName := "tap1"
|
||||
defaultMac := "55:44:33:22:11:00"
|
||||
assert := assert.New(t)
|
||||
clh := &cloudHypervisor{
|
||||
ctx: context.Background(),
|
||||
cliBuilder: &DefaultCLIBuilder{},
|
||||
}
|
||||
|
||||
tep, _ := createTapNetworkEndpoint(0, defaultEndpointName)
|
||||
tep.TapInterface.TAPIface.HardAddr = defaultMac
|
||||
|
||||
err := clh.addDevice(Endpoint(tep), netDev)
|
||||
assert.NoError(err)
|
||||
|
||||
director := &CommandLineDirector{}
|
||||
|
||||
cli, err := director.Build(clh.cliBuilder)
|
||||
assert.NoError(err)
|
||||
assert.NotNil(cli)
|
||||
|
||||
netarg, err := getCliOption(cli.args, "--net")
|
||||
assert.NoError(err)
|
||||
assert.Equal(netarg, "tap="+defaultEndpointName+",mac="+defaultMac)
|
||||
|
||||
}
|
||||
|
||||
func TestClhAddDeviceVSock(t *testing.T) {
|
||||
defaultCid := uint64(12345)
|
||||
defaultUdsPath := "/a/b/c"
|
||||
defaultPort := uint32(1024)
|
||||
assert := assert.New(t)
|
||||
clh := &cloudHypervisor{
|
||||
ctx: context.Background(),
|
||||
cliBuilder: &DefaultCLIBuilder{},
|
||||
}
|
||||
|
||||
vsock := types.HybridVSock{
|
||||
UdsPath: defaultUdsPath,
|
||||
ContextID: defaultCid,
|
||||
Port: defaultPort,
|
||||
}
|
||||
|
||||
err := clh.addDevice(vsock, netDev)
|
||||
assert.NoError(err)
|
||||
|
||||
director := &CommandLineDirector{}
|
||||
|
||||
cli, err := director.Build(clh.cliBuilder)
|
||||
assert.NoError(err)
|
||||
assert.NotNil(cli)
|
||||
|
||||
netarg, err := getCliOption(cli.args, "--vsock")
|
||||
assert.NoError(err)
|
||||
assert.Equal(netarg, "cid="+strconv.FormatUint(defaultCid, 10)+",sock="+defaultUdsPath+",iommu=off")
|
||||
|
||||
}
|
||||
|
||||
func TestClhCapabilities(t *testing.T) {
|
||||
assert := assert.New(t)
|
||||
clh := &cloudHypervisor{
|
||||
ctx: context.Background(),
|
||||
cliBuilder: &DefaultCLIBuilder{},
|
||||
}
|
||||
|
||||
caps := clh.capabilities()
|
||||
|
||||
assert.False(caps.IsBlockDeviceHotplugSupported())
|
||||
assert.True(caps.IsFsSharingSupported())
|
||||
assert.False(caps.IsMultiQueueSupported())
|
||||
}
|
||||
|
||||
func TestClhGenerateSocket(t *testing.T) {
|
||||
defaultID := "123-456-99"
|
||||
defaultPort := uint32(1024)
|
||||
|
||||
assert := assert.New(t)
|
||||
clh := &cloudHypervisor{
|
||||
ctx: context.Background(),
|
||||
cliBuilder: &DefaultCLIBuilder{},
|
||||
}
|
||||
|
||||
rtnval, err := clh.generateSocket(defaultID, true)
|
||||
assert.NoError(err)
|
||||
assert.NotNil(rtnval)
|
||||
|
||||
if socket, ok := rtnval.(types.HybridVSock); ok {
|
||||
assert.Equal(socket.UdsPath, "/run/vc/vm/"+defaultID+"/"+clhSocket)
|
||||
assert.Equal(socket.Port, defaultPort)
|
||||
assert.NotEqual(socket.ContextID, 0, "ContextID 0 is reserved for the hypervisor communication")
|
||||
assert.NotEqual(socket.ContextID, 1, "ContextID 1 is reserved")
|
||||
assert.NotEqual(socket.ContextID, 2, "ContextID 2 is reserved for the host communication")
|
||||
assert.NotEqual(socket.ContextID, 0xffffffff, "ContextID 0xffffffff is reserved")
|
||||
} else {
|
||||
t.Fail()
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
func TestClhReset(t *testing.T) {
|
||||
assert := assert.New(t)
|
||||
clh := &cloudHypervisor{
|
||||
ctx: context.Background(),
|
||||
cliBuilder: &DefaultCLIBuilder{},
|
||||
}
|
||||
|
||||
clh.state.PID = 10
|
||||
clh.state.VirtiofsdPID = 11
|
||||
clh.state.state = clhReady
|
||||
|
||||
clh.reset()
|
||||
|
||||
assert.Equal(clh.state.PID, 0)
|
||||
assert.Equal(clh.state.VirtiofsdPID, 0)
|
||||
assert.Equal(clh.state.state, clhNotReady)
|
||||
|
||||
}
|
||||
|
||||
func TestClhVirtiofsdArgs(t *testing.T) {
|
||||
assert := assert.New(t)
|
||||
|
||||
defaultSocketPath := "/a/b/c/doit.sock"
|
||||
|
||||
clhConfig := newClhConfig()
|
||||
clh := &cloudHypervisor{
|
||||
ctx: context.Background(),
|
||||
cliBuilder: &DefaultCLIBuilder{},
|
||||
config: clhConfig,
|
||||
}
|
||||
|
||||
args, err := clh.virtiofsdArgs(defaultSocketPath)
|
||||
assert.NoError(err)
|
||||
assert.Equal(strings.Join(args, " "), "-f -o vhost_user_socket="+defaultSocketPath+" -o source=/run/kata-containers/shared/sandboxes -o cache="+virtioFsCacheAlways)
|
||||
|
||||
}
|
||||
|
||||
func TestClhPath(t *testing.T) {
|
||||
assert := assert.New(t)
|
||||
|
||||
clhConfig := newClhConfig()
|
||||
clh := &cloudHypervisor{
|
||||
ctx: context.Background(),
|
||||
cliBuilder: &DefaultCLIBuilder{},
|
||||
config: clhConfig,
|
||||
}
|
||||
|
||||
defaultPath, _ := clh.config.HypervisorAssetPath()
|
||||
|
||||
path, err := clh.clhPath()
|
||||
assert.NoError(err)
|
||||
assert.Equal(path, defaultPath)
|
||||
|
||||
}
|
||||
@ -42,6 +42,9 @@ var testQemuKernelPath = ""
|
||||
var testQemuInitrdPath = ""
|
||||
var testQemuImagePath = ""
|
||||
var testQemuPath = ""
|
||||
var testClhKernelPath = ""
|
||||
var testClhImagePath = ""
|
||||
var testClhPath = ""
|
||||
var testAcrnKernelPath = ""
|
||||
var testAcrnImagePath = ""
|
||||
var testAcrnPath = ""
|
||||
@ -85,6 +88,18 @@ func setupAcrn() {
|
||||
}
|
||||
}
|
||||
|
||||
func setupClh() {
|
||||
os.Mkdir(filepath.Join(testDir, testBundle), store.DirMode)
|
||||
|
||||
for _, filename := range []string{testClhKernelPath, testClhImagePath, testClhPath} {
|
||||
_, err := os.Create(filename)
|
||||
if err != nil {
|
||||
fmt.Printf("Could not recreate %s:%v", filename, err)
|
||||
os.Exit(1)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// TestMain is the common main function used by ALL the test functions
|
||||
// for this package.
|
||||
func TestMain(m *testing.M) {
|
||||
@ -127,6 +142,12 @@ func TestMain(m *testing.M) {
|
||||
|
||||
setupAcrn()
|
||||
|
||||
testClhKernelPath = filepath.Join(testDir, testKernel)
|
||||
testClhImagePath = filepath.Join(testDir, testImage)
|
||||
testClhPath = filepath.Join(testDir, testHypervisor)
|
||||
|
||||
setupClh()
|
||||
|
||||
ConfigStoragePathSaved := store.ConfigStoragePath
|
||||
RunStoragePathSaved := store.RunStoragePath
|
||||
// allow the tests to run without affecting the host system.
|
||||
|
||||
Loading…
Reference in New Issue
Block a user