add unit tests for adding files to a denied stating directory

kdomanski · Sep 17, 2022 · 740d005 · 740d005
1 parent 85afb8b
commit 740d005
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/image_writer_test.go b/image_writer_test.go
@@ -162,3 +162,15 @@ func TestWriterAddLocalDirectoryWithFile(t *testing.T) {
 
 	assert.False(t, os.IsNotExist(err))
 }
+
+func TestWriter_DeniedStagingDir(t *testing.T) {
+	w := &ImageWriter{stagingDir: "/usr/access_denied"}
+
+	err := w.AddLocalFile("/etc/hosts", "foo")
+	assert.Error(t, err)
+	assert.True(t, os.IsPermission(err), "err should have been a permission denied directory, but is: %+v", err)
+
+	err = w.AddFile(strings.NewReader("somestring"), "foo")
+	assert.Error(t, err)
+	assert.True(t, os.IsPermission(err), "err should have been a permission denied directory, but is: %+v", err)
+}