mirror of
https://github.com/aljazceru/kata-containers.git
synced 2026-01-23 00:04:24 +01:00
efa19c41eb
Currently, the block driver option is specifed by hard coding, maybe it is better to use const string variables instead of hard coded strings. Another modification is to remove duplicate consts for virtio driver in manager.go. Fixes: #3321 Signed-off-by: Jason Zhang <zhanghj.lc@inspur.com>
84 lines
2.3 KiB
Go
84 lines
2.3 KiB
Go
// Copyright (c) 2019 Huawei Corporation
|
|
//
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
//
|
|
|
|
package virtcontainers
|
|
|
|
import (
|
|
"context"
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/kata-containers/kata-containers/src/runtime/virtcontainers/device/config"
|
|
"github.com/kata-containers/kata-containers/src/runtime/virtcontainers/device/manager"
|
|
"github.com/kata-containers/kata-containers/src/runtime/virtcontainers/persist"
|
|
"github.com/kata-containers/kata-containers/src/runtime/virtcontainers/types"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestSandboxRestore(t *testing.T) {
|
|
var err error
|
|
assert := assert.New(t)
|
|
sconfig := SandboxConfig{
|
|
ID: "test-exp",
|
|
}
|
|
container := make(map[string]*Container)
|
|
container["test-exp"] = &Container{}
|
|
|
|
network, err := NewNetwork()
|
|
assert.NoError(err)
|
|
|
|
sandbox := Sandbox{
|
|
id: "test-exp",
|
|
containers: container,
|
|
devManager: manager.NewDeviceManager(config.VirtioSCSI, false, "", nil),
|
|
hypervisor: &mockHypervisor{},
|
|
network: network,
|
|
ctx: context.Background(),
|
|
config: &sconfig,
|
|
state: types.SandboxState{BlockIndexMap: make(map[int]struct{})},
|
|
}
|
|
|
|
sandbox.store, err = persist.GetDriver()
|
|
assert.NoError(err)
|
|
assert.NotNil(sandbox.store)
|
|
|
|
// if we don't call Save(), we can get nothing from disk
|
|
err = sandbox.Restore()
|
|
assert.NotNil(t, err)
|
|
assert.True(os.IsNotExist(err))
|
|
|
|
// disk data are empty
|
|
err = sandbox.Save()
|
|
assert.NoError(err)
|
|
|
|
err = sandbox.Restore()
|
|
assert.NoError(err)
|
|
assert.Equal(sandbox.state.State, types.StateString(""))
|
|
assert.Equal(sandbox.state.GuestMemoryBlockSizeMB, uint32(0))
|
|
assert.Equal(len(sandbox.state.BlockIndexMap), 0)
|
|
|
|
// set state data and Save again
|
|
sandbox.state.State = types.StateString("running")
|
|
sandbox.state.GuestMemoryBlockSizeMB = uint32(1024)
|
|
sandbox.state.BlockIndexMap[2] = struct{}{}
|
|
// flush data to disk
|
|
err = sandbox.Save()
|
|
assert.Nil(err)
|
|
|
|
// empty the sandbox
|
|
sandbox.state = types.SandboxState{}
|
|
if sandbox.store, err = persist.GetDriver(); err != nil || sandbox.store == nil {
|
|
t.Fatal("failed to get persist driver")
|
|
}
|
|
|
|
// restore data from disk
|
|
err = sandbox.Restore()
|
|
assert.NoError(err)
|
|
assert.Equal(sandbox.state.State, types.StateString("running"))
|
|
assert.Equal(sandbox.state.GuestMemoryBlockSizeMB, uint32(1024))
|
|
assert.Equal(len(sandbox.state.BlockIndexMap), 1)
|
|
assert.Equal(sandbox.state.BlockIndexMap[2], struct{}{})
|
|
}
|