85 lines
2.2 KiB
JavaScript
85 lines
2.2 KiB
JavaScript
// Unless explicitly stated otherwise all files in this repository are licensed under the Apache 2.0 License.
|
|
//
|
|
// This product includes software developed at Datadog (https://www.datadoghq.com/). Copyright 2021 Datadog, Inc.
|
|
|
|
import Hook from '../../index.js'
|
|
import {
|
|
name1 as n1,
|
|
name2 as n2,
|
|
name3 as n3,
|
|
name4 as n4,
|
|
ClassName as cn,
|
|
generatorFunctionName as gfn,
|
|
name5 as n5,
|
|
bar as n6,
|
|
name7 as n7,
|
|
name8 as n8,
|
|
asyncFunctionName as afn,
|
|
asyncGeneratorFunctionName as agfn,
|
|
arrowFunction as arfn,
|
|
asyncArrowFunction as aarfn
|
|
, functionName
|
|
} from '../fixtures/export-types/declarations.mjs'
|
|
import { strictEqual } from 'assert'
|
|
|
|
Hook((exports, name) => {
|
|
if (name.match(/declarations\.m?js/)) {
|
|
exports.name1 += 1
|
|
exports.name2 += 1
|
|
exports.name3 += 1
|
|
exports.name4 += 1
|
|
const orig = exports.functionName
|
|
exports.functionName = function () {
|
|
return orig() + 1
|
|
}
|
|
exports.ClassName.prototype.getFoo = function () {
|
|
return 2
|
|
}
|
|
const orig2 = exports.generatorFunctionName
|
|
exports.generatorFunctionName = function * () {
|
|
return orig2().next().value + 1
|
|
}
|
|
exports.name5 += 1
|
|
exports.bar += 1
|
|
exports.name7 += 1
|
|
exports.name8 += 1
|
|
const asyncOrig = exports.asyncFunctionName
|
|
exports.asyncFunctionName = async function () {
|
|
return await asyncOrig() + 1
|
|
}
|
|
const asyncOrig2 = exports.asyncGeneratorFunctionName
|
|
exports.asyncGeneratorFunctionName = async function * () {
|
|
for await (const value of asyncOrig2()) {
|
|
yield value + 1
|
|
}
|
|
}
|
|
const arrowOrig = exports.arrowFunction
|
|
exports.arrowFunction = () => {
|
|
return arrowOrig() + 1
|
|
}
|
|
const asyncArrowOrig = exports.asyncArrowFunction
|
|
exports.asyncArrowFunction = async () => {
|
|
return await asyncArrowOrig() + 1
|
|
}
|
|
}
|
|
})
|
|
|
|
strictEqual(n1, 2)
|
|
strictEqual(n2, 2)
|
|
strictEqual(n3, 2)
|
|
strictEqual(n4, 2)
|
|
strictEqual(functionName(), 2)
|
|
/* eslint-disable new-cap */
|
|
strictEqual(new cn().getFoo(), 2)
|
|
strictEqual(gfn().next().value, 2)
|
|
strictEqual(n5, 2)
|
|
strictEqual(n6, 2)
|
|
strictEqual(n7, 2)
|
|
strictEqual(n8, 2)
|
|
strictEqual(await afn(), 2)
|
|
for await (const value of agfn()) {
|
|
strictEqual(value, 2)
|
|
}
|
|
strictEqual(arfn(), 2)
|
|
strictEqual(await aarfn(), 2)
|