mirror of
https://github.com/aljazceru/kata-containers.git
synced 2026-01-07 16:34:28 +01:00
6de95bf36c
To use the kata-containers repo path. Most of the change is generated by script: find . -type f -name "*.go" |xargs sed -i -e \ 's|github.com/kata-containers/runtime|github.com/kata-containers/kata-containers/src/runtime|g' Fixes: #201 Signed-off-by: Peng Tao <bergwolf@hyper.sh>
82 lines
2.3 KiB
Go
82 lines
2.3 KiB
Go
// Copyright (c) 2019 Huawei Corporation
|
|
//
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
//
|
|
|
|
package virtcontainers
|
|
|
|
import (
|
|
"context"
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
|
|
"github.com/kata-containers/kata-containers/src/runtime/virtcontainers/device/manager"
|
|
exp "github.com/kata-containers/kata-containers/src/runtime/virtcontainers/experimental"
|
|
"github.com/kata-containers/kata-containers/src/runtime/virtcontainers/persist"
|
|
"github.com/kata-containers/kata-containers/src/runtime/virtcontainers/types"
|
|
)
|
|
|
|
func TestSandboxRestore(t *testing.T) {
|
|
var err error
|
|
assert := assert.New(t)
|
|
sconfig := SandboxConfig{
|
|
ID: "test-exp",
|
|
Experimental: []exp.Feature{persist.NewStoreFeature},
|
|
}
|
|
container := make(map[string]*Container)
|
|
container["test-exp"] = &Container{}
|
|
|
|
sandbox := Sandbox{
|
|
id: "test-exp",
|
|
containers: container,
|
|
devManager: manager.NewDeviceManager(manager.VirtioSCSI, false, "", nil),
|
|
hypervisor: &mockHypervisor{},
|
|
ctx: context.Background(),
|
|
config: &sconfig,
|
|
state: types.SandboxState{BlockIndexMap: make(map[int]struct{})},
|
|
}
|
|
|
|
sandbox.newStore, err = persist.GetDriver()
|
|
assert.NoError(err)
|
|
assert.NotNil(sandbox.newStore)
|
|
|
|
// 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.newStore, err = persist.GetDriver(); err != nil || sandbox.newStore == 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{}{})
|
|
}
|