index.js 3.15 KB
var test = require('tape')
var assert = require('assert')
var parse = require('acorn').parse
var getAssignedIdentifiers = require('../')

function getName (node) {
  assert.equal(node.type, 'Identifier', 'Returned node must be an Identifier')
  return node.name
}

test('example', function (t) {
  t.plan(1)

  var ast = parse(`
    var { a, b: [ c,, ...x ], d } = whatever()
  `)
  var node = ast.body[0].declarations[0].id
  t.deepEqual(getAssignedIdentifiers(node).map(getName), [
    'a',
    'c',
    'x',
    'd'
  ])
})

test('simple identifiers', function (t) {
  t.plan(1)
  var ast = parse(`
    var xyz = whatever()
  `)
  var node = ast.body[0].declarations[0].id
  t.deepEqual(getAssignedIdentifiers(node).map(getName), [ 'xyz' ])
})

test('array destructuring', function (t) {
  t.plan(1)
  var ast = parse(`
    var [a, b, c] = whatever()
  `)
  var node = ast.body[0].declarations[0].id
  t.deepEqual(getAssignedIdentifiers(node).map(getName), [ 'a', 'b', 'c' ])
})

test('array destructuring with rest element', function (t) {
  t.plan(1)
  var ast = parse(`
    var [a, b, ...rest] = whatever()
  `)
  var node = ast.body[0].declarations[0].id
  t.deepEqual(getAssignedIdentifiers(node).map(getName), [ 'a', 'b', 'rest' ])
})

test('array destructuring with holes', function (t) {
  t.plan(1)
  var ast = parse(`
    var [a, b,,,,,, boop] = whatever()
  `)
  var node = ast.body[0].declarations[0].id
  t.deepEqual(getAssignedIdentifiers(node).map(getName), [ 'a', 'b', 'boop' ])
})

test('nested array destructuring', function (t) {
  t.plan(1)
  var ast = parse(`
    var [a, [[[b]], ...c], boop] = whatever()
  `)
  var node = ast.body[0].declarations[0].id
  t.deepEqual(getAssignedIdentifiers(node).map(getName), [ 'a', 'b', 'c', 'boop' ])
})

test('object destructuring', function (t) {
  t.plan(1)
  var ast = parse(`
    var {a, b} = whatever()
  `)
  var node = ast.body[0].declarations[0].id
  t.deepEqual(getAssignedIdentifiers(node).map(getName), [ 'a', 'b' ])
})

test('object destructuring with different names', function (t) {
  t.plan(1)
  var ast = parse(`
    var {a: b, b: lol} = whatever()
  `)
  var node = ast.body[0].declarations[0].id
  t.deepEqual(getAssignedIdentifiers(node).map(getName), [ 'b', 'lol' ])
})

test('nested object destructuring', function (t) {
  t.plan(1)
  var ast = parse(`
    var {a: {b}, b: lol, c: {
      d, e: { f: g }
    }} = whatever()
  `)
  var node = ast.body[0].declarations[0].id
  t.deepEqual(getAssignedIdentifiers(node).map(getName), [ 'b', 'lol', 'd', 'g' ])
})

test('object rest destructuring', function (t) {
  t.plan(1)
  var ast = parse(`
    var {a, ...b} = whatever()
  `, { ecmaVersion: 9 })
  var node = ast.body[0].declarations[0].id
  t.deepEqual(getAssignedIdentifiers(node).map(getName), [ 'a', 'b' ])
})

test('import declarations', function (t) {
  t.plan(2)
  var ast = parse(`
    import x, { y, z as a } from 'module'
  `, { sourceType: 'module' })
  var node = ast.body[0]
  t.deepEqual(getAssignedIdentifiers(node).map(getName), [ 'x', 'y', 'a' ])

  ast = parse(`
    import * as ns from 'module'
  `, { sourceType: 'module' })
  node = ast.body[0]
  t.deepEqual(getAssignedIdentifiers(node).map(getName), [ 'ns' ])
})