21 lines
752 B
Python
21 lines
752 B
Python
from src.document_processor import _is_text_file, _process_text_file
|
|
from src.upload_handler import UploadHandler
|
|
|
|
|
|
def test_nix_files_are_treated_as_readable_documents(tmp_path):
|
|
handler = UploadHandler(str(tmp_path), str(tmp_path / "uploads"))
|
|
|
|
assert handler.is_document_file("configuration.nix")
|
|
assert _is_text_file("configuration.nix")
|
|
|
|
|
|
def test_nix_file_processing_includes_content_in_code_block(tmp_path):
|
|
nix_file = tmp_path / "configuration.nix"
|
|
nix_file.write_text("{ pkgs, ... }:\n{\n services.openssh.enable = true;\n}\n", encoding="utf-8")
|
|
|
|
rendered = _process_text_file(str(nix_file))
|
|
|
|
assert "[Type: nix" in rendered
|
|
assert "```nix" in rendered
|
|
assert "services.openssh.enable = true;" in rendered
|