// SetupHostPathVolumes will create multiple PersistentVolumes with given capacity func SetupHostPathVolumes(c kclient.PersistentVolumeInterface, prefix, capacity string, count int) (volumes []*kapi.PersistentVolume, err error) { rootDir, err := ioutil.TempDir(TestContext.OutputDir, "persistent-volumes") if err != nil { return volumes, err } for i := 0; i < count; i++ { dir, err := ioutil.TempDir(rootDir, fmt.Sprintf("%0.4d", i)) if err != nil { return volumes, err } if _, err = exec.LookPath("chcon"); err != nil { err := exec.Command("chcon", "-t", "svirt_sandbox_file_t", dir).Run() if err != nil { return volumes, err } } if err = os.Chmod(dir, 0777); err != nil { return volumes, err } pv, err := c.Create(CreatePersistentVolume(fmt.Sprintf("%s%s-%0.4d", pvPrefix, prefix, i), capacity, dir)) if err != nil { return volumes, err } volumes = append(volumes, pv) } return volumes, err }
// CleanupHostPathVolumes removes all PersistentVolumes created by // SetupHostPathVolumes, with a given prefix func CleanupHostPathVolumes(c kclient.PersistentVolumeInterface, prefix string) error { pvs, err := c.List(kapi.ListOptions{}) if err != nil { return err } prefix = fmt.Sprintf("%s%s-", pvPrefix, prefix) for _, pv := range pvs.Items { if strings.HasPrefix(pv.Name, prefix) { c.Delete(pv.Name) } } return nil }
// CleanupHostPathVolumes removes all PersistentVolumes created by // SetupHostPathVolumes, with a given prefix func CleanupHostPathVolumes(c kclient.PersistentVolumeInterface, prefix string) error { pvs, err := c.List(kapi.ListOptions{}) if err != nil { return err } prefix = fmt.Sprintf("%s%s-", pvPrefix, prefix) for _, pv := range pvs.Items { if !strings.HasPrefix(pv.Name, prefix) { continue } if err = c.Delete(pv.Name); err != nil { fmt.Fprintf(g.GinkgoWriter, "WARNING: couldn't remove PV %s: %v\n", pv.Name, err) continue } pvInfo, err := c.Get(pv.Name) if err != nil { fmt.Fprintf(g.GinkgoWriter, "WARNING: couldn't get meta info for PV %s: %v\n", pv.Name, err) continue } volumeDir := pvInfo.Spec.HostPath.Path if err = os.RemoveAll(volumeDir); err != nil { fmt.Fprintf(g.GinkgoWriter, "WARNING: couldn't remove directory %q: %v\n", volumeDir, err) continue } parentDir := filepath.Dir(volumeDir) if parentDir == "." || parentDir == "/" { continue } if err = os.Remove(parentDir); err != nil { fmt.Fprintf(g.GinkgoWriter, "WARNING: couldn't remove directory %q: %v\n", parentDir, err) continue } } return nil }