Merge pull request #409 from PR0M3TH3AN/testing-improvement-cmd-rollback-5981959177572314725

Add tests for cmdRollback

src/cmd-rollback.mjs

@@ -1,18 +1,31 @@
-import { rollbackPrompt, listPromptVersions } from './services/governance/index.js';
+import {
+  rollbackPrompt as _rollbackPrompt,
+  listPromptVersions as _listPromptVersions
+} from './services/governance/index.js';
 import { ExitError } from './errors.mjs';
 
-export async function cmdRollback(target, strategy, { list = false } = {}) {
+export async function cmdRollback(target, strategy, options = {}, deps = {}) {
+  // Backwards compatibility: if options is destructured directly
+  const { list = false } = options;
+
+  const {
+    rollbackPrompt = _rollbackPrompt,
+    listPromptVersions = _listPromptVersions,
+    log = console.log,
+    error = console.error
+  } = deps;
+
   if (!target) {
-    console.error('Usage: torch-lock rollback --target <path> [--strategy <hash|latest>] [--list]');
+    error('Usage: torch-lock rollback --target <path> [--strategy <hash|latest>] [--list]');
     throw new ExitError(1, 'Missing target');
   }
 
   if (list) {
     try {
       const versions = await listPromptVersions(target);
-      console.log(JSON.stringify(versions, null, 2));
+      log(JSON.stringify(versions, null, 2));
     } catch (e) {
-      console.error(`Failed to list versions: ${e.message}`);
+      error(`Failed to list versions: ${e.message}`);
       throw new ExitError(1, 'List versions failed');
     }
     return;
@@ -22,9 +35,9 @@ export async function cmdRollback(target, strategy, { list = false } = {}) {
 
   try {
     const result = await rollbackPrompt(target, effectiveStrategy);
-    console.log(JSON.stringify(result, null, 2));
+    log(JSON.stringify(result, null, 2));
   } catch (e) {
-    console.error(`Rollback failed: ${e.message}`);
+    error(`Rollback failed: ${e.message}`);
     throw new ExitError(1, 'Rollback failed');
   }
 }