1 files changed, 34 insertions, 0 deletions
diff --git a/internal/notify/service_test.go b/internal/notify/service_test.go
index fb47b98..871a0c7 100644
--- a/internal/notify/service_test.go
+++ b/internal/notify/service_test.go
@@ -144,6 +144,40 @@ func TestCloseNotificationEmitsReasonThree(t *testing.T) {
 	}
 }
 
+func TestRemoveImageRefusesPathsOutsideTheImageDir(t *testing.T) {
+	dir := t.TempDir()
+	imgDir := ImagesDir(dir)
+	if err := os.MkdirAll(imgDir, 0o700); err != nil {
+		t.Fatal(err)
+	}
+	sibling := filepath.Join(dir, "queue.json")
+	if err := os.WriteFile(sibling, []byte("x"), 0o600); err != nil {
+		t.Fatal(err)
+	}
+	owned := filepath.Join(imgDir, "1.png")
+	if err := os.WriteFile(owned, []byte("x"), 0o600); err != nil {
+		t.Fatal(err)
+	}
+	s := &Service{dir: dir}
+
+	s.removeImage(filepath.Join(imgDir, "..", "queue.json"))
+	if _, err := os.Stat(sibling); err != nil {
+		t.Fatalf("traversal removed a sibling: %v", err)
+	}
+	s.removeImage(sibling)
+	if _, err := os.Stat(sibling); err != nil {
+		t.Fatalf("unrelated path removed a sibling: %v", err)
+	}
+	s.removeImage("")
+	if _, err := os.Stat(sibling); err != nil {
+		t.Fatalf("empty path removed a sibling: %v", err)
+	}
+	s.removeImage(owned)
+	if _, err := os.Stat(owned); !os.IsNotExist(err) {
+		t.Fatalf("owned image not removed: %v", err)
+	}
+}
+
 // A natural expiry must drop the timer entry, or the map grows by one timer per
 // id for the whole session. No bus is needed: the store's emit is a no-op here,
 // so the test exercises arm's closure directly.