package plugin // import "github.com/docker/docker/plugin" import ( "io/ioutil" "os" "path/filepath" "testing" ) func TestAtomicRemoveAllNormal(t *testing.T) { dir, err := ioutil.TempDir("", "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 := ioutil.TempDir("", "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 := ioutil.TempDir("", "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) } }