mirror of
https://github.com/aljazceru/kata-containers.git
synced 2026-01-08 17:04:26 +01:00
128 lines
2.2 KiB
Go
128 lines
2.2 KiB
Go
// Copyright (c) 2018 HyperHQ Inc.
|
|
//
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
//
|
|
|
|
package template
|
|
|
|
import (
|
|
"context"
|
|
"io/ioutil"
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
|
|
vc "github.com/kata-containers/runtime/virtcontainers"
|
|
"github.com/kata-containers/runtime/virtcontainers/persist/fs"
|
|
)
|
|
|
|
const testDisabledAsNonRoot = "Test disabled as requires root privileges"
|
|
|
|
var rootPathSave = fs.StorageRootPath()
|
|
|
|
func TestTemplateFactory(t *testing.T) {
|
|
if os.Geteuid() != 0 {
|
|
t.Skip(testDisabledAsNonRoot)
|
|
}
|
|
|
|
assert := assert.New(t)
|
|
|
|
templateWaitForAgent = 1 * time.Microsecond
|
|
|
|
testDir, _ := ioutil.TempDir("", "vmfactory-tmp-")
|
|
fs.TestSetStorageRootPath(filepath.Join(testDir, "vc"))
|
|
|
|
defer func() {
|
|
os.RemoveAll(testDir)
|
|
fs.TestSetStorageRootPath(rootPathSave)
|
|
}()
|
|
|
|
hyperConfig := vc.HypervisorConfig{
|
|
KernelPath: testDir,
|
|
ImagePath: testDir,
|
|
}
|
|
vmConfig := vc.VMConfig{
|
|
HypervisorType: vc.MockHypervisor,
|
|
HypervisorConfig: hyperConfig,
|
|
AgentType: vc.NoopAgentType,
|
|
ProxyType: vc.NoopProxyType,
|
|
}
|
|
|
|
err := vmConfig.Valid()
|
|
assert.Nil(err)
|
|
|
|
ctx := context.Background()
|
|
|
|
// New
|
|
f, err := New(ctx, vmConfig, testDir)
|
|
assert.Nil(err)
|
|
|
|
// Config
|
|
assert.Equal(f.Config(), vmConfig)
|
|
|
|
// GetBaseVM
|
|
vm, err := f.GetBaseVM(ctx, vmConfig)
|
|
assert.Nil(err)
|
|
|
|
err = vm.Stop()
|
|
assert.Nil(err)
|
|
|
|
// Fetch
|
|
tt := template{
|
|
statePath: testDir,
|
|
config: vmConfig,
|
|
}
|
|
|
|
assert.Equal(tt.Config(), vmConfig)
|
|
|
|
err = tt.checkTemplateVM()
|
|
assert.Error(err)
|
|
|
|
_, err = os.Create(tt.statePath + "/memory")
|
|
assert.Nil(err)
|
|
err = tt.checkTemplateVM()
|
|
assert.Error(err)
|
|
|
|
_, err = os.Create(tt.statePath + "/state")
|
|
assert.Nil(err)
|
|
err = tt.checkTemplateVM()
|
|
assert.Nil(err)
|
|
|
|
err = tt.createTemplateVM(ctx)
|
|
assert.Nil(err)
|
|
|
|
vm, err = tt.GetBaseVM(ctx, vmConfig)
|
|
assert.Nil(err)
|
|
|
|
err = vm.Stop()
|
|
assert.Nil(err)
|
|
|
|
vm, err = f.GetBaseVM(ctx, vmConfig)
|
|
assert.Nil(err)
|
|
|
|
err = vm.Stop()
|
|
assert.Nil(err)
|
|
|
|
err = tt.createTemplateVM(ctx)
|
|
assert.Nil(err)
|
|
|
|
vm, err = tt.GetBaseVM(ctx, vmConfig)
|
|
assert.Nil(err)
|
|
|
|
err = vm.Stop()
|
|
assert.Nil(err)
|
|
|
|
vm, err = f.GetBaseVM(ctx, vmConfig)
|
|
assert.Nil(err)
|
|
|
|
err = vm.Stop()
|
|
assert.Nil(err)
|
|
|
|
// CloseFactory
|
|
f.CloseFactory(ctx)
|
|
tt.CloseFactory(ctx)
|
|
}
|