import os
import pytest
from unittest.mock import patch

from rook.files import read_file, write_file, delete_file, PolicyError


def test_read_file_returns_content(tmp_path):
 tmp_file = tmp_path / "hello.txt"
 tmp_file.write_text("hello world")
 with patch("rook.files.is_approved_dir", return_value=True):
  result = read_file(str(tmp_file))
 assert result == "hello world"


def test_read_file_unapproved_raises():
 with patch("rook.files.is_approved_dir", return_value=False), \
      patch("rook.files.confirm_spawn", return_value=False):
  with pytest.raises(PolicyError):
   read_file("/etc/passwd")


def test_write_file_creates_backup(tmp_path):
 tmp_file = tmp_path / "data.txt"
 tmp_file.write_text("old")
 with patch("rook.files.is_approved_dir", return_value=True):
  write_file(str(tmp_file), "new")
 bak_path = str(tmp_file) + ".bak"
 assert os.path.exists(bak_path)
 with open(bak_path) as f:
  assert f.read() == "old"
 assert tmp_file.read_text() == "new"


def test_write_file_new_file_no_backup(tmp_path):
 path = str(tmp_path / "new.txt")
 with patch("rook.files.is_approved_dir", return_value=True), \
      patch("rook.files.backup_before_edit"):
  write_file(path, "content")
 assert os.path.exists(path)
 with open(path) as f:
  assert f.read() == "content"
 assert not os.path.exists(path + ".bak")


def test_delete_file_requires_confirmation_and_deletes(tmp_path):
 tmp_file = tmp_path / "todelete.txt"
 tmp_file.write_text("bye")
 with patch("rook.files.is_approved_dir", return_value=True), \
      patch("rook.files.confirm_spawn", return_value=True):
  delete_file(str(tmp_file))
 assert not tmp_file.exists()


def test_delete_file_denied_leaves_file(tmp_path):
 tmp_file = tmp_path / "safe.txt"
 tmp_file.write_text("keep me")
 with patch("rook.files.is_approved_dir", return_value=True), \
      patch("rook.files.confirm_spawn", return_value=False):
  with pytest.raises(PolicyError):
   delete_file(str(tmp_file))
 assert tmp_file.exists()