Compare commits
1 Commits
burn/118-1
...
feat/sessi
| Author | SHA1 | Date | |
|---|---|---|---|
| 160dfcf419 |
@@ -1,239 +0,0 @@
|
||||
"""
|
||||
Diff Analyzer — Pipeline 6.1
|
||||
|
||||
Reads PR diffs and categorizes changes: new code, deleted code, modified code, moved code.
|
||||
Produces a change summary with line counts per category.
|
||||
|
||||
Usage:
|
||||
from diff_analyzer import DiffAnalyzer
|
||||
analyzer = DiffAnalyzer()
|
||||
summary = analyzer.analyze(diff_text)
|
||||
"""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
import re
|
||||
from dataclasses import dataclass, field
|
||||
from enum import Enum
|
||||
from typing import List, Optional
|
||||
|
||||
|
||||
class ChangeCategory(Enum):
|
||||
"""Categories of code changes in a diff hunk."""
|
||||
ADDED = "added"
|
||||
DELETED = "deleted"
|
||||
MODIFIED = "modified"
|
||||
MOVED = "moved"
|
||||
CONTEXT = "context"
|
||||
|
||||
|
||||
@dataclass
|
||||
class Hunk:
|
||||
"""A single diff hunk with metadata."""
|
||||
header: str
|
||||
old_start: int
|
||||
old_count: int
|
||||
new_start: int
|
||||
new_count: int
|
||||
lines: List[str] = field(default_factory=list)
|
||||
category: ChangeCategory = ChangeCategory.CONTEXT
|
||||
old_lines: int = 0
|
||||
new_lines: int = 0
|
||||
|
||||
|
||||
@dataclass
|
||||
class FileChange:
|
||||
"""Changes within a single file."""
|
||||
path: str
|
||||
old_path: Optional[str] = None # For renames
|
||||
is_new: bool = False
|
||||
is_deleted: bool = False
|
||||
is_renamed: bool = False
|
||||
hunks: List[Hunk] = field(default_factory=list)
|
||||
added_lines: int = 0
|
||||
deleted_lines: int = 0
|
||||
context_lines: int = 0
|
||||
|
||||
|
||||
@dataclass
|
||||
class ChangeSummary:
|
||||
"""Summary of all changes in a diff."""
|
||||
files_changed: int = 0
|
||||
files_added: int = 0
|
||||
files_deleted: int = 0
|
||||
files_renamed: int = 0
|
||||
files_modified: int = 0
|
||||
total_added: int = 0
|
||||
total_deleted: int = 0
|
||||
total_context: int = 0
|
||||
hunks_added: int = 0
|
||||
hunks_deleted: int = 0
|
||||
hunks_modified: int = 0
|
||||
hunks_moved: int = 0
|
||||
file_changes: List[FileChange] = field(default_factory=list)
|
||||
|
||||
def to_dict(self) -> dict:
|
||||
"""Serialize to dict for JSON output."""
|
||||
return {
|
||||
"files_changed": self.files_changed,
|
||||
"files_added": self.files_added,
|
||||
"files_deleted": self.files_deleted,
|
||||
"files_renamed": self.files_renamed,
|
||||
"files_modified": self.files_modified,
|
||||
"total_added": self.total_added,
|
||||
"total_deleted": self.total_deleted,
|
||||
"total_context": self.total_context,
|
||||
"hunks_added": self.hunks_added,
|
||||
"hunks_deleted": self.hunks_deleted,
|
||||
"hunks_modified": self.hunks_modified,
|
||||
"hunks_moved": self.hunks_moved,
|
||||
"files": [
|
||||
{
|
||||
"path": fc.path,
|
||||
"old_path": fc.old_path,
|
||||
"is_new": fc.is_new,
|
||||
"is_deleted": fc.is_deleted,
|
||||
"is_renamed": fc.is_renamed,
|
||||
"added": fc.added_lines,
|
||||
"deleted": fc.deleted_lines,
|
||||
"context": fc.context_lines,
|
||||
}
|
||||
for fc in self.file_changes
|
||||
],
|
||||
}
|
||||
|
||||
|
||||
# Regex for unified diff headers
|
||||
_HUNK_RE = re.compile(
|
||||
r"^@@\s+-(\d+)(?:,(\d+))?\s+\+(\d+)(?:,(\d+))?\s+@@(.*)$"
|
||||
)
|
||||
_FILE_HEADER_RE = re.compile(r"^diff --git a/(.*) b/(.*)$")
|
||||
_RENAME_RE = re.compile(r"^rename from (.+)$|^rename to (.+)$")
|
||||
|
||||
|
||||
class DiffAnalyzer:
|
||||
"""Parses unified diffs and categorizes changes."""
|
||||
|
||||
def analyze(self, diff_text: str) -> ChangeSummary:
|
||||
"""Analyze a unified diff string and return a ChangeSummary."""
|
||||
summary = ChangeSummary()
|
||||
lines = diff_text.splitlines(keepends=False)
|
||||
|
||||
current_file: Optional[FileChange] = None
|
||||
current_hunk: Optional[Hunk] = None
|
||||
old_path: Optional[str] = None
|
||||
new_path: Optional[str] = None
|
||||
|
||||
for line in lines:
|
||||
# File header
|
||||
m = _FILE_HEADER_RE.match(line)
|
||||
if m:
|
||||
# Save previous file
|
||||
if current_file:
|
||||
self._classify_file(current_file)
|
||||
summary.file_changes.append(current_file)
|
||||
|
||||
old_path = m.group(1)
|
||||
new_path = m.group(2)
|
||||
current_file = FileChange(path=new_path, old_path=old_path)
|
||||
current_hunk = None
|
||||
continue
|
||||
|
||||
if current_file is None:
|
||||
continue
|
||||
|
||||
# Detect new/deleted file markers
|
||||
if line.startswith("new file mode"):
|
||||
current_file.is_new = True
|
||||
continue
|
||||
if line.startswith("deleted file mode"):
|
||||
current_file.is_deleted = True
|
||||
continue
|
||||
|
||||
# Detect renames
|
||||
rm = _RENAME_RE.match(line)
|
||||
if rm:
|
||||
current_file.is_renamed = True
|
||||
continue
|
||||
|
||||
# Hunk header
|
||||
hm = _HUNK_RE.match(line)
|
||||
if hm:
|
||||
if current_hunk:
|
||||
self._classify_hunk(current_hunk)
|
||||
current_file.hunks.append(current_hunk)
|
||||
|
||||
current_hunk = Hunk(
|
||||
header=line,
|
||||
old_start=int(hm.group(1)),
|
||||
old_count=int(hm.group(2) or 1),
|
||||
new_start=int(hm.group(3)),
|
||||
new_count=int(hm.group(4) or 1),
|
||||
)
|
||||
continue
|
||||
|
||||
if current_hunk is None:
|
||||
continue
|
||||
|
||||
# Hunk content
|
||||
current_hunk.lines.append(line)
|
||||
|
||||
if line.startswith("+"):
|
||||
current_hunk.new_lines += 1
|
||||
current_file.added_lines += 1
|
||||
elif line.startswith("-"):
|
||||
current_hunk.old_lines += 1
|
||||
current_file.deleted_lines += 1
|
||||
elif line.startswith(" "):
|
||||
current_file.context_lines += 1
|
||||
|
||||
# Finalize last hunk and file
|
||||
if current_hunk:
|
||||
self._classify_hunk(current_hunk)
|
||||
if current_file:
|
||||
current_file.hunks.append(current_hunk)
|
||||
if current_file:
|
||||
self._classify_file(current_file)
|
||||
summary.file_changes.append(current_file)
|
||||
|
||||
# Aggregate
|
||||
summary.files_changed = len(summary.file_changes)
|
||||
for fc in summary.file_changes:
|
||||
summary.total_added += fc.added_lines
|
||||
summary.total_deleted += fc.deleted_lines
|
||||
summary.total_context += fc.context_lines
|
||||
if fc.is_new:
|
||||
summary.files_added += 1
|
||||
elif fc.is_deleted:
|
||||
summary.files_deleted += 1
|
||||
elif fc.is_renamed:
|
||||
summary.files_renamed += 1
|
||||
else:
|
||||
summary.files_modified += 1
|
||||
|
||||
for h in fc.hunks:
|
||||
if h.category == ChangeCategory.ADDED:
|
||||
summary.hunks_added += 1
|
||||
elif h.category == ChangeCategory.DELETED:
|
||||
summary.hunks_deleted += 1
|
||||
elif h.category == ChangeCategory.MODIFIED:
|
||||
summary.hunks_modified += 1
|
||||
elif h.category == ChangeCategory.MOVED:
|
||||
summary.hunks_moved += 1
|
||||
|
||||
return summary
|
||||
|
||||
def _classify_hunk(self, hunk: Hunk) -> None:
|
||||
"""Classify a hunk based on its add/delete ratio."""
|
||||
if hunk.new_lines > 0 and hunk.old_lines == 0:
|
||||
hunk.category = ChangeCategory.ADDED
|
||||
elif hunk.old_lines > 0 and hunk.new_lines == 0:
|
||||
hunk.category = ChangeCategory.DELETED
|
||||
elif hunk.new_lines > 0 and hunk.old_lines > 0:
|
||||
hunk.category = ChangeCategory.MODIFIED
|
||||
else:
|
||||
hunk.category = ChangeCategory.CONTEXT
|
||||
|
||||
def _classify_file(self, fc: FileChange) -> None:
|
||||
"""Final file classification (renames already detected via headers)."""
|
||||
pass
|
||||
276
scripts/session_metadata.py
Normal file
276
scripts/session_metadata.py
Normal file
@@ -0,0 +1,276 @@
|
||||
#!/usr/bin/env python3
|
||||
"""
|
||||
session_metadata.py - Extract structured metadata from Hermes session transcripts.
|
||||
Works alongside session_reader.py to provide higher-level session analysis.
|
||||
"""
|
||||
|
||||
import json
|
||||
import re
|
||||
import sys
|
||||
from dataclasses import dataclass, asdict
|
||||
from datetime import datetime
|
||||
from pathlib import Path
|
||||
from typing import Dict, List, Optional, Any
|
||||
|
||||
# Import from session_reader (the canonical reader)
|
||||
from session_reader import read_session
|
||||
|
||||
|
||||
@dataclass
|
||||
class SessionSummary:
|
||||
"""Structured summary of a Hermes session transcript."""
|
||||
session_id: str
|
||||
model: str
|
||||
repo: str
|
||||
outcome: str
|
||||
message_count: int
|
||||
tool_calls: int
|
||||
duration_estimate: str
|
||||
key_actions: List[str]
|
||||
errors_encountered: List[str]
|
||||
start_time: Optional[str] = None
|
||||
end_time: Optional[str] = None
|
||||
total_tokens_estimate: int = 0
|
||||
user_messages: int = 0
|
||||
assistant_messages: int = 0
|
||||
tool_outputs: int = 0
|
||||
|
||||
|
||||
def extract_session_metadata(file_path: str) -> SessionSummary:
|
||||
"""
|
||||
Extract structured metadata from a Hermes session JSONL transcript.
|
||||
Uses session_reader.read_session() for file reading.
|
||||
"""
|
||||
session_id = Path(file_path).stem
|
||||
messages = []
|
||||
model = "unknown"
|
||||
repo = "unknown"
|
||||
tool_calls_count = 0
|
||||
key_actions = []
|
||||
errors = []
|
||||
start_time = None
|
||||
end_time = None
|
||||
total_tokens = 0
|
||||
|
||||
# Common repo patterns to look for
|
||||
repo_patterns = [
|
||||
r"(?:the-nexus|compounding-intelligence|timmy-config|hermes-agent)",
|
||||
r"(?:forge\.alexanderwhitestone\.com/([^/]+/[^/\\s]+))",
|
||||
r"(?:github\.com/([^/]+/[^/\\s]+))",
|
||||
r"(?:Timmy_Foundation/([^/\\s]+))",
|
||||
]
|
||||
|
||||
try:
|
||||
# Use the canonical reader from session_reader.py
|
||||
messages = read_session(file_path)
|
||||
except FileNotFoundError:
|
||||
return SessionSummary(
|
||||
session_id=session_id,
|
||||
model="unknown",
|
||||
repo="unknown",
|
||||
outcome="failure",
|
||||
message_count=0,
|
||||
tool_calls=0,
|
||||
duration_estimate="0m",
|
||||
key_actions=[],
|
||||
errors_encountered=[f"File not found: {file_path}"]
|
||||
)
|
||||
|
||||
# Process messages for metadata
|
||||
for entry in messages:
|
||||
# Extract model from assistant messages
|
||||
if entry.get("role") == "assistant" and entry.get("model"):
|
||||
model = entry["model"]
|
||||
|
||||
# Extract timestamps
|
||||
if entry.get("timestamp"):
|
||||
ts = entry["timestamp"]
|
||||
if start_time is None:
|
||||
start_time = ts
|
||||
end_time = ts
|
||||
|
||||
# Count tool calls
|
||||
if entry.get("tool_calls"):
|
||||
tool_calls_count += len(entry["tool_calls"])
|
||||
for tc in entry["tool_calls"]:
|
||||
if tc.get("function", {}).get("name"):
|
||||
action = f"{tc['function']['name']}"
|
||||
if action not in key_actions:
|
||||
key_actions.append(action)
|
||||
|
||||
# Estimate tokens from content length
|
||||
content = entry.get("content", "")
|
||||
if isinstance(content, str):
|
||||
total_tokens += len(content.split())
|
||||
elif isinstance(content, list):
|
||||
for item in content:
|
||||
if isinstance(item, dict) and "text" in item:
|
||||
total_tokens += len(item["text"].split())
|
||||
|
||||
# Look for repo mentions in content
|
||||
if entry.get("content"):
|
||||
content_str = str(entry["content"])
|
||||
for pattern in repo_patterns:
|
||||
match = re.search(pattern, content_str, re.IGNORECASE)
|
||||
if match:
|
||||
if match.groups():
|
||||
repo = match.group(1)
|
||||
else:
|
||||
repo = match.group(0)
|
||||
break
|
||||
|
||||
# Look for error messages
|
||||
if entry.get("role") == "tool" and entry.get("is_error"):
|
||||
error_msg = entry.get("content", "Unknown error")
|
||||
if isinstance(error_msg, str) and len(error_msg) < 200:
|
||||
errors.append(error_msg[:200])
|
||||
|
||||
# Count message types
|
||||
user_messages = sum(1 for m in messages if m.get("role") == "user")
|
||||
assistant_messages = sum(1 for m in messages if m.get("role") == "assistant")
|
||||
tool_outputs = sum(1 for m in messages if m.get("role") == "tool")
|
||||
|
||||
# Calculate duration estimate
|
||||
duration_estimate = "unknown"
|
||||
if start_time and end_time:
|
||||
try:
|
||||
# Try to parse timestamps
|
||||
start_dt = None
|
||||
end_dt = None
|
||||
|
||||
# Handle various timestamp formats
|
||||
for fmt in ["%Y-%m-%dT%H:%M:%S.%fZ", "%Y-%m-%dT%H:%M:%SZ", "%Y-%m-%d %H:%M:%S"]:
|
||||
try:
|
||||
if start_dt is None:
|
||||
start_dt = datetime.strptime(start_time, fmt)
|
||||
if end_dt is None:
|
||||
end_dt = datetime.strptime(end_time, fmt)
|
||||
except ValueError:
|
||||
continue
|
||||
|
||||
if start_dt and end_dt:
|
||||
duration = end_dt - start_dt
|
||||
minutes = duration.total_seconds() / 60
|
||||
duration_estimate = f"{minutes:.0f}m"
|
||||
except Exception:
|
||||
pass
|
||||
|
||||
# Classify outcome
|
||||
outcome = "unknown"
|
||||
if errors:
|
||||
# Check if any errors are fatal
|
||||
fatal_errors = any("405" in e or "permission" in e.lower() or "authentication" in e.lower()
|
||||
for e in errors)
|
||||
if fatal_errors:
|
||||
outcome = "failure"
|
||||
else:
|
||||
outcome = "partial"
|
||||
elif messages:
|
||||
# Check last message for success indicators
|
||||
last_msg = messages[-1]
|
||||
if last_msg.get("role") == "assistant":
|
||||
content = last_msg.get("content", "")
|
||||
if isinstance(content, str):
|
||||
success_indicators = ["done", "completed", "success", "merged", "pushed"]
|
||||
if any(indicator in content.lower() for indicator in success_indicators):
|
||||
outcome = "success"
|
||||
else:
|
||||
outcome = "unknown"
|
||||
|
||||
# Deduplicate key actions (keep unique, limit to 10)
|
||||
unique_actions = []
|
||||
for action in key_actions:
|
||||
if action not in unique_actions:
|
||||
unique_actions.append(action)
|
||||
if len(unique_actions) >= 10:
|
||||
break
|
||||
|
||||
# Deduplicate errors (keep unique, limit to 5)
|
||||
unique_errors = []
|
||||
for error in errors:
|
||||
if error not in unique_errors:
|
||||
unique_errors.append(error)
|
||||
if len(unique_errors) >= 5:
|
||||
break
|
||||
|
||||
return SessionSummary(
|
||||
session_id=session_id,
|
||||
model=model,
|
||||
repo=repo,
|
||||
outcome=outcome,
|
||||
message_count=len(messages),
|
||||
tool_calls=tool_calls_count,
|
||||
duration_estimate=duration_estimate,
|
||||
key_actions=unique_actions,
|
||||
errors_encountered=unique_errors,
|
||||
start_time=start_time,
|
||||
end_time=end_time,
|
||||
total_tokens_estimate=total_tokens,
|
||||
user_messages=user_messages,
|
||||
assistant_messages=assistant_messages,
|
||||
tool_outputs=tool_outputs
|
||||
)
|
||||
|
||||
|
||||
def process_session_directory(directory_path: str, output_file: Optional[str] = None) -> List[SessionSummary]:
|
||||
"""
|
||||
Process all JSONL files in a directory.
|
||||
"""
|
||||
directory = Path(directory_path)
|
||||
if not directory.exists():
|
||||
print(f"Error: Directory {directory_path} does not exist", file=sys.stderr)
|
||||
return []
|
||||
|
||||
jsonl_files = list(directory.glob("*.jsonl"))
|
||||
if not jsonl_files:
|
||||
print(f"Warning: No JSONL files found in {directory_path}", file=sys.stderr)
|
||||
return []
|
||||
|
||||
summaries = []
|
||||
for jsonl_file in sorted(jsonl_files):
|
||||
print(f"Processing {jsonl_file.name}...", file=sys.stderr)
|
||||
summary = extract_session_metadata(str(jsonl_file))
|
||||
summaries.append(summary)
|
||||
|
||||
if output_file:
|
||||
with open(output_file, 'w', encoding='utf-8') as f:
|
||||
json.dump([asdict(s) for s in summaries], f, indent=2)
|
||||
print(f"Wrote {len(summaries)} summaries to {output_file}", file=sys.stderr)
|
||||
|
||||
return summaries
|
||||
|
||||
|
||||
def main():
|
||||
"""CLI entry point."""
|
||||
import argparse
|
||||
|
||||
parser = argparse.ArgumentParser(description="Extract metadata from Hermes session JSONL transcripts")
|
||||
parser.add_argument("path", help="Path to JSONL file or directory of session files")
|
||||
parser.add_argument("-o", "--output", help="Output JSON file (default: stdout)")
|
||||
parser.add_argument("-v", "--verbose", action="store_true", help="Verbose output")
|
||||
|
||||
args = parser.parse_args()
|
||||
|
||||
path = Path(args.path)
|
||||
|
||||
if path.is_file():
|
||||
summary = extract_session_metadata(str(path))
|
||||
if args.output:
|
||||
with open(args.output, 'w') as f:
|
||||
json.dump(asdict(summary), f, indent=2)
|
||||
print(f"Wrote summary to {args.output}", file=sys.stderr)
|
||||
else:
|
||||
print(json.dumps(asdict(summary), indent=2))
|
||||
|
||||
elif path.is_dir():
|
||||
summaries = process_session_directory(str(path), args.output)
|
||||
if not args.output:
|
||||
print(json.dumps([asdict(s) for s in summaries], indent=2))
|
||||
|
||||
else:
|
||||
print(f"Error: {args.path} is not a file or directory", file=sys.stderr)
|
||||
sys.exit(1)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
@@ -1,224 +0,0 @@
|
||||
"""Tests for diff_analyzer module."""
|
||||
|
||||
import sys
|
||||
import os
|
||||
sys.path.insert(0, os.path.join(os.path.dirname(__file__), '..', 'scripts'))
|
||||
|
||||
from diff_analyzer import DiffAnalyzer, ChangeCategory
|
||||
|
||||
|
||||
def test_parse_simple_addition():
|
||||
diff = """diff --git a/foo.py b/foo.py
|
||||
new file mode 100644
|
||||
--- /dev/null
|
||||
+++ b/foo.py
|
||||
@@ -0,0 +1,3 @@
|
||||
+def hello():
|
||||
+ return "world"
|
||||
+# end
|
||||
"""
|
||||
analyzer = DiffAnalyzer()
|
||||
summary = analyzer.analyze(diff)
|
||||
|
||||
assert summary.files_changed == 1
|
||||
assert summary.files_added == 1
|
||||
assert summary.files_modified == 0
|
||||
assert summary.total_added == 3
|
||||
assert summary.total_deleted == 0
|
||||
assert summary.hunks_added == 1
|
||||
assert len(summary.file_changes) == 1
|
||||
assert summary.file_changes[0].is_new is True
|
||||
assert summary.file_changes[0].path == "foo.py"
|
||||
|
||||
|
||||
def test_parse_simple_deletion():
|
||||
diff = """diff --git a/old.py b/old.py
|
||||
deleted file mode 100644
|
||||
--- a/old.py
|
||||
+++ /dev/null
|
||||
@@ -1,2 +0,0 @@
|
||||
-x = 1
|
||||
-y = 2
|
||||
"""
|
||||
analyzer = DiffAnalyzer()
|
||||
summary = analyzer.analyze(diff)
|
||||
|
||||
assert summary.files_changed == 1
|
||||
assert summary.files_deleted == 1
|
||||
assert summary.total_deleted == 2
|
||||
assert summary.total_added == 0
|
||||
assert summary.hunks_deleted == 1
|
||||
assert summary.file_changes[0].is_deleted is True
|
||||
|
||||
|
||||
def test_parse_modification():
|
||||
diff = """diff --git a/bar.py b/bar.py
|
||||
--- a/bar.py
|
||||
+++ b/bar.py
|
||||
@@ -10,3 +10,4 @@ def foo():
|
||||
existing()
|
||||
- old_call()
|
||||
+ new_call()
|
||||
+ extra_step()
|
||||
return
|
||||
"""
|
||||
analyzer = DiffAnalyzer()
|
||||
summary = analyzer.analyze(diff)
|
||||
|
||||
assert summary.files_changed == 1
|
||||
assert summary.files_modified == 1
|
||||
assert summary.total_added == 2 # +new_call(), +extra_step()
|
||||
assert summary.total_deleted == 1 # -old_call()
|
||||
assert summary.total_context == 2 # 2 context lines
|
||||
assert summary.hunks_modified == 1
|
||||
|
||||
|
||||
def test_parse_multiple_files():
|
||||
diff = """diff --git a/a.py b/a.py
|
||||
--- a/a.py
|
||||
+++ b/a.py
|
||||
@@ -1,1 +1,2 @@
|
||||
existing
|
||||
+added
|
||||
diff --git a/b.py b/b.py
|
||||
new file mode 100644
|
||||
--- /dev/null
|
||||
+++ b/b.py
|
||||
@@ -0,0 +1,1 @@
|
||||
+new file
|
||||
diff --git a/c.py b/c.py
|
||||
deleted file mode 100644
|
||||
--- a/c.py
|
||||
+++ /dev/null
|
||||
@@ -1,1 +0,0 @@
|
||||
-gone
|
||||
"""
|
||||
analyzer = DiffAnalyzer()
|
||||
summary = analyzer.analyze(diff)
|
||||
|
||||
assert summary.files_changed == 3
|
||||
assert summary.files_added == 1
|
||||
assert summary.files_deleted == 1
|
||||
assert summary.files_modified == 1
|
||||
assert summary.total_added == 2
|
||||
assert summary.total_deleted == 1
|
||||
|
||||
|
||||
def test_parse_rename():
|
||||
diff = """diff --git a/old_name.py b/new_name.py
|
||||
rename from old_name.py
|
||||
rename to new_name.py
|
||||
--- a/old_name.py
|
||||
+++ b/new_name.py
|
||||
@@ -1,1 +1,1 @@
|
||||
-old_func()
|
||||
+new_func()
|
||||
"""
|
||||
analyzer = DiffAnalyzer()
|
||||
summary = analyzer.analyze(diff)
|
||||
|
||||
assert summary.files_changed == 1
|
||||
assert summary.files_renamed == 1
|
||||
assert summary.file_changes[0].is_renamed is True
|
||||
assert summary.file_changes[0].old_path == "old_name.py"
|
||||
assert summary.file_changes[0].path == "new_name.py"
|
||||
|
||||
|
||||
def test_parse_mixed_hunks():
|
||||
"""A file with one add hunk and one delete hunk."""
|
||||
diff = """diff --git a/mixed.py b/mixed.py
|
||||
--- a/mixed.py
|
||||
+++ b/mixed.py
|
||||
@@ -5,0 +6,2 @@
|
||||
+new_line_1
|
||||
+new_line_2
|
||||
@@ -20,2 +22,0 @@
|
||||
-removed_1
|
||||
-removed_2
|
||||
"""
|
||||
analyzer = DiffAnalyzer()
|
||||
summary = analyzer.analyze(diff)
|
||||
|
||||
assert summary.files_changed == 1
|
||||
assert summary.hunks_added == 1
|
||||
assert summary.hunks_deleted == 1
|
||||
assert summary.total_added == 2
|
||||
assert summary.total_deleted == 2
|
||||
|
||||
|
||||
def test_empty_diff():
|
||||
analyzer = DiffAnalyzer()
|
||||
summary = analyzer.analyze("")
|
||||
|
||||
assert summary.files_changed == 0
|
||||
assert summary.total_added == 0
|
||||
assert summary.total_deleted == 0
|
||||
|
||||
|
||||
def test_to_dict():
|
||||
diff = """diff --git a/test.py b/test.py
|
||||
new file mode 100644
|
||||
--- /dev/null
|
||||
+++ b/test.py
|
||||
@@ -0,0 +1,2 @@
|
||||
+line1
|
||||
+line2
|
||||
"""
|
||||
analyzer = DiffAnalyzer()
|
||||
summary = analyzer.analyze(diff)
|
||||
d = summary.to_dict()
|
||||
|
||||
assert d["files_changed"] == 1
|
||||
assert d["files_added"] == 1
|
||||
assert d["total_added"] == 2
|
||||
assert d["total_deleted"] == 0
|
||||
assert len(d["files"]) == 1
|
||||
assert d["files"][0]["path"] == "test.py"
|
||||
assert d["files"][0]["is_new"] is True
|
||||
|
||||
|
||||
def test_context_only_hunk():
|
||||
"""A hunk with only context lines (rare but possible)."""
|
||||
diff = """diff --git a/noop.py b/noop.py
|
||||
--- a/noop.py
|
||||
+++ b/noop.py
|
||||
@@ -5,3 +5,3 @@
|
||||
context1
|
||||
context2
|
||||
context3
|
||||
"""
|
||||
analyzer = DiffAnalyzer()
|
||||
summary = analyzer.analyze(diff)
|
||||
|
||||
assert summary.total_context == 3
|
||||
assert summary.total_added == 0
|
||||
assert summary.total_deleted == 0
|
||||
|
||||
|
||||
def test_binary_files_skipped():
|
||||
"""Binary file diffs have no content lines — just headers."""
|
||||
diff = """diff --git a/image.png b/image.png
|
||||
--- a/image.png
|
||||
+++ b/image.png
|
||||
Binary files a/image.png and b/image.png differ
|
||||
"""
|
||||
analyzer = DiffAnalyzer()
|
||||
summary = analyzer.analyze(diff)
|
||||
|
||||
assert summary.files_changed == 1
|
||||
assert summary.total_added == 0
|
||||
assert summary.total_deleted == 0
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
test_parse_simple_addition()
|
||||
test_parse_simple_deletion()
|
||||
test_parse_modification()
|
||||
test_parse_multiple_files()
|
||||
test_parse_rename()
|
||||
test_parse_mixed_hunks()
|
||||
test_empty_diff()
|
||||
test_to_dict()
|
||||
test_context_only_hunk()
|
||||
test_binary_files_skipped()
|
||||
print("All 10 tests passed.")
|
||||
Reference in New Issue
Block a user