Fix tests

TryQuiet · May 10, 2024 · 6f71eb3 · 6f71eb3
1 parent 6bccb85
commit 6f71eb3
Showing 1 changed file with 9 additions and 8 deletions.
diff --git a/packages/state-manager/src/utils/functions/storybookLog/storybookLog.test.ts b/packages/state-manager/src/utils/functions/storybookLog/storybookLog.test.ts
@@ -1,10 +1,11 @@
+import { QuietLogger } from '@quiet/logger'
 import { storybookLog } from './storybookLog.function'
 
 describe('storybookLog function', () => {
   const consoleLogMessage = 'storybookLog called'
 
   beforeEach(() => {
-    jest.spyOn(console, 'info')
+    jest.spyOn(QuietLogger.prototype, 'info')
   })
 
   afterEach(() => {
@@ -20,17 +21,17 @@ describe('storybookLog function', () => {
   it('should call `logger.info` with passed message', () => {
     storybookLog(consoleLogMessage)()
 
-    expect(console.info).toHaveBeenCalledWith(consoleLogMessage)
-    expect(console.info).toHaveBeenCalledTimes(1)
+    expect(QuietLogger.prototype.info).toHaveBeenCalledWith(consoleLogMessage)
+    expect(QuietLogger.prototype.info).toHaveBeenCalledTimes(1)
   })
 
-  it('should call `console.info` with passed args', () => {
+  it('should call `logger.info` with passed args', () => {
     const args = ['something', 5]
     storybookLog(consoleLogMessage)(...args)
 
-    expect(console.info).toHaveBeenCalledWith(consoleLogMessage)
-    expect(console.info).toHaveBeenCalledWith(args[0])
-    expect(console.info).toHaveBeenCalledWith(args[1])
-    expect(console.info).toHaveBeenCalledTimes(3)
+    expect(QuietLogger.prototype.info).toHaveBeenCalledWith(consoleLogMessage)
+    expect(QuietLogger.prototype.info).toHaveBeenCalledWith(args[0])
+    expect(QuietLogger.prototype.info).toHaveBeenCalledWith(args[1])
+    expect(QuietLogger.prototype.info).toHaveBeenCalledTimes(3)
   })
 })