pdfkit/tests/unit/reference.spec.js

const PDFReference = require("../../lib/reference").default;
const PDFDocument = require("../../lib/document").default;
const zlib = require('zlib')

describe('PDFReference', () => {
  let document
  beforeEach(() => {
    document = new PDFDocument()
  })
  
  test('instantiated without data', () => {
    const ref = new PDFReference(document, 1);
    
    expect(ref.id).toBeDefined()          
    expect(ref.data).toBeDefined()
    expect(ref.data).toBeInstanceOf(Object)          
  })

  test('instantiated with data', () => {
    const refData = {Pages: 0}
    const ref = new PDFReference(document, 1, refData);    

    expect(ref.id).toBe(1)
    expect(ref.data).toBe(refData)        
  })

  test('written data of empty reference', (done) => {
    const dataLog = []
    const expected = [
      '1 0 obj', 
      '<<\n>>', 
      'endobj'
    ]
    const ref = new PDFReference(document, 1);
    document._write = function(data) {
      dataLog.push(data)
    }
    ref.finalize()
    setTimeout(() => {
      expect(dataLog).toEqual(expected)
      done()
    }, 1)
  })

  test('written data of reference with uncompressed data', (done) => {
    const dataLog = []
    const chunk = new Buffer('test')
    const expected = [
      '1 0 obj', 
      `<<
/Length ${chunk.length}
>>`,
      'stream',
      chunk,
      '\nendstream',
      'endobj'
    ]
    const ref = new PDFReference(document, 1);
    ref.compress = false
    ref.write(chunk)
    document._write = function(data) {
      dataLog.push(data)
    }
    ref.finalize()
    setTimeout(() => {
      expect(dataLog).toEqual(expected)
      done()
    }, 1)
  })

  test('written data of reference with compressed data', (done) => {
    const dataLog = []
    const chunk = new Buffer('test')
    const compressed = zlib.deflateSync(chunk);
    const expected = [
      '1 0 obj', 
      `<<
/Length ${compressed.length}
/Filter /FlateDecode
>>`,
      'stream',
      compressed,
      '\nendstream',
      'endobj'
    ]
    const ref = new PDFReference(document, 1);  
    ref.write(chunk)
    document._write = function(data) {
      dataLog.push(data)
    }
    ref.finalize()
    setTimeout(() => {
      expect(dataLog).toEqual(expected)
      done()
    }, 1)
  })
})