Update: config

increments · Sep 2, 2023 · 79283e6 · 79283e6
1 parent 56451df
commit 79283e6
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/src/lib/config.test.ts b/src/lib/config.test.ts
@@ -55,7 +55,7 @@ jest.mock("node:fs/promises", () => {
     writeFile: jest.fn((filePath: string, text: string) => {
       setFile(filePath, text);
     }),
-    mkdir: jest.fn(() => {}),
+    mkdir: jest.fn(() => { }),
   };
 });
 jest.mock("node:fs", () => {
@@ -224,6 +224,8 @@ describe("config", () => {
       beforeEach(() => {
         const userConfigData = {
           includePrivate: true,
+          address: "localhost",
+          family: "IPv4",
           port: 9999,
         };
         resetFiles();
@@ -234,6 +236,8 @@ describe("config", () => {
         const userConfig = await config.getUserConfig();
         expect(userConfig).toStrictEqual({
           includePrivate: true,
+          address: "localhost",
+          family: "IPv4",
           port: 9999,
         });
       });
@@ -248,6 +252,8 @@ describe("config", () => {
         const userConfig = await config.getUserConfig();
         expect(userConfig).toStrictEqual({
           includePrivate: false,
+          address: "localhost",
+          family: "IPv4",
           port: 8888,
         });
       });

diff --git a/src/lib/config.ts b/src/lib/config.ts
@@ -14,6 +14,8 @@ interface Options {
 
 type UserConfig = {
   includePrivate: boolean;
+  address: string;
+  family: string;
   port: number;
 };
 
@@ -107,6 +109,8 @@ class Config {
   async getUserConfig() {
     const defaultConfig = {
       includePrivate: false,
+      address: "localhost",
+      family: "IPv4",
       port: 8888,
     } as UserConfig;