c55a4ac779
The io/ioutil package has been deprecated in Go 1.16. This commit replaces the existing io/ioutil functions with their new definitions in io and os packages. Signed-off-by: Eng Zer Jun <engzerjun@gmail.com>
80 lines
1.9 KiB
Go
80 lines
1.9 KiB
Go
package plugin // import "github.com/docker/docker/plugin"
|
|
|
|
import (
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
)
|
|
|
|
func TestAtomicRemoveAllNormal(t *testing.T) {
|
|
dir, err := os.MkdirTemp("", "atomic-remove-with-normal")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer os.RemoveAll(dir) // just try to make sure this gets cleaned up
|
|
|
|
if err := atomicRemoveAll(dir); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if _, err := os.Stat(dir); !os.IsNotExist(err) {
|
|
t.Fatalf("dir should be gone: %v", err)
|
|
}
|
|
if _, err := os.Stat(dir + "-removing"); !os.IsNotExist(err) {
|
|
t.Fatalf("dir should be gone: %v", err)
|
|
}
|
|
}
|
|
|
|
func TestAtomicRemoveAllAlreadyExists(t *testing.T) {
|
|
dir, err := os.MkdirTemp("", "atomic-remove-already-exists")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer os.RemoveAll(dir) // just try to make sure this gets cleaned up
|
|
|
|
if err := os.MkdirAll(dir+"-removing", 0755); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer os.RemoveAll(dir + "-removing")
|
|
|
|
if err := atomicRemoveAll(dir); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if _, err := os.Stat(dir); !os.IsNotExist(err) {
|
|
t.Fatalf("dir should be gone: %v", err)
|
|
}
|
|
if _, err := os.Stat(dir + "-removing"); !os.IsNotExist(err) {
|
|
t.Fatalf("dir should be gone: %v", err)
|
|
}
|
|
}
|
|
|
|
func TestAtomicRemoveAllNotExist(t *testing.T) {
|
|
if err := atomicRemoveAll("/not-exist"); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
dir, err := os.MkdirTemp("", "atomic-remove-already-exists")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer os.RemoveAll(dir) // just try to make sure this gets cleaned up
|
|
|
|
// create the removing dir, but not the "real" one
|
|
foo := filepath.Join(dir, "foo")
|
|
removing := dir + "-removing"
|
|
if err := os.MkdirAll(removing, 0755); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if err := atomicRemoveAll(dir); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if _, err := os.Stat(foo); !os.IsNotExist(err) {
|
|
t.Fatalf("dir should be gone: %v", err)
|
|
}
|
|
if _, err := os.Stat(removing); !os.IsNotExist(err) {
|
|
t.Fatalf("dir should be gone: %v", err)
|
|
}
|
|
}
|