79 lines
2.3 KiB
JavaScript
Executable File
79 lines
2.3 KiB
JavaScript
Executable File
'use strict'
|
|
|
|
var test = require('tape')
|
|
var redeyed = require('..')
|
|
|
|
test('given i skip 2 more tokens after each semicolon', function(t) {
|
|
var calls = 0
|
|
var opts = {
|
|
Punctuator: {
|
|
';': function identity(s, info) {
|
|
// tell it to skip past second to last token that is 2 ahead of the current one
|
|
calls++
|
|
var skipToken = info.tokens[info.tokenIndex + 2]
|
|
return skipToken ? { replacement: s, skipPastToken: skipToken } : s
|
|
}
|
|
}
|
|
}
|
|
|
|
;[ { code: ';;;', expectedCalls: 1 }
|
|
, { code: ';;;;', expectedCalls: 2 }
|
|
, { code: '; ; ; ;', expectedCalls: 2 }
|
|
, { code: ';;; ;;; ;;; ;', expectedCalls: 4 }
|
|
, { code: ';;; ;;; ;;; ;;; ;', expectedCalls: 5 }
|
|
, { code: ';;; ;;; ;;; ;;; ;;;', expectedCalls: 5 }
|
|
, { code: ';;; ;;; ;;; ;;; ;;; ;', expectedCalls: 6 }
|
|
].forEach(function(x) {
|
|
calls = 0
|
|
redeyed(x.code, opts)
|
|
t.equals(calls, x.expectedCalls, 'calls ' + x.expectedCalls + ' times for ' + x.code)
|
|
})
|
|
t.end()
|
|
})
|
|
|
|
test('replace log', function(t) {
|
|
var opts = {
|
|
Identifier: {
|
|
console: function replaceLog(s, info) {
|
|
var code = info.code
|
|
var idx = info.tokenIndex
|
|
var tokens = info.tokens
|
|
var kind = tokens[idx + 2].value
|
|
var firstArgTkn = tokens[idx + 4]
|
|
var argIdx = idx + 3
|
|
var open
|
|
var tkn
|
|
|
|
open = 1
|
|
while (open) {
|
|
tkn = tokens[++argIdx]
|
|
|
|
if (tkn.value === '(') open++
|
|
if (tkn.value === ')') open--
|
|
}
|
|
|
|
var argsIncludingClosingParen = code.slice(firstArgTkn.range[0], tkn.range[1])
|
|
var result = 'log.' + kind + '("main-logger", ' + argsIncludingClosingParen
|
|
|
|
return { replacement: result, skipPastToken: tkn }
|
|
}
|
|
}
|
|
}
|
|
|
|
var origCode = [
|
|
'console.info("info ", 1);'
|
|
, 'console.warn("warn ", 3);'
|
|
, 'console.error("error ", new Error("oh my!"));'
|
|
].join('\n')
|
|
|
|
var expectedCode = [
|
|
'log.info("main-logger", "info ", 1));'
|
|
, 'log.warn("main-logger", "warn ", 3));'
|
|
, 'log.error("main-logger", "error ", new Error("oh my!")));'
|
|
].join('\n')
|
|
var code = redeyed(origCode, opts).code
|
|
|
|
t.equals(code, expectedCode, 'transforms all log statements')
|
|
t.end()
|
|
})
|