mirror of
https://github.com/aljazceru/kata-containers.git
synced 2026-01-02 14:04:22 +01:00
59c7165ee1
The directory created by `T.TempDir` is automatically removed when the test and all its subtests complete. This commit also updates the unit test advice to use `T.TempDir` to create temporary directory in tests. Fixes: #3924 Reference: https://pkg.go.dev/testing#T.TempDir Signed-off-by: Eng Zer Jun <engzerjun@gmail.com>
129 lines
2.6 KiB
Go
129 lines
2.6 KiB
Go
// Copyright (c) 2018 HyperHQ Inc.
|
|
//
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
//
|
|
|
|
package virtcontainers
|
|
|
|
import (
|
|
"context"
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
"github.com/kata-containers/kata-containers/src/runtime/virtcontainers/utils"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestNewVM(t *testing.T) {
|
|
assert := assert.New(t)
|
|
|
|
testDir := t.TempDir()
|
|
|
|
config := VMConfig{
|
|
HypervisorType: MockHypervisor,
|
|
}
|
|
hyperConfig := HypervisorConfig{
|
|
KernelPath: testDir,
|
|
ImagePath: testDir,
|
|
}
|
|
|
|
ctx := WithNewAgentFunc(context.Background(), newMockAgent)
|
|
|
|
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(context.Background())
|
|
assert.Nil(err)
|
|
err = vm.Resume(context.Background())
|
|
assert.Nil(err)
|
|
err = vm.Start(context.Background())
|
|
assert.Nil(err)
|
|
err = vm.Disconnect(context.Background())
|
|
assert.Nil(err)
|
|
err = vm.Save()
|
|
assert.Nil(err)
|
|
err = vm.Stop(context.Background())
|
|
assert.Nil(err)
|
|
err = vm.AddCPUs(context.Background(), 2)
|
|
assert.Nil(err)
|
|
err = vm.AddMemory(context.Background(), 128)
|
|
assert.Nil(err)
|
|
err = vm.OnlineCPUMemory(context.Background())
|
|
assert.Nil(err)
|
|
|
|
// mock urandom device
|
|
savedUrandomDev := urandomDev
|
|
defer func() {
|
|
urandomDev = savedUrandomDev
|
|
}()
|
|
tmpdir := t.TempDir()
|
|
urandomDev = filepath.Join(tmpdir, "urandom")
|
|
data := make([]byte, 512)
|
|
err = os.WriteFile(urandomDev, data, os.FileMode(0640))
|
|
assert.NoError(err)
|
|
|
|
err = vm.ReseedRNG(context.Background())
|
|
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 := t.TempDir()
|
|
|
|
config.HypervisorConfig = HypervisorConfig{
|
|
KernelPath: testDir,
|
|
InitrdPath: testDir,
|
|
}
|
|
err = config.Valid()
|
|
assert.Nil(err)
|
|
}
|
|
|
|
func TestVMConfigGrpc(t *testing.T) {
|
|
assert := assert.New(t)
|
|
config := VMConfig{
|
|
HypervisorType: QemuHypervisor,
|
|
HypervisorConfig: newQemuConfig(),
|
|
AgentConfig: KataAgentConfig{
|
|
LongLiveConn: true,
|
|
Debug: false,
|
|
Trace: false,
|
|
EnableDebugConsole: false,
|
|
ContainerPipeSize: 0,
|
|
KernelModules: []string{}},
|
|
}
|
|
|
|
p, err := config.ToGrpc()
|
|
assert.Nil(err)
|
|
|
|
config2, err := GrpcToVMConfig(p)
|
|
assert.Nil(err)
|
|
|
|
assert.True(utils.DeepCompare(config, *config2))
|
|
}
|