test.js 2.95 KB
Newer Older
liang ce committed
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111
var Buffer = require('safe-buffer').Buffer
var CipherBase = require('./')

var test = require('tape')
var inherits = require('inherits')

test('basic version', function (t) {
  function Cipher () {
    CipherBase.call(this)
  }
  inherits(Cipher, CipherBase)
  Cipher.prototype._update = function (input) {
    t.ok(Buffer.isBuffer(input))
    return input
  }
  Cipher.prototype._final = function () {
    // noop
  }
  var cipher = new Cipher()
  var utf8 = 'abc123abcd'
  var update = cipher.update(utf8, 'utf8', 'base64') + cipher.final('base64')
  var string = (Buffer.from(update, 'base64')).toString()
  t.equals(utf8, string)
  t.end()
})
test('hash mode', function (t) {
  function Cipher () {
    CipherBase.call(this, 'finalName')
    this._cache = []
  }
  inherits(Cipher, CipherBase)
  Cipher.prototype._update = function (input) {
    t.ok(Buffer.isBuffer(input))
    this._cache.push(input)
  }
  Cipher.prototype._final = function () {
    return Buffer.concat(this._cache)
  }
  var cipher = new Cipher()
  var utf8 = 'abc123abcd'
  var update = cipher.update(utf8, 'utf8').finalName('base64')
  var string = (Buffer.from(update, 'base64')).toString()

  t.equals(utf8, string)
  t.end()
})
test('hash mode as stream', function (t) {
  function Cipher () {
    CipherBase.call(this, 'finalName')
    this._cache = []
  }
  inherits(Cipher, CipherBase)
  Cipher.prototype._update = function (input) {
    t.ok(Buffer.isBuffer(input))
    this._cache.push(input)
  }
  Cipher.prototype._final = function () {
    return Buffer.concat(this._cache)
  }
  var cipher = new Cipher()
  cipher.on('error', function (e) {
    t.notOk(e)
  })
  var utf8 = 'abc123abcd'
  cipher.end(utf8, 'utf8')
  var update = cipher.read().toString('base64')
  var string = (Buffer.from(update, 'base64')).toString()

  t.equals(utf8, string)
  t.end()
})

test('encodings', function (t) {
  inherits(Cipher, CipherBase)
  function Cipher () {
    CipherBase.call(this)
  }
  Cipher.prototype._update = function (input) {
    return input
  }
  Cipher.prototype._final = function () {
    // noop
  }
  t.test('mix and match encoding', function (t) {
    t.plan(2)

    var cipher = new Cipher()
    cipher.update('foo', 'utf8', 'utf8')
    t.throws(function () {
      cipher.update('foo', 'utf8', 'base64')
    })
    cipher = new Cipher()
    cipher.update('foo', 'utf8', 'base64')
    t.doesNotThrow(function () {
      cipher.update('foo', 'utf8')
      cipher.final('base64')
    })
  })
  t.test('handle long uft8 plaintexts', function (t) {
    t.plan(1)
    var txt = 'ふっかつ あきる すぶり はやい つける まゆげ たんさん みんぞく ねほりはほり せまい たいまつばな ひはん'

    var cipher = new Cipher()
    var decipher = new Cipher()
    var enc = decipher.update(cipher.update(txt, 'utf8', 'base64'), 'base64', 'utf8')
    enc += decipher.update(cipher.final('base64'), 'base64', 'utf8')
    enc += decipher.final('utf8')

    t.equals(txt, enc)
  })
})