s3-files/test/test_index.js

Stream selected files from an Amazon S3 bucket/folder

const sinon = require('sinon')
const t = require('tap')
const PassThrough = require('stream').PassThrough

const proxyquire = require('proxyquire')
const s3Stub = {}
const s3Files = proxyquire('../s3-files.js', {
  '@aws-sdk/client-s3': { S3Client: sinon.stub().returns(s3Stub) }
})

// Connect
t.type(s3Files.s3, undefined)
s3Files.connect({})
t.type(s3Files.s3, 'object')

// Keystream
const keyStream = s3Files.createKeyStream('folder', undefined)
t.same(keyStream, null)

t.test('keyStream', function (child) {
  const keyStream = s3Files.createKeyStream('folder/', ['a', 'b'])
  let cnt = 0
  keyStream.on('data', function (chunk) {
    if (cnt === 0) child.equal(chunk.toString(), 'folder/a')
    if (cnt === 1) child.equal(chunk.toString(), 'folder/b')
    cnt++
  })
  keyStream.on('end', function () {
    child.end()
  })
})

t.test('keyStream without folder having trailing slash', function (child) {
  const keyStream = s3Files.createKeyStream('folder', ['a', 'b'])
  let cnt = 0
  keyStream.on('data', function (chunk) {
    if (cnt === 0) child.equal(chunk.toString(), 'folder/a')
    if (cnt === 1) child.equal(chunk.toString(), 'folder/b')
    cnt++
  })
  keyStream.on('end', function () {
    child.end()
  })
})

t.test('keyStream without folder', function (child) {
  const keyStream = s3Files.createKeyStream('', ['a', 'b'])
  let cnt = 0
  keyStream.on('data', function (chunk) {
    if (cnt === 0) child.equal(chunk.toString(), 'a')
    if (cnt === 1) child.equal(chunk.toString(), 'b')
    cnt++
  })
  keyStream.on('end', function () {
    child.end()
  })
})

// Filestream
t.test('Filestream needs a bucket', function (child) {
  let fileStream = s3Files.createFileStream()
  child.same(fileStream, null)

  const keyStream = s3Files
    .connect({ bucket: 'bucket' })
    .createKeyStream('folder/', ['a', 'b', 'c'])

  const s = new PassThrough()
  s.end('hi')
  s3Stub.send = function () {
    return new Promise(function (resolve) {
      resolve({
        Body: s
      })
    })
  }
  let cnt = 0
  fileStream = s3Files.createFileStream(keyStream)
  fileStream.on('data', function (chunk) {
    child.equal(chunk.data.toString(), 'hi')
    if (cnt === 0) child.equal(chunk.path, 'a')
    if (cnt === 1) {
      child.equal(chunk.path, 'b')
    }
    if (cnt === 2) {
      s.emit('error', new Error('fail'))
    }
    cnt++
  })
  fileStream.on('error', function (chunk) {
    child.ok(chunk)
  })
  fileStream.on('end', function (chunk) {
    setTimeout(function () {
      child.end()
    })
  })
})

t.end()