From 740d005028775750108a2604572ff3e3748d9f83 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Kamil=20Doma=C5=84ski?= <kamil@domanski.co>
Date: Sat, 17 Sep 2022 14:20:38 +0200
Subject: [PATCH] add unit tests for adding files to a denied stating directory

---
 image_writer_test.go | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/image_writer_test.go b/image_writer_test.go
index 96c74cf..49b51a7 100644
--- 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)
+}