added test for main function

asyncapi · Nov 28, 2024 · 02e221b · 02e221b
1 parent 0fbdc06
commit 02e221b
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 1 deletion.
diff --git a/scripts/markdown/check-markdown.js b/scripts/markdown/check-markdown.js
@@ -157,4 +157,4 @@ async function main() {
 
 main();
 
-module.exports = { validateBlogs, validateDocs, checkMarkdownFiles }
+module.exports = { validateBlogs, validateDocs, checkMarkdownFiles, main }
diff --git a/tests/markdown/check-markdown.test.js b/tests/markdown/check-markdown.test.js
@@ -2,6 +2,7 @@ const fs = require('fs').promises;
 const path = require('path');
 const os = require('os');
 const {
+    main,
     validateBlogs,
     validateDocs,
     checkMarkdownFiles
@@ -116,4 +117,19 @@ describe('Frontmatter Validator', () => {
         mockReadFile.mockRestore();
     });
 
+    it('should handle main function errors', async () => {
+        const mockExit = jest.spyOn(process, 'exit').mockImplementation(() => {});
+        jest.spyOn(fs, 'readdir').mockRejectedValue(new Error('Test error'));
+
+        await main();
+
+        expect(mockConsoleError).toHaveBeenCalledWith(
+            'Failed to validate markdown files:', 
+            expect.any(Error)
+        );
+        expect(mockExit).toHaveBeenCalledWith(1);
+
+        mockExit.mockRestore();
+    });
+
 });