1 files changed, 79 insertions, 0 deletions
diff --git a/pkg/aflow/tool/codeeditor/codeeditor_test.go b/pkg/aflow/tool/codeeditor/codeeditor_test.go
new file mode 100644
index 000000000..4ba556f1b
--- /dev/null
+++ b/pkg/aflow/tool/codeeditor/codeeditor_test.go
@@ -0,0 +1,79 @@
+// Copyright 2026 syzkaller project authors. All rights reserved.
+// Use of this source code is governed by Apache 2 LICENSE that can be found in the LICENSE file.
+
+package codeeditor
+
+import (
+	"path/filepath"
+	"testing"
+
+	"github.com/google/syzkaller/pkg/aflow"
+	"github.com/google/syzkaller/pkg/osutil"
+	"github.com/stretchr/testify/require"
+)
+
+func TestCodeeditorEscapingPath(t *testing.T) {
+	aflow.TestTool(t, Tool,
+		state{
+			KernelScratchSrc: "whatever",
+		},
+		args{
+			SourceFile: "../../passwd",
+		},
+		struct{}{},
+		`SourceFile "../../passwd" is outside of the source tree`,
+	)
+}
+
+func TestCodeeditorMissingPath(t *testing.T) {
+	aflow.TestTool(t, Tool,
+		state{
+			KernelScratchSrc: t.TempDir(),
+		},
+		args{
+			SourceFile: "missing-file",
+		},
+		struct{}{},
+		`SourceFile "missing-file" does not exist`,
+	)
+}
+
+func TestCodeeditorEmptyCurrentCode(t *testing.T) {
+	dir := writeTestFile(t, "foo", "data")
+	aflow.TestTool(t, Tool,
+		state{
+			KernelScratchSrc: dir,
+		},
+		args{
+			SourceFile: "foo",
+		},
+		struct{}{},
+		`CurrentCode snippet is empty`,
+	)
+}
+
+func writeTestFile(t *testing.T, filename, data string) string {
+	dir := t.TempDir()
+	if err := osutil.WriteFile(filepath.Join(dir, filename), []byte(data)); err != nil {
+		t.Fatal(err)
+	}
+	return dir
+}
+
+func Fuzz(f *testing.F) {
+	dir := f.TempDir()
+	const filename = "src.c"
+	fullFilename := filepath.Join(dir, filename)
+	f.Fuzz(func(t *testing.T, fileData []byte, curCode, newCode string) {
+		require.NoError(t, osutil.WriteFile(fullFilename, fileData))
+		aflow.FuzzTool(t, Tool,
+			state{
+				KernelScratchSrc: dir,
+			},
+			args{
+				SourceFile:  filename,
+				CurrentCode: curCode,
+				NewCode:     newCode,
+			})
+	})
+}