test: improved test_fileblock and fixed --no-confirm for overwrite

gptme/tools/save.py

@@ -33,23 +33,31 @@ def execute_save(
             return
 
         with open(path, "a") as f:
-            f.write("\n" + code)
+            f.write(code)
         yield Message("system", f"Appended to {fn}")
         return
 
     # if the file exists, ask to overwrite
     if path.exists():
-        overwrite = ask_execute("File exists, overwrite?")
-        print()
+        if ask:
+            overwrite = ask_execute("File exists, overwrite?")
+            print()
+        else:
+            overwrite = True
+            print("Skipping overwrite confirmation.")
         if not overwrite:
             # early return
             yield Message("system", "Save cancelled.")
             return
 
     # if the folder doesn't exist, ask to create it
     if not path.parent.exists():
-        create = ask_execute("Folder doesn't exist, create it?")
-        print()
+        if ask:
+            create = ask_execute("Folder doesn't exist, create it?")
+            print()
+        else:
+            create = True
+            print("Skipping folder creation confirmation.")
         if create:
             path.parent.mkdir(parents=True)
         else:

tests/test_cli.py

@@ -96,6 +96,8 @@ def test_command_fork(args: list[str], runner: CliRunner, name: str):
 
 @pytest.mark.slow
 def test_fileblock(args: list[str], runner: CliRunner):
+    args_orig = args.copy()
+
     # tests saving with a ```filename.txt block
     args.append(f"{CMDFIX}impersonate ```hello.py\nprint('hello')\n```")
     result = runner.invoke(gptme.cli.main, args)
@@ -106,6 +108,23 @@ def test_fileblock(args: list[str], runner: CliRunner):
         content = f.read()
     assert content == "print('hello')\n"
 
+    # test append
+    args = args_orig.copy()
+    args.append(f"{CMDFIX}impersonate ```append hello.py\nprint('world')\n```")
+    result = runner.invoke(gptme.cli.main, args)
+    assert result.exit_code == 0
+
+    # read the file
+    with open("hello.py", "r") as f:
+        content = f.read()
+    assert content == "print('hello')\nprint('world')\n"
+
+    # test write file to directory that doesn't exist
+    args = args_orig.copy()
+    args.append(f"{CMDFIX}impersonate ```hello/hello.py\nprint('hello')\n```")
+    result = runner.invoke(gptme.cli.main, args)
+    assert result.exit_code == 0
+
 
 def test_shell(args: list[str], runner: CliRunner):
     args.append(f"{CMDFIX}shell echo 'yes'")