添加单元测试

Author: yinxulai

src/logger/index.test.ts

@@ -0,0 +1,87 @@
+import { Logger } from './index'
+
+let isCallReport = false
+
+jest.mock('./report-v3', () => ({
+  reportV3: () => {
+    isCallReport = true
+  }
+}))
+
+describe('test logger', () => {
+  const logMessage: unknown[] = []
+  const warnMessage: unknown[] = []
+  const errorMessage: unknown[] = []
+  console.log = jest.fn((...args: unknown[]) => logMessage.push(...args))
+  console.warn = jest.fn((...args: unknown[]) => warnMessage.push(...args))
+  console.error = jest.fn((...args: unknown[]) => errorMessage.push(...args))
+
+  test('test level', () => {
+    const infoLogger = new Logger('', true, 'INFO')
+    infoLogger.info('test1')
+    expect(logMessage).toStrictEqual([`Qiniu-JS-SDK [INFO][1]: `, 'test1'])
+    infoLogger.warn('test2')
+    expect(warnMessage).toStrictEqual(['Qiniu-JS-SDK [WARN][1]: ', 'test2'])
+    infoLogger.error('test3')
+    expect(errorMessage).toStrictEqual(['Qiniu-JS-SDK [ERROR][1]: ', 'test3'])
+
+    // 清空消息
+    logMessage.splice(0, logMessage.length)
+    warnMessage.splice(0, warnMessage.length)
+    errorMessage.splice(0, errorMessage.length)
+
+    const warnLogger = new Logger('', true, 'WARN')
+    warnLogger.info('test1')
+    expect(logMessage).toStrictEqual([])
+    warnLogger.warn('test2')
+    expect(warnMessage).toStrictEqual(['Qiniu-JS-SDK [WARN][2]: ', 'test2'])
+    warnLogger.error('test3')
+    expect(errorMessage).toStrictEqual(['Qiniu-JS-SDK [ERROR][2]: ', 'test3'])
+
+    // 清空消息
+    logMessage.splice(0, logMessage.length)
+    warnMessage.splice(0, warnMessage.length)
+    errorMessage.splice(0, errorMessage.length)
+
+    const errorLogger = new Logger('', true, 'ERROR')
+    errorLogger.info('test1')
+    expect(logMessage).toStrictEqual([])
+    errorLogger.warn('test2')
+    expect(warnMessage).toStrictEqual([])
+    errorLogger.error('test3')
+    expect(errorMessage).toStrictEqual(['Qiniu-JS-SDK [ERROR][3]: ', 'test3'])
+
+    // 清空消息
+    logMessage.splice(0, logMessage.length)
+    warnMessage.splice(0, warnMessage.length)
+    errorMessage.splice(0, errorMessage.length)
+
+    const offLogger = new Logger('', true, 'OFF')
+    offLogger.info('test1')
+    expect(logMessage).toStrictEqual([])
+    offLogger.warn('test2')
+    expect(warnMessage).toStrictEqual([])
+    offLogger.error('test3')
+    expect(errorMessage).toStrictEqual([])
+  })
+
+  test('test unique id', () => {
+    // @ts-ignore
+    const startId = Logger.id
+    new Logger('', true, 'OFF')
+    new Logger('', true, 'OFF')
+    const last = new Logger('', true, 'OFF')
+    // @ts-ignore
+    expect(last.id).toStrictEqual(startId + 3)
+  })
+
+  test('test report', () => {
+    const logger1 = new Logger('', false, 'OFF')
+    logger1.report(null as any)
+    expect(isCallReport).toBeTruthy()
+    isCallReport = false
+    const logger2 = new Logger('', true, 'OFF')
+    logger2.report(null as any)
+    expect(isCallReport).toBeFalsy()
+  })
+})

src/logger/report-v3.test.ts

@@ -0,0 +1,103 @@
+import { reportV3, V3LogInfo } from './report-v3'
+
+class MockXHR {
+  sendData: string
+  openData: string[]
+  openCount: number
+  headerData: string[]
+
+  status: number
+  readyState: number
+  onreadystatechange() { }
+
+  clear() {
+    this.sendData = ''
+    this.openData = []
+    this.headerData = []
+
+    this.status = 0
+    this.readyState = 0
+  }
+
+  open(...args: string[]) {
+    this.clear()
+    this.openCount += 1
+    this.openData = args
+  }
+
+  send(args: string) {
+    this.sendData = args
+  }
+
+  setRequestHeader(...args: string[]) {
+    this.headerData.push(...args)
+  }
+
+  changeStatusAndState(readyState: number, status: number) {
+    this.status = status
+    this.readyState = readyState
+    this.onreadystatechange()
+  }
+}
+
+const mockXHR = new MockXHR()
+
+jest.mock('../utils', () => ({
+  createXHR: () => (mockXHR),
+  getAuthHeaders: (t: string) => t
+}))
+
+describe('test report-v3', () => {
+  const testData: V3LogInfo = {
+    code: 200,
+    reqId: 'reqId',
+    host: 'host',
+    remoteIp: 'remoteIp',
+    port: 'port',
+    duration: 1,
+    time: 1,
+    bytesSent: 1,
+    upType: 'jssdk-h5',
+    size: 1
+  }
+
+  test('test stringify send Data', () => {
+    reportV3('token', testData, 3)
+    mockXHR.changeStatusAndState(0, 0)
+    expect(mockXHR.sendData).toBe([
+      testData.code || '',
+      testData.reqId || '',
+      testData.host || '',
+      testData.remoteIp || '',
+      testData.port || '',
+      testData.duration || '',
+      testData.time || '',
+      testData.bytesSent || '',
+      testData.upType || '',
+      testData.size || ''
+    ].join(','))
+  })
+
+  test('test retry', () => {
+    mockXHR.openCount = 0
+    reportV3('token', testData)
+    for (let index = 1; index <= 10; index++) {
+      mockXHR.changeStatusAndState(4, 0)
+    }
+    expect(mockXHR.openCount).toBe(4)
+
+    mockXHR.openCount = 0
+    reportV3('token', testData, 4)
+    for (let index = 1; index < 10; index++) {
+      mockXHR.changeStatusAndState(4, 0)
+    }
+    expect(mockXHR.openCount).toBe(5)
+
+    mockXHR.openCount = 0
+    reportV3('token', testData, 0)
+    for (let index = 1; index < 10; index++) {
+      mockXHR.changeStatusAndState(4, 0)
+    }
+    expect(mockXHR.openCount).toBe(1)
+  })
+})