84 lines
1.8 KiB
Go
84 lines
1.8 KiB
Go
package system // import "github.com/docker/docker/pkg/system"
|
|
|
|
import (
|
|
"io/ioutil"
|
|
"os"
|
|
"path/filepath"
|
|
"runtime"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/docker/docker/pkg/mount"
|
|
"gotest.tools/skip"
|
|
)
|
|
|
|
func TestEnsureRemoveAllNotExist(t *testing.T) {
|
|
// should never return an error for a non-existent path
|
|
if err := EnsureRemoveAll("/non/existent/path"); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
|
|
func TestEnsureRemoveAllWithDir(t *testing.T) {
|
|
dir, err := ioutil.TempDir("", "test-ensure-removeall-with-dir")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err := EnsureRemoveAll(dir); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
|
|
func TestEnsureRemoveAllWithFile(t *testing.T) {
|
|
tmp, err := ioutil.TempFile("", "test-ensure-removeall-with-dir")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
tmp.Close()
|
|
if err := EnsureRemoveAll(tmp.Name()); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
|
|
func TestEnsureRemoveAllWithMount(t *testing.T) {
|
|
skip.If(t, runtime.GOOS == "windows", "mount not supported on Windows")
|
|
skip.If(t, os.Getuid() != 0, "skipping test that requires root")
|
|
|
|
dir1, err := ioutil.TempDir("", "test-ensure-removeall-with-dir1")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
dir2, err := ioutil.TempDir("", "test-ensure-removeall-with-dir2")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer os.RemoveAll(dir2)
|
|
|
|
bindDir := filepath.Join(dir1, "bind")
|
|
if err := os.MkdirAll(bindDir, 0755); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if err := mount.Mount(dir2, bindDir, "none", "bind"); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
done := make(chan struct{})
|
|
go func() {
|
|
err = EnsureRemoveAll(dir1)
|
|
close(done)
|
|
}()
|
|
|
|
select {
|
|
case <-done:
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
case <-time.After(5 * time.Second):
|
|
t.Fatal("timeout waiting for EnsureRemoveAll to finish")
|
|
}
|
|
|
|
if _, err := os.Stat(dir1); !os.IsNotExist(err) {
|
|
t.Fatalf("expected %q to not exist", dir1)
|
|
}
|
|
}
|