feat: update

Author: johackim

__tests__/index.spec.js

@@ -1,16 +1,37 @@
-import { execSync } from 'child_process';
+import { program } from '../src/index';
+import * as utilsModule from '../src/utils';
 import { description, version } from '../package.json';
 
-const ulysse = 'node -r @babel/register src/index.js';
+const runCommand = (command) => {
+    const args = command.split(' ');
+    return program.parseAsync(['node', 'src/index.js', ...args]);
+};
+
+beforeEach(async () => {
+    jest.spyOn(console, 'log').mockImplementation(() => {});
+    jest.spyOn(utilsModule, 'isDaemonRunning').mockReturnValue(true);
+});
 
 test('Should display the version', async () => {
-    const output = execSync(`${ulysse} -v`);
+    const mockWrite = jest.fn();
+    program.exitOverride().configureOutput({ writeOut: mockWrite });
+
+    await expect(runCommand('--version')).rejects.toThrow();
+
+    expect(mockWrite).toHaveBeenCalledWith(expect.stringContaining(version));
+});
+
+test('Should display the help message', async () => {
+    const mockWrite = jest.fn();
+    program.exitOverride().configureOutput({ writeOut: mockWrite });
+
+    await expect(runCommand('--help')).rejects.toThrow();
 
-    expect(output.toString().trim()).toBe(version);
+    expect(mockWrite).toHaveBeenCalledWith(expect.stringContaining(description));
 });
 
-test('Should display the help', async () => {
-    const output = execSync(`${ulysse} -h`);
+test('Should block a domain', async () => {
+    await runCommand('blocklist add --website example.com');
 
-    expect(output.toString().trim()).toContain(description);
+    expect(console.log).toHaveBeenCalledWith('Blocking website example.com');
 });

src/index.js

@@ -310,8 +310,14 @@ program.commands.forEach((cmd) => {
     }
 });
 
-program.parse(process.argv);
+const main = () => program.parse(process.argv);
+
+if (process.env.NODE_ENV !== 'test') {
+    main();
+}
 
 if (!process.argv.slice(2).length) {
     program.outputHelp();
 }
+
+export { program };