blob: 3fbb2dd9ac518686975667b0f8caae5a60c08df6 [file] [log] [blame]
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' ])
})