feat: add unit tests for output formatting and enhance file system schema

Author: NicolasIRAGNE

src/gitingest/output_formatter.py

@@ -324,7 +324,7 @@ def summary(self, node: Source, query):
 
     @summary.register
     def _(self, node: FileSystemDirectory, query):
-        template = """
+        template = """ \
 Directory structure:
 {{ node.tree }}
 """
@@ -334,7 +334,9 @@ def _(self, node: FileSystemDirectory, query):
     @summary.register
     def _(self, context: Context, query):
         template = """
-{{ context.summary }}
+Repository: {{ context.query.user_name }}/{{ context.query.repo_name }}
+Commit: {{ context.query.commit }}
+Files analyzed: {{ context.sources[0].file_count }}
 """
         summary_template = self.env.from_string(template)
         return summary_template.render(context=context, query=query)

src/gitingest/schemas/filesystem.py

@@ -70,6 +70,7 @@ class FileSystemDirectory(FileSystemNode):
     children: list[FileSystemNode] = field(default_factory=list)
     file_count: int = 0
     dir_count: int = 0
+    file_count_total: int = 0
     type: FileSystemNodeType = FileSystemNodeType.DIRECTORY
 
     def sort_children(self) -> None:

src/server/query_processor.py

@@ -7,7 +7,7 @@
 
 from gitingest.clone import clone_repo
 from gitingest.ingestion import ingest_query
-from gitingest.output_formatter import DefaultFormatter, SummaryFormatter
+from gitingest.output_formatter import DebugFormatter, DefaultFormatter, SummaryFormatter
 from gitingest.query_parser import parse_remote_repo
 from gitingest.utils.git_utils import resolve_commit, validate_github_token
 from gitingest.utils.logging_config import get_logger

test_formatting.py

@@ -0,0 +1,36 @@
+from pathlib import Path
+from unittest.mock import Mock
+from gitingest.output_formatter import DefaultFormatter
+from gitingest.schemas.filesystem import Context, GitRepository, FileSystemFile
+
+# Create a mock query
+mock_query = Mock()
+mock_query.user_name = "test_user"
+mock_query.repo_name = "test_repo"
+
+# Create a simple file
+mock_file = FileSystemFile(
+    name="test.py",
+    path_str="test.py", 
+    path=Path("test.py"),
+)
+mock_file.content = "print('hello world')"
+
+# Create a git repository with the file
+mock_repo = GitRepository(
+    name="test_repo",
+    path_str="",
+    path=Path("."),
+    children=[mock_file]
+)
+
+# Create context
+context = Context([mock_repo], mock_query)
+
+# Test formatting
+formatter = DefaultFormatter()
+result = formatter.format(context, mock_query)
+print("RESULT:")
+print(repr(result))
+print("\nFORMATTED:")
+print(result)