update

tests/test_finish_ollama.sh

@@ -0,0 +1,112 @@
+#!/usr/bin/env bash
+set -euo pipefail
+
+# ----- tiny test framework -----
+pass() { printf "✔ %s\n" "$1"; }
+fail() { printf "✘ %s\n" "$1"; exit 1; }
+
+assert_ok() {
+    if [[ $1 -ne 0 ]]; then fail "$2"; fi
+}
+
+assert_nonempty() {
+    if [[ -z "$1" ]]; then fail "$2"; fi
+}
+
+assert_contains() {
+    if [[ ! "$1" =~ $2 ]]; then fail "$3"; fi
+}
+
+# Ensure output does not contain any of the common error phrases.
+assert_no_common_errors() {
+    local output="$1"
+    local -a patterns=(
+        "ACSH_ACTIVE_API_KEY not set"
+        "Bad Request"
+        "Unauthorized"
+        "Too Many Requests"
+        "Internal Server Error"
+        "Unknown Error"
+        "Failed to parse completions"
+        "SyntaxError"
+        "ERROR:"
+    )
+    for pat in "${patterns[@]}"; do
+        if grep -qE "$pat" <<<"$output"; then
+            printf "Test failed: output contains error pattern '%s'\nFull output follows:\n%s\n" "$pat" "$output"
+            exit 1
+        fi
+    done
+}
+# --------------------------------
+
+echo "=== OLLAMA TEST: SETUP ==="
+
+# Always run from the repo root for predictable paths
+SCRIPT_DIR="$(cd -- "$(dirname -- "${BASH_SOURCE[0]}")" && pwd)"
+cd "$SCRIPT_DIR/.."
+
+# Install finish (adds to PATH and shell rc files)
+bash ./docs/install.sh main
+
+# Load the updated shell rc so `finish` is available on PATH in this session
+if [[ -f ~/.bashrc ]]; then
+  # shellcheck disable=SC1090
+  source ~/.bashrc || true
+fi
+
+# ---- Check prerequisites for local Ollama ----
+OLLAMA_URL=${OLLAMA_URL:-"http://localhost:11434"}
+
+echo "Checking Ollama at $OLLAMA_URL ..."
+if ! curl -fsS --max-time 2 "$OLLAMA_URL/api/version" >/dev/null 2>&1; then
+  echo "SKIP: Ollama is not reachable at $OLLAMA_URL. Start it with: 'ollama serve' and ensure a model is pulled (e.g., 'ollama pull llama3')."
+  exit 0
+fi
+
+# Try to discover an installed model; prefer the first listed
+MODEL=$(curl -fsS "$OLLAMA_URL/api/tags" | jq -r '.models[0].name // empty' || echo "")
+if [[ -z "$MODEL" ]]; then
+  echo "SKIP: No local Ollama models found. Pull one first, e.g.: 'ollama pull llama3:latest'"
+  exit 0
+fi
+
+echo "Using Ollama model: $MODEL"
+
+# -------------------------------- TESTS --------------------------------
+
+# 1) configure finish to use local Ollama
+finish config set provider ollama
+finish config set endpoint "$OLLAMA_URL/api/chat"
+finish config set model "$MODEL"
+
+# 2) which finish should return something
+out=$(which finish 2>&1) ; code=$?
+assert_ok $code     "which finish should exit 0"
+assert_nonempty "$out" "which finish returned empty output"
+pass "which finish returns path"
+
+# 3) finish output should contain 'finish.sh'
+out=$(finish 2>&1) ; code=$?
+assert_ok $code     "finish should exit 0"
+assert_contains "$out" "finish\.sh" "finish output does not contain finish.sh"
+pass "finish outputs reference to finish.sh"
+
+# 4) config should mention ollama
+out=$(finish config 2>&1) ; code=$?
+assert_ok $code     "finish config should exit 0"
+assert_contains "$out" "ollama" "finish config missing ollama provider"
+pass "finish config contains ollama"
+
+# 5) finish command should run and return output
+out=$(finish command "ls # show largest files" 2>&1) ; code=$?
+assert_ok $code     "finish command did not exit 0"
+assert_nonempty "$out" "finish command returned empty output"
+assert_no_common_errors "$out"
+pass "finish command executed and returned output (ollama)"
+
+# ------------------------------- CLEANUP --------------------------------
+echo "=== OLLAMA TEST: CLEANUP ==="
+finish remove -y || true
+
+echo "Ollama test passed."