Merge pull request #55 from ca-dp/fix-error-handle

Fix error handle
ca-dp · Oct 19, 2023 · 78e0967 · 78e0967
2 parents 676e304 + 1b90ff3
commit 78e0967
Show file tree

Hide file tree

Showing 4 changed files with 17 additions and 7 deletions.
diff --git a/__tests__/main.test.ts b/__tests__/main.test.ts
@@ -90,7 +90,9 @@ describe('run', () => {
     await main.run()
 
     // Ensure the error message was set correctly
-    expect(setFailedMock).toHaveBeenCalledWith('Response content is missing')
+    expect(setFailedMock).toHaveBeenCalledWith(
+      '[review]Response content is missing'
+    )
 
     // Clear environment variables
     delete process.env.OPENAI_API_KEY
@@ -180,7 +182,9 @@ describe('run', () => {
     await main.run()
 
     // Ensure the error message was set correctly
-    expect(setFailedMock).toHaveBeenCalledWith('Response content is missing')
+    expect(setFailedMock).toHaveBeenCalledWith(
+      '[chat]Response content is missing'
+    )
 
     // Clear environment variables
     delete process.env.OPENAI_API_KEY

diff --git a/badges/coverage.svg b/badges/coverage.svg
diff --git a/dist/index.js b/dist/index.js
diff --git a/src/main.ts b/src/main.ts
@@ -13,6 +13,9 @@ export async function run(): Promise<void> {
     switch (cmd) {
       case 'review': {
         const diff = await github.getPullRequestDiff()
+        if (diff === '') {
+          core.setFailed('Pull request diff is missing')
+        }
         const sysPrompt = prompt.getCodeReviewSystemPrompt()
         const messagePromise = ai.completionRequest(
           core.getInput('OPENAI_API_KEY', { required: true }),
@@ -22,7 +25,7 @@ export async function run(): Promise<void> {
         const message = await messagePromise
 
         if (message === '') {
-          core.setFailed('Response content is missing')
+          core.setFailed('[review]Response content is missing')
         }
 
         await github.createGitHubComment(message)
@@ -44,7 +47,7 @@ export async function run(): Promise<void> {
         )
         const response = await responseMessage
         if (response === '') {
-          core.setFailed('Response content is missing')
+          core.setFailed('[chat]Response content is missing')
         }
 
         await github.createGitHubComment(response)