mirror of
https://github.com/kata-containers/kata-containers.git
synced 2025-11-13 00:53:04 +00: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>
99 lines
2.4 KiB
Go
99 lines
2.4 KiB
Go
// Copyright (c) 2018 Intel Corporation
|
|
// Copyright (c) 2022 Apple Inc.
|
|
//
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
//
|
|
|
|
package virtcontainers
|
|
|
|
import (
|
|
"context"
|
|
"os"
|
|
"path"
|
|
"path/filepath"
|
|
"syscall"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestSandboxSharedFilesystem(t *testing.T) {
|
|
if os.Getuid() != 0 {
|
|
t.Skip("Test disabled as requires root user")
|
|
}
|
|
|
|
assert := assert.New(t)
|
|
// create temporary files to mount:
|
|
testMountPath := t.TempDir()
|
|
|
|
// create a new shared directory for our test:
|
|
kataHostSharedDirSaved := kataHostSharedDir
|
|
testHostDir := t.TempDir()
|
|
kataHostSharedDir = func() string {
|
|
return testHostDir
|
|
}
|
|
defer func() {
|
|
kataHostSharedDir = kataHostSharedDirSaved
|
|
}()
|
|
|
|
m1Path := filepath.Join(testMountPath, "foo.txt")
|
|
f1, err := os.Create(m1Path)
|
|
assert.NoError(err)
|
|
defer f1.Close()
|
|
|
|
m2Path := filepath.Join(testMountPath, "bar.txt")
|
|
f2, err := os.Create(m2Path)
|
|
assert.NoError(err)
|
|
defer f2.Close()
|
|
|
|
// create sandbox for mounting into
|
|
sandbox := &Sandbox{
|
|
ctx: context.Background(),
|
|
id: "foobar",
|
|
config: &SandboxConfig{
|
|
SandboxBindMounts: []string{m1Path, m2Path},
|
|
},
|
|
}
|
|
|
|
fsShare, err := NewFilesystemShare(sandbox)
|
|
assert.Nil(err)
|
|
sandbox.fsShare = fsShare
|
|
|
|
// make the shared directory for our test:
|
|
dir := kataHostSharedDir()
|
|
err = os.MkdirAll(path.Join(dir, sandbox.id), 0777)
|
|
assert.Nil(err)
|
|
|
|
// Test the prepare function. We expect it to succeed
|
|
err = sandbox.fsShare.Prepare(sandbox.ctx)
|
|
assert.NoError(err)
|
|
|
|
// Test the Cleanup function. We expect it to succeed for the mount to be removed.
|
|
err = sandbox.fsShare.Cleanup(sandbox.ctx)
|
|
assert.NoError(err)
|
|
|
|
// After successful Cleanup, verify there are not any mounts left behind.
|
|
stat := syscall.Stat_t{}
|
|
mount1CheckPath := filepath.Join(getMountPath(sandbox.id), sandboxMountsDir, filepath.Base(m1Path))
|
|
err = syscall.Stat(mount1CheckPath, &stat)
|
|
assert.Error(err)
|
|
assert.True(os.IsNotExist(err))
|
|
|
|
mount2CheckPath := filepath.Join(getMountPath(sandbox.id), sandboxMountsDir, filepath.Base(m2Path))
|
|
err = syscall.Stat(mount2CheckPath, &stat)
|
|
assert.Error(err)
|
|
assert.True(os.IsNotExist(err))
|
|
|
|
// Verify that Prepare is idempotent.
|
|
err = sandbox.fsShare.Prepare(sandbox.ctx)
|
|
assert.NoError(err)
|
|
err = sandbox.fsShare.Prepare(sandbox.ctx)
|
|
assert.NoError(err)
|
|
|
|
// Verify that Cleanup is idempotent.
|
|
err = sandbox.fsShare.Cleanup(sandbox.ctx)
|
|
assert.NoError(err)
|
|
err = sandbox.fsShare.Cleanup(sandbox.ctx)
|
|
assert.NoError(err)
|
|
}
|