mirror of
https://github.com/aljazceru/kata-containers.git
synced 2025-12-25 18:14:21 +01:00
ed64240df2
Add configuration options to support the various Kata agent tracing modes and types. See the comments in the built configuration files for details: - `cli/config/configuration-fc.toml` - `cli/config/configuration-qemu.toml` Fixes #1369. Signed-off-by: James O. D. Hunt <james.o.hunt@intel.com>
133 lines
2.7 KiB
Go
133 lines
2.7 KiB
Go
// Copyright (c) 2018 HyperHQ Inc.
|
|
//
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
//
|
|
|
|
package virtcontainers
|
|
|
|
import (
|
|
"context"
|
|
"io/ioutil"
|
|
"testing"
|
|
|
|
"github.com/kata-containers/runtime/virtcontainers/utils"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestNewVM(t *testing.T) {
|
|
assert := assert.New(t)
|
|
|
|
testDir, _ := ioutil.TempDir("", "vmfactory-tmp-")
|
|
config := VMConfig{
|
|
HypervisorType: MockHypervisor,
|
|
AgentType: NoopAgentType,
|
|
ProxyType: NoopProxyType,
|
|
}
|
|
hyperConfig := HypervisorConfig{
|
|
KernelPath: testDir,
|
|
ImagePath: testDir,
|
|
}
|
|
|
|
ctx := context.Background()
|
|
|
|
var vm *VM
|
|
_, err := NewVM(ctx, config)
|
|
assert.Error(err)
|
|
|
|
config.HypervisorConfig = hyperConfig
|
|
vm, err = NewVM(ctx, config)
|
|
assert.Nil(err)
|
|
|
|
// VM operations
|
|
err = vm.Pause()
|
|
assert.Nil(err)
|
|
err = vm.Resume()
|
|
assert.Nil(err)
|
|
err = vm.Start()
|
|
assert.Nil(err)
|
|
err = vm.Disconnect()
|
|
assert.Nil(err)
|
|
err = vm.Save()
|
|
assert.Nil(err)
|
|
err = vm.Stop()
|
|
assert.Nil(err)
|
|
err = vm.AddCPUs(2)
|
|
assert.Nil(err)
|
|
err = vm.AddMemory(128)
|
|
assert.Nil(err)
|
|
err = vm.OnlineCPUMemory()
|
|
assert.Nil(err)
|
|
err = vm.ReseedRNG()
|
|
assert.Nil(err)
|
|
|
|
// template VM
|
|
config.HypervisorConfig.BootFromTemplate = true
|
|
_, err = NewVM(ctx, config)
|
|
assert.Error(err)
|
|
|
|
config.HypervisorConfig.MemoryPath = testDir
|
|
_, err = NewVM(ctx, config)
|
|
assert.Error(err)
|
|
|
|
config.HypervisorConfig.DevicesStatePath = testDir
|
|
_, err = NewVM(ctx, config)
|
|
assert.Nil(err)
|
|
}
|
|
|
|
func TestVMConfigValid(t *testing.T) {
|
|
assert := assert.New(t)
|
|
|
|
config := VMConfig{}
|
|
|
|
err := config.Valid()
|
|
assert.Error(err)
|
|
|
|
testDir, _ := ioutil.TempDir("", "vmfactory-tmp-")
|
|
config.HypervisorConfig = HypervisorConfig{
|
|
KernelPath: testDir,
|
|
InitrdPath: testDir,
|
|
}
|
|
err = config.Valid()
|
|
assert.Nil(err)
|
|
}
|
|
|
|
func TestSetupProxy(t *testing.T) {
|
|
assert := assert.New(t)
|
|
|
|
config := VMConfig{
|
|
HypervisorType: MockHypervisor,
|
|
AgentType: NoopAgentType,
|
|
}
|
|
|
|
hypervisor := &mockHypervisor{}
|
|
agent := &noopAgent{}
|
|
|
|
// wrong proxy type
|
|
config.ProxyType = ProxyType("invalidProxyType")
|
|
_, _, _, err := setupProxy(hypervisor, agent, config, "foobar")
|
|
assert.NotNil(err)
|
|
|
|
config.ProxyType = NoopProxyType
|
|
_, _, _, err = setupProxy(hypervisor, agent, config, "foobar")
|
|
assert.Nil(err)
|
|
}
|
|
|
|
func TestVMConfigGrpc(t *testing.T) {
|
|
assert := assert.New(t)
|
|
config := VMConfig{
|
|
HypervisorType: QemuHypervisor,
|
|
HypervisorConfig: newQemuConfig(),
|
|
AgentType: KataContainersAgent,
|
|
AgentConfig: KataAgentConfig{false, true, false, false, "", ""},
|
|
ProxyType: NoopProxyType,
|
|
}
|
|
|
|
p, err := config.ToGrpc()
|
|
assert.Nil(err)
|
|
|
|
config2, err := GrpcToVMConfig(p)
|
|
assert.Nil(err)
|
|
|
|
assert.True(utils.DeepCompare(config, *config2))
|
|
}
|