commitbee 0.5.0

AI-powered commit message generator using tree-sitter semantic analysis and local LLMs
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144

// SPDX-FileCopyrightText: 2026 Sephyi <me@sephy.io>
//
// SPDX-License-Identifier: AGPL-3.0-only OR LicenseRef-Commercial

//! Tests for the template engine (custom prompt templates).

use std::collections::HashMap;
use std::path::PathBuf;

use commitbee::services::template;

// ─── Template loading and variable substitution ─────────────────────────────

#[test]
fn render_template_substitutes_variables() {
    let dir = tempfile::tempdir().unwrap();
    let path = dir.path().join("template.txt");
    std::fs::write(
        &path,
        "Diff:\n{{diff}}\n\nFiles: {{files}}\nSymbols: {{symbols}}",
    )
    .unwrap();

    let mut vars = HashMap::new();
    vars.insert("diff", "+fn foo() {}");
    vars.insert("files", "src/main.rs");
    vars.insert("symbols", "Function foo (added)");

    let result = template::render_template(&path, &vars).unwrap();

    assert_eq!(
        result,
        "Diff:\n+fn foo() {}\n\nFiles: src/main.rs\nSymbols: Function foo (added)"
    );
}

// ─── Missing file produces clear error ──────────────────────────────────────

#[test]
fn render_template_missing_file_returns_error() {
    let path = PathBuf::from("/nonexistent/template.txt");
    let vars = HashMap::new();

    let result = template::render_template(&path, &vars);

    assert!(result.is_err(), "expected error for missing file");
    let err_msg = result.unwrap_err().to_string();
    assert!(
        err_msg.contains("failed to read template file"),
        "expected descriptive error, got: {err_msg}"
    );
    assert!(
        err_msg.contains("/nonexistent/template.txt"),
        "expected path in error, got: {err_msg}"
    );
}

// ─── Unknown variables left as-is ───────────────────────────────────────────

#[test]
fn render_template_unknown_variables_left_intact() {
    let dir = tempfile::tempdir().unwrap();
    let path = dir.path().join("template.txt");
    std::fs::write(&path, "Known: {{diff}}\nUnknown: {{future_var}}").unwrap();

    let mut vars = HashMap::new();
    vars.insert("diff", "the diff");

    let result = template::render_template(&path, &vars).unwrap();

    assert_eq!(result, "Known: the diff\nUnknown: {{future_var}}");
}

// ─── Empty template file works ──────────────────────────────────────────────

#[test]
fn render_template_empty_file_returns_empty_string() {
    let dir = tempfile::tempdir().unwrap();
    let path = dir.path().join("empty.txt");
    std::fs::write(&path, "").unwrap();

    let vars = HashMap::new();

    let result = template::render_template(&path, &vars).unwrap();
    assert_eq!(result, "");
}

// ─── load_file works ────────────────────────────────────────────────────────

#[test]
fn load_file_reads_content() {
    let dir = tempfile::tempdir().unwrap();
    let path = dir.path().join("prompt.txt");
    std::fs::write(&path, "You are a helpful assistant.").unwrap();

    let result = template::load_file(&path).unwrap();
    assert_eq!(result, "You are a helpful assistant.");
}

#[test]
fn load_file_missing_returns_error() {
    let path = PathBuf::from("/nonexistent/prompt.txt");

    let result = template::load_file(&path);

    assert!(result.is_err(), "expected error for missing file");
    let err_msg = result.unwrap_err().to_string();
    assert!(
        err_msg.contains("failed to read prompt file"),
        "expected descriptive error, got: {err_msg}"
    );
}

// ─── All supported variables ────────────────────────────────────────────────

#[test]
fn render_template_all_supported_variables() {
    let dir = tempfile::tempdir().unwrap();
    let path = dir.path().join("full.txt");
    std::fs::write(
        &path,
        "Type: {{type}}\nScope: {{scope}}\nDiff: {{diff}}\nFiles: {{files}}\nSymbols: {{symbols}}\nEvidence: {{evidence}}\nConstraints: {{constraints}}",
    )
    .unwrap();

    let mut vars = HashMap::new();
    vars.insert("type", "feat");
    vars.insert("scope", "cli");
    vars.insert("diff", "+new code");
    vars.insert("files", "src/cli.rs");
    vars.insert("symbols", "fn main");
    vars.insert("evidence", "has bug evidence: no");
    vars.insert("constraints", "no constraints");

    let result = template::render_template(&path, &vars).unwrap();

    assert!(result.contains("Type: feat"));
    assert!(result.contains("Scope: cli"));
    assert!(result.contains("Diff: +new code"));
    assert!(result.contains("Files: src/cli.rs"));
    assert!(result.contains("Symbols: fn main"));
    assert!(result.contains("Evidence: has bug evidence: no"));
    assert!(result.contains("Constraints: no constraints"));
}