mirror of
https://github.com/aljazceru/kata-containers.git
synced 2026-01-05 15:34:21 +01:00
00307a70ee
Fix sandbox implementation and unit tests to support the new persist API Signed-off-by: Julio Montes <julio.montes@intel.com>
77 lines
2.0 KiB
Go
77 lines
2.0 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/runtime/virtcontainers/device/manager"
|
|
exp "github.com/kata-containers/runtime/virtcontainers/experimental"
|
|
"github.com/kata-containers/runtime/virtcontainers/persist"
|
|
"github.com/kata-containers/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, nil),
|
|
hypervisor: &mockHypervisor{},
|
|
ctx: context.Background(),
|
|
config: &sconfig,
|
|
}
|
|
|
|
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(sandbox.state.BlockIndex, 0)
|
|
|
|
// set state data and save again
|
|
sandbox.state.State = types.StateString("running")
|
|
sandbox.state.GuestMemoryBlockSizeMB = uint32(1024)
|
|
sandbox.state.BlockIndex = 2
|
|
// flush data to disk
|
|
err = sandbox.Save()
|
|
assert.Nil(err)
|
|
|
|
// empty the sandbox
|
|
sandbox.state = types.SandboxState{}
|
|
|
|
// restore data from disk
|
|
err = sandbox.Restore()
|
|
assert.Nil(err)
|
|
assert.Equal(sandbox.state.State, types.StateString("running"))
|
|
assert.Equal(sandbox.state.GuestMemoryBlockSizeMB, uint32(1024))
|
|
assert.Equal(sandbox.state.BlockIndex, 2)
|
|
}
|