Commit 8af2206

committed

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

1 parent e6b6e74 commit 8af2206Copy full SHA for 8af2206

File tree

+42

-3

lines changed

+42

-3

lines changed

Lines changed: 4 additions & 2 deletions

Original file line number	Diff line number	Diff line change
`@@ -324,7 +324,7 @@ def summary(self, node: Source, query):`
`324`	`324`
`325`	`325`	`@summary.register`
`326`	`326`	`def _(self, node: FileSystemDirectory, query):`
`327`		`- template = """`
	`327`	`+ template = """\`
`328`	`328`	`Directory structure:`
`329`	`329`	`{{ node.tree }}`
`330`	`330`	`"""`
`@@ -334,7 +334,9 @@ def _(self, node: FileSystemDirectory, query):`
`334`	`334`	`@summary.register`
`335`	`335`	`def _(self, context: Context, query):`
`336`	`336`	`template = """`
`337`		`-{{ context.summary }}`
	`337`	`+Repository: {{ context.query.user_name }}/{{ context.query.repo_name }}`
	`338`	`+Commit: {{ context.query.commit }}`
	`339`	`+Files analyzed: {{ context.sources[0].file_count }}`
`338`	`340`	`"""`
`339`	`341`	`summary_template = self.env.from_string(template)`
`340`	`342`	`return summary_template.render(context=context, query=query)`

Lines changed: 1 addition & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -70,6 +70,7 @@ class FileSystemDirectory(FileSystemNode):`
`70`	`70`	`children: list[FileSystemNode] = field(default_factory=list)`
`71`	`71`	`file_count: int = 0`
`72`	`72`	`dir_count: int = 0`
	`73`	`+ file_count_total: int = 0`
`73`	`74`	`type: FileSystemNodeType = FileSystemNodeType.DIRECTORY`
`74`	`75`
`75`	`76`	`def sort_children(self) -> None:`

Lines changed: 1 addition & 1 deletion

Original file line number	Diff line number	Diff line change
`@@ -8,7 +8,7 @@`
`8`	`8`
`9`	`9`	`from gitingest.clone import clone_repo`
`10`	`10`	`from gitingest.ingestion import ingest_query`
`11`		`-from gitingest.output_formatter import DefaultFormatter, SummaryFormatter`
	`11`	`+from gitingest.output_formatter import DebugFormatter, DefaultFormatter, SummaryFormatter`
`12`	`12`	`from gitingest.query_parser import parse_remote_repo`
`13`	`13`	`from gitingest.utils.git_utils import resolve_commit, validate_github_token`
`14`	`14`	`from gitingest.utils.logging_config import get_logger`

Lines changed: 36 additions & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -0,0 +1,36 @@`
	`1`	`+from pathlib import Path`
	`2`	`+from unittest.mock import Mock`
	`3`	`+from gitingest.output_formatter import DefaultFormatter`
	`4`	`+from gitingest.schemas.filesystem import Context, GitRepository, FileSystemFile`
	`5`	`+`
	`6`	`+# Create a mock query`
	`7`	`+mock_query = Mock()`
	`8`	`+mock_query.user_name = "test_user"`
	`9`	`+mock_query.repo_name = "test_repo"`
	`10`	`+`
	`11`	`+# Create a simple file`
	`12`	`+mock_file = FileSystemFile(`
	`13`	`+ name="test.py",`
	`14`	`+ path_str="test.py",`
	`15`	`+ path=Path("test.py"),`
	`16`	`+)`
	`17`	`+mock_file.content = "print('hello world')"`
	`18`	`+`
	`19`	`+# Create a git repository with the file`
	`20`	`+mock_repo = GitRepository(`
	`21`	`+ name="test_repo",`
	`22`	`+ path_str="",`
	`23`	`+ path=Path("."),`
	`24`	`+ children=[mock_file]`
	`25`	`+)`
	`26`	`+`
	`27`	`+# Create context`
	`28`	`+context = Context([mock_repo], mock_query)`
	`29`	`+`
	`30`	`+# Test formatting`
	`31`	`+formatter = DefaultFormatter()`
	`32`	`+result = formatter.format(context, mock_query)`
	`33`	`+print("RESULT:")`
	`34`	`+print(repr(result))`
	`35`	`+print("\nFORMATTED:")`
	`36`	`+print(result)`

Comments

(0)