2026-06-10 07:09:11 +02:00
parent dc35579135
commit eb0437b0ed
7 changed files with 233 additions and 0 deletions
							
							
							
						
@@ -0,0 +1,68 @@
"""Feature D — KI-Test-Generierung (Vorschlag als PR-Kommentar).
Per Slash-Command `/gen-tests` an einem PR (issue_comment) ODER aus einem
pull_request-Event. Holt den Diff, lässt das LLM passende Unit-Tests vorschlagen
und postet sie als idempotenten Kommentar. Bewusst NUR Vorschlag — kein
Auto-Commit (der Mensch entscheidet, was übernommen wird).
"""
import sys
import context
import gitea_api
import llm
MARKER = "<!-- ci-ai:tests -->"
DIFF_CAP = 30000
SYSTEM = (
    "Du bist ein erfahrener Test-Engineer. Schlage sinnvolle, lauffähige Unit-Tests "
    "für die geänderten Stellen vor. Erkenne Sprache und Test-Framework aus dem Diff. "
    "Antworte auf Deutsch, Test-Code in passenden Code-Blöcken. Keine Tests für reine "
    "Trivialitäten; erfinde keine nicht vorhandenen APIs."
)
PROMPT = """\
Generiere Unit-Tests für die Änderungen in diesem Pull-Request.
PR-Titel: {title}
Pro relevanter Datei/Funktion: ein kurzer Hinweis + der Test-Code (richtige Sprache,
lauffähig, Framework benennen). Wenn nichts sinnvoll testbar ist, sag das ehrlich.
--- DIFF ---
{diff}
"""
def _pr_index():
    pr = context.pull_request()
    if pr and pr.get("index"):
        return pr["index"], pr.get("title", "")
    iss = context.issue()
    if iss and iss.get("is_pr"):
        return iss["index"], iss.get("title", "")
    return None, ""
def main() -> int:
    owner, repo = context.repo_slug()
    index, title = _pr_index()
    if not index:
        print("[ai_tests] Kein PR-Kontext — Test-Generierung läuft nur auf PRs. Übersprungen.")
        return 0
    ok, why = llm.configured()
    if not ok:
        print(f"[ai_tests] KI inaktiv ({why}) — übersprungen, kein Fehler.")
        return 0
    diff = gitea_api.get_pr_diff(owner, repo, index)
    if not diff.strip():
        print("[ai_tests] Leerer Diff — nichts zu testen.")
        return 0
    answer = llm.complete(SYSTEM, PROMPT.format(title=title, diff=diff[:DIFF_CAP]), max_tokens=2000)
    gitea_api.upsert_comment(owner, repo, index, MARKER, f"## 🧪 KI-Test-Vorschläge\n\n{answer}")
    print("[ai_tests] Kommentar gepostet/aktualisiert.")
    return 0
if __name__ == "__main__":
    sys.exit(main())