179 lines
3.4 KiB
Go
179 lines
3.4 KiB
Go
//go:build linux
|
|
// +build linux
|
|
|
|
package mount
|
|
|
|
import (
|
|
"os"
|
|
"path"
|
|
"slices"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestMountOptionsParsing(t *testing.T) {
|
|
options := "noatime,ro,size=10k"
|
|
|
|
flag, data := ParseOptions(options)
|
|
|
|
if data != "size=10k" {
|
|
t.Fatalf("Expected size=10 got %s", data)
|
|
}
|
|
|
|
expectedFlag := NOATIME | RDONLY
|
|
|
|
if flag != expectedFlag {
|
|
t.Fatalf("Expected %d got %d", expectedFlag, flag)
|
|
}
|
|
}
|
|
|
|
func TestMounted(t *testing.T) {
|
|
if os.Getuid() != 0 {
|
|
t.Skip("root required")
|
|
}
|
|
|
|
tmp := path.Join(os.TempDir(), "mount-tests")
|
|
if err := os.MkdirAll(tmp, 0o777); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer os.RemoveAll(tmp)
|
|
|
|
var (
|
|
sourceDir = path.Join(tmp, "source")
|
|
targetDir = path.Join(tmp, "target")
|
|
sourcePath = path.Join(sourceDir, "file.txt")
|
|
targetPath = path.Join(targetDir, "file.txt")
|
|
)
|
|
|
|
if err := os.Mkdir(sourceDir, 0o777); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if err := os.Mkdir(targetDir, 0o777); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
f, err := os.Create(sourcePath)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if _, err = f.WriteString("hello"); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
f.Close()
|
|
|
|
f, err = os.Create(targetPath)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
f.Close()
|
|
|
|
if err := Mount(sourceDir, targetDir, none, "bind,rw"); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer func() {
|
|
if err := Unmount(targetDir); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}()
|
|
|
|
mounted, err := Mounted(targetDir)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if !mounted {
|
|
t.Fatalf("Expected %s to be mounted", targetDir)
|
|
}
|
|
if _, err := os.Stat(targetDir); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
|
|
func TestMountReadonly(t *testing.T) {
|
|
if os.Getuid() != 0 {
|
|
t.Skip("root required")
|
|
}
|
|
|
|
tmp := path.Join(os.TempDir(), "mount-tests")
|
|
if err := os.MkdirAll(tmp, 0o777); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer os.RemoveAll(tmp)
|
|
|
|
var (
|
|
sourceDir = path.Join(tmp, "source")
|
|
targetDir = path.Join(tmp, "target")
|
|
sourcePath = path.Join(sourceDir, "file.txt")
|
|
targetPath = path.Join(targetDir, "file.txt")
|
|
)
|
|
|
|
if err := os.Mkdir(sourceDir, 0o777); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err := os.Mkdir(targetDir, 0o777); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
f, err := os.Create(sourcePath)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
_, err = f.WriteString("hello")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
f.Close()
|
|
|
|
f, err = os.Create(targetPath)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
f.Close()
|
|
|
|
if err := Mount(sourceDir, targetDir, none, "bind,ro"); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer func() {
|
|
if err := Unmount(targetDir); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}()
|
|
|
|
f, err = os.OpenFile(targetPath, os.O_RDWR, 0o777)
|
|
if err == nil {
|
|
f.Close()
|
|
t.Fatal("Should not be able to open a ro file as rw")
|
|
}
|
|
}
|
|
|
|
func TestGetMounts(t *testing.T) {
|
|
mounts, err := GetMounts()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if !slices.ContainsFunc(mounts, func(entry *Info) bool {
|
|
return entry.Mountpoint == "/"
|
|
}) {
|
|
t.Fatal("/ should be mounted at least")
|
|
}
|
|
}
|
|
|
|
func TestMergeTmpfsOptions(t *testing.T) {
|
|
options := []string{"noatime", "ro", "size=10k", "defaults", "atime", "defaults", "rw", "rprivate", "size=1024k", "slave"}
|
|
expected := []string{"atime", "rw", "size=1024k", "slave"}
|
|
merged, err := MergeTmpfsOptions(options)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
require.Equal(t, expected, merged)
|
|
|
|
options = []string{"noatime", "ro", "size=10k", "atime", "rw", "rprivate", "size=1024k", "slave", "size"}
|
|
_, err = MergeTmpfsOptions(options)
|
|
if err == nil {
|
|
t.Fatal("Expected error got nil")
|
|
}
|
|
}
|