style: Update mock function signature in handler and remove unnecessa…

…ry comment in util test.
wajeht · Aug 2, 2024 · 442ef66 · 442ef66
1 parent 4b10396
commit 442ef66
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/src/handler.test.ts b/src/handler.test.ts
@@ -216,7 +216,9 @@ describe('getIndexHandler', () => {
 
 describe('postGenerateCommitMessageHandler', () => {
 	const createMockAIService = (mockMessage: string | null) => ({
-		generate: mock.fn<(diff: string) => Promise<string | null>>(() => Promise.resolve(mockMessage)),
+		generate: mock.fn<(diff: string, apiKey?: string) => Promise<string | null>>(() =>
+			Promise.resolve(mockMessage),
+		),
 	});
 
 	const createMockAIFactory = (mockService: AIService) =>

diff --git a/src/util.test.ts b/src/util.test.ts
@@ -223,7 +223,6 @@ describe('validateConfig', () => {
 	let loggerOutput: string;
 
 	beforeEach(() => {
-		// Mock logger.error
 		originalLoggerError = logger.error;
 		loggerOutput = '';