Commit 77f3547

committed

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

1 parent 1e8547e commit 77f3547Copy full SHA for 77f3547

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
`@@ -295,7 +295,7 @@ def summary(self, node: Source, query):`
`295`	`295`
`296`	`296`	`@summary.register`
`297`	`297`	`def _(self, node: FileSystemDirectory, query):`
`298`		`- template = """`
	`298`	`+ template = """\`
`299`	`299`	`Directory structure:`
`300`	`300`	`{{ node.tree }}`
`301`	`301`	`"""`
`@@ -305,7 +305,9 @@ def _(self, node: FileSystemDirectory, query):`
`305`	`305`	`@summary.register`
`306`	`306`	`def _(self, context: Context, query):`
`307`	`307`	`template = """`
`308`		`-{{ context.summary }}`
	`308`	`+Repository: {{ context.query.user_name }}/{{ context.query.repo_name }}`
	`309`	`+Commit: {{ context.query.commit }}`
	`310`	`+Files analyzed: {{ context.sources[0].file_count }}`
`309`	`311`	`"""`
`310`	`312`	`summary_template = self.env.from_string(template)`
`311`	`313`	`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
`@@ -7,7 +7,7 @@`
`7`	`7`
`8`	`8`	`from gitingest.clone import clone_repo`
`9`	`9`	`from gitingest.ingestion import ingest_query`
`10`		`-from gitingest.output_formatter import DefaultFormatter, SummaryFormatter`
	`10`	`+from gitingest.output_formatter import DebugFormatter, DefaultFormatter, SummaryFormatter`
`11`	`11`	`from gitingest.query_parser import parse_remote_repo`
`12`	`12`	`from gitingest.utils.git_utils import validate_github_token`
`13`	`13`	`from gitingest.utils.pattern_utils import process_patterns`

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)