Add ignore socket test

dmcgowan · dmcgowan · commit d77d4bab26e7 · 2018-03-22T16:24:53.000-07:00
Signed-off-by: Derek McGowan &lt;derek@mcgstyle.net&gt;
diff --git a/archive/tar_test.go b/archive/tar_test.go
@@ -1006,6 +1006,22 @@ func TestDiffTar(t *testing.T) {
 				fstest.CreateDir("/d3/", 0755),
 			),
 		},
+		{
+			name: "IgnoreSockets",
+			validators: []tarEntryValidator{
+				fileEntry("f2", []byte("content"), 0644),
+				// There should be _no_ socket here, despite the fstest.CreateSocket below
+				fileEntry("f3", []byte("content"), 0644),
+			},
+			a: fstest.Apply(
+				fstest.CreateFile("/f1", []byte("content"), 0644),
+			),
+			b: fstest.Apply(
+				fstest.CreateFile("/f2", []byte("content"), 0644),
+				fstest.CreateSocket("/s0", 0644),
+				fstest.CreateFile("/f3", []byte("content"), 0644),
+			),
+		},
 	}
 
 	for _, at := range tests {
diff --git a/fs/fstest/file.go b/fs/fstest/file.go
@@ -2,6 +2,7 @@ package fstest
 
 import (
 	"io/ioutil"
+	"net"
 	"os"
 	"path/filepath"
 	"time"
@@ -107,6 +108,19 @@ func Link(oldname, newname string) Applier {
 //	}
 //}
 
+// CreateSocket returns a file applier which creates a unix socket
+func CreateSocket(name string, perm os.FileMode) Applier {
+	return applyFn(func(root string) error {
+		fullPath := filepath.Join(root, name)
+		ln, err := net.Listen("unix", fullPath)
+		if err != nil {
+			return err
+		}
+		defer ln.Close()
+		return os.Chmod(fullPath, perm)
+	})
+}
+
 // Apply returns a new applier from the given appliers
 func Apply(appliers ...Applier) Applier {
 	return applyFn(func(root string) error {
diff --git a/fs/fstest/testsuite.go b/fs/fstest/testsuite.go
@@ -74,6 +74,7 @@ var (
 		Symlink("libnothing.so", "/usr/local/lib/libnothing.so.2"),
 		CreateDir("/home", 0755),
 		CreateDir("/home/derek", 0700),
+		CreateDir("/var/run/socket", 0700),
 	)
 
 	// basicTest covers basic operations

Original file line number	Diff line number	Diff line change
`@@ -74,6 +74,7 @@ var (`
`74`	`74`	`Symlink("libnothing.so", "/usr/local/lib/libnothing.so.2"),`
`75`	`75`	`CreateDir("/home", 0755),`
`76`	`76`	`CreateDir("/home/derek", 0700),`
	`77`	`+ CreateDir("/var/run/socket", 0700),`
`77`	`78`	`)`
`78`	`79`
`79`	`80`	`// basicTest covers basic operations`