var {runDocTest} = require('./helpers');
var PDFDocument = require('../../lib/document').default;
var CryptoJS = require('crypto-js');

describe('pdfmake', function () {
  let generateRandomWordArray = null;

  beforeAll(function () {
    const doc = new PDFDocument({ userPassword: 'user' });
    generateRandomWordArray = Object.getPrototypeOf(doc._security).constructor.generateRandomWordArray;
    Object.getPrototypeOf(doc._security).constructor.generateRandomWordArray = function (bytes) {
      return CryptoJS.lib.WordArray.create(null, bytes);
    };
  });

  afterAll(function () {
    const doc = new PDFDocument({ userPassword: 'user' });
    Object.getPrototypeOf(doc._security).constructor.generateRandomWordArray = generateRandomWordArray;
  });

  test('encryption with RC-40 (PDF 1.3)', function () {
    return runDocTest({ pdfVersion: '1.3', userPassword: 'user', ownerPassword: 'owner' }, function(doc) {
      doc.text('test');
    });
  });

  test('encryption with RC-128 (PDF 1.4)', function () {
    return runDocTest({ pdfVersion: '1.4', userPassword: 'user', ownerPassword: 'owner' }, function(doc) {
      doc.text('test');
    });
  });

  test('encryption with RC-128 (PDF 1.5)', function () {
    return runDocTest({ pdfVersion: '1.5', userPassword: 'user', ownerPassword: 'owner' }, function(doc) {
      doc.text('test');
    });
  });

  test('encryption with AES-128 (PDF 1.6)', function () {
    return runDocTest({ pdfVersion: '1.6', userPassword: 'user', ownerPassword: 'owner' }, function(doc) {
      doc.text('test');
    });
  });

  test('encryption with AES-128 (PDF 1.7)', function () {
    return runDocTest({ pdfVersion: '1.7', userPassword: 'user', ownerPassword: 'owner' }, function(doc) {
      doc.text('test');
    });
  });

  test('encryption with AES-256 (PDF 1.7 extension 3)', function () {
    return runDocTest({ pdfVersion: '1.7ext3', userPassword: 'user', ownerPassword: 'owner' }, function(doc) {
      doc.text('test');
    });
  });
});