From 5f1ed83225d81863a20c43ce13e73482c7835953 Mon Sep 17 00:00:00 2001
From: Faisal Salman <f@faisalman.com>
Date: Wed, 8 Oct 2025 13:13:17 +0700
Subject: [PATCH] [chore] Update CLI import & unit test

---
 script/cli.js             |  4 ++--
 test/unit/cli/cli.spec.js | 16 ++++++++--------
 2 files changed, 10 insertions(+), 10 deletions(-)

diff --git a/script/cli.js b/script/cli.js
index 368d6c5..813cfc3 100755
--- a/script/cli.js
+++ b/script/cli.js
@@ -6,8 +6,8 @@ try {
     const { performance } = require('node:perf_hooks');
     const readline = require('node:readline');
     const { parseArgs } = require('node:util');
-    const UAParser = require('ua-parser-js');
-    const { Bots, Emails, ExtraDevices, InApps, Vehicles } = require('ua-parser-js/extensions');
+    const UAParser = require('../src/main/ua-parser');
+    const { Bots, Emails, ExtraDevices, InApps, Vehicles } = require('../src/extensions/ua-parser-extensions');
 
     if (!process.argv[2].startsWith('-')) {
 
diff --git a/test/unit/cli/cli.spec.js b/test/unit/cli/cli.spec.js
index 7365fe4..ab5523e 100644
--- a/test/unit/cli/cli.spec.js
+++ b/test/unit/cli/cli.spec.js
@@ -4,6 +4,12 @@ const fs = require('node:fs');
 const { UAParser } = require('../../../src/main/ua-parser');
 const uap = new UAParser();
 
+const input = [
+    'Opera/9.25 (Windows NT 6.0; U; ru)',
+    'Mozilla/4.0 (compatible; MSIE 5.5; Windows NT)'
+];
+const output = input.map(x => uap.setUA(x).getResult());
+
 describe('npx ua-parser-js <string>', () => {
     it ('print result to stdout', () => {
         exec('npx ua-parser-js "TEST"', (err, stdout, stderr) => {
@@ -15,10 +21,7 @@ describe('npx ua-parser-js <string>', () => {
 describe('npx ua-parser-js --input-file=<filepath>', () => {
     it ('load file and print result to stdout', () => {
         exec('npx ua-parser-js --input-file="../test/unit/cli/input.txt"', (err, stdout, stderr) => {
-            assert.deepEqual(JSON.parse(stdout), JSON.parse(JSON.stringify([
-                uap.setUA('Opera/9.25 (Windows NT 6.0; U; ru)').getResult(),
-                uap.setUA('Mozilla/4.0 (compatible; MSIE 5.5; Windows NT)').getResult()
-            ])));
+            assert.deepEqual(JSON.parse(stdout), JSON.parse(JSON.stringify(output)));
         });
     });
 });
@@ -27,10 +30,7 @@ describe('npx ua-parser-js --input-file=<filepath> --output-file=<filepath>', ()
     it ('load file and save result to file', () => {
         exec('npx ua-parser-js --input-file="../test/unit/cli/input.txt" --output-file="../test/unit/cli/output.json"', (err, stdout, stderr) => {
             fs.readFile('test/unit/cli/output.json', (err, data) => {
-                assert.deepEqual(JSON.parse(data), JSON.parse(JSON.stringify([
-                    uap.setUA('Opera/9.25 (Windows NT 6.0; U; ru)').getResult(),
-                    uap.setUA('Mozilla/4.0 (compatible; MSIE 5.5; Windows NT)').getResult()
-                ])));
+                assert.deepEqual(JSON.parse(data), JSON.parse(JSON.stringify(output)));
             });
         });
     });