add new test

src/commands/repl.ts

@@ -34,6 +34,7 @@ type DynamicDiagramClient = {
   onReload: () => void,
   enabled: boolean,
   app?: Express, // only for testing purposes
+  server?: http.Server, // only for testing purposes
 }
 
 export default async function (autoImportPath: string | undefined, options: Options): Promise<void> {
@@ -223,13 +224,13 @@ export async function initializeClient(options: Options, repl: Interface, interp
   const server = http.createServer(app)
 
   server.addListener('error', ({ port, code }: { port: string, code: string }) => {
-    console.info('')
+    logger.info('')
     if (code === 'EADDRINUSE') {
-      console.info(yellow(bold(`⚡ We couldn't start dynamic diagram at port ${port}, because it is already in use. ⚡`)))
+      logger.info(yellow(bold(`⚡ We couldn't start dynamic diagram at port ${port}, because it is already in use. ⚡`)))
       // eslint-disable-next-line @typescript-eslint/quotes
-      console.info(yellow(`Please make sure you don't have another REPL session running in another terminal. \nIf you want to start another instance, you can use "--port xxxx" option, where xxxx should be any available port.`))
+      logger.info(yellow(`Please make sure you don't have another REPL session running in another terminal. \nIf you want to start another instance, you can use "--port xxxx" option, where xxxx should be any available port.`))
     } else {
-      console.info(yellow(bold(`⚡ REPL couldn't be started at port ${port}, error code ["${code}]. ⚡`)))
+      logger.info(yellow(bold(`⚡ REPL couldn't be started at port ${port}, error code ["${code}]. ⚡`)))
     }
     process.exit(1)
   })
@@ -261,6 +262,7 @@ export async function initializeClient(options: Options, repl: Interface, interp
     },
     enabled: true,
     app,
+    server,
   }
 }
 

test/dynamicDiagramClient.test.ts

@@ -34,10 +34,23 @@ describe('dynamic diagram client', () => {
   })
 
   it('should work for root path', async () => {
-    const { enabled, app } = await initializeClient(options, repl, interpreter)
+    const { enabled, app, server } = await initializeClient(options, repl, interpreter)
     expect(enabled).to.be.true
     const result = await chai.request(app).get('/index.html')
     expect(result).to.have.status(200)
+    server!.close()
   })
 
+  it('should return a fake client if noDiagram is set', async () => {
+    const noDiagramOptions = {
+      ...options,
+      noDiagram: true,
+    }
+    const { enabled, app } = await initializeClient(noDiagramOptions, repl, interpreter)
+    expect(enabled).to.be.false
+    expect(app).to.be.undefined
+  })
+
+  // testing failure cases are extremely complicated due to server listeners and async notifications
+
 })