fs: add DirFromPath

Author: dnephin

fs/file.go

@@ -113,3 +113,18 @@ func (d *Dir) Remove() {
 func (d *Dir) Join(parts ...string) string {
 	return filepath.Join(append([]string{d.Path()}, parts...)...)
 }
+
+// DirFromPath returns a Dir for a path that already exists. No directory is created.
+// DirFromPath can be used with Apply to modify an existing directory.
+//
+// If the path does not already exist, use NewDir instead.
+func DirFromPath(t assert.TestingT, path string, ops ...PathOp) *Dir {
+	if ht, ok := t.(helperT); ok {
+		ht.Helper()
+	}
+
+	dir := &Dir{path: path}
+	cleanup.Cleanup(t, dir.Remove)
+	assert.NilError(t, applyPathOps(dir, ops))
+	return dir
+}

fs/file_test.go

@@ -1,7 +1,10 @@
 package fs_test
 
 import (
+	"errors"
+	"io/ioutil"
 	"os"
+	"path/filepath"
 	"testing"
 
 	"gotest.tools/v3/assert"
@@ -92,3 +95,24 @@ func TestNewDir_IntegrationWithCleanup(t *testing.T) {
 		assert.ErrorType(t, err, os.IsNotExist)
 	})
 }
+
+func TestDirFromPath(t *testing.T) {
+	tmpdir, err := ioutil.TempDir("", t.Name())
+	assert.NilError(t, err)
+	t.Cleanup(func() {
+		os.RemoveAll(tmpdir)
+	})
+
+	dir := fs.DirFromPath(t, tmpdir, fs.WithFile("newfile", ""))
+
+	_, err = os.Stat(dir.Join("newfile"))
+	assert.NilError(t, err)
+
+	assert.Equal(t, dir.Path(), tmpdir)
+	assert.Equal(t, dir.Join("newfile"), filepath.Join(tmpdir, "newfile"))
+
+	dir.Remove()
+
+	_, err = os.Stat(tmpdir)
+	assert.Assert(t, errors.Is(err, os.ErrNotExist))
+}