Chore: improve test coverage

Author: mysticatea

lib/parse-component.js

@@ -82,19 +82,21 @@ function parseScriptNode(scriptNode, scriptParser, tokenGenerator) {
 
     // Needs the tokens of start/end tags for `lines-around-*` rules to work
     // correctly.
-    const startTag = tokenGenerator.createToken(
-        "Punctuator",
-        startLoc.startOffset,
-        startLoc.endOffset
-    )
-    const endTag = tokenGenerator.createToken(
-        "Punctuator",
-        endLoc.startOffset,
-        endLoc.endOffset
-    )
-    ast.start = startTag.end
-    ast.tokens.unshift(startTag)
-    ast.tokens.push(endTag)
+    if (ast.tokens) {
+        const startTag = tokenGenerator.createToken(
+            "Punctuator",
+            startLoc.startOffset,
+            startLoc.endOffset
+        )
+        const endTag = tokenGenerator.createToken(
+            "Punctuator",
+            endLoc.startOffset,
+            endLoc.endOffset
+        )
+        ast.start = startTag.end
+        ast.tokens.unshift(startTag)
+        ast.tokens.push(endTag)
+    }
 
     return ast
 }
@@ -137,33 +139,37 @@ function parse(code, options) {
     const templateOptions = options.template || {}
     const isVue = path.extname(filePath || "unknown.js") === ".vue"
     const scriptParser = new ScriptParser(code, scriptOptions)
-
-    if (!isVue) {
-        return {
-            ast: scriptParser._parseScript(code),
-            services: {registerTemplateBodyVisitor},
-        }
+    let ast = null
+
+    if (isVue) {
+        const tokenGenerator = new TokenGenerator(code)
+        const info = parseComponent(code)
+
+        // <script>
+        ast = (info.script != null)
+            ? parseScriptNode(info.script, scriptParser, tokenGenerator)
+            : scriptParser._parseScript("")
+
+        // <template>
+        ast.templateBody = info.template && parseTemplateNode(
+            info.template,
+            scriptParser,
+            tokenGenerator,
+            templateOptions
+        )
     }
-    const info = parseComponent(code)
-    if (!info.script) {
-        return {
-            ast: scriptParser._parseScript(""),
-            services: {registerTemplateBodyVisitor},
-        }
+    else {
+        ast = scriptParser._parseScript(code)
+        ast.templateBody = null
     }
 
-    const tokenGenerator = new TokenGenerator(code)
-    const ast = parseScriptNode(
-        info.script,
-        scriptParser,
-        tokenGenerator
-    )
-    ast.templateBody = info.template && parseTemplateNode(
-        info.template,
-        scriptParser,
-        tokenGenerator,
-        templateOptions
-    )
+    // Ensure ast.comments and ast.tokens.
+    if (ast.comments == null) {
+        ast.comments = []
+    }
+    if (ast.tokens == null) {
+        ast.tokens = []
+    }
 
     return {
         ast,

lib/register-template-body-visitor.js

@@ -180,3 +180,4 @@ module.exports = (context, visitor) => {
         emitter.on(selector, visitor[selector])
     }
 }
+module.exports.traverse = traverse

lib/transform-html.js

@@ -292,7 +292,7 @@ class HTMLTransformer {
      */
     createVAttributeValue(start, end, value) {
         const literal = this.appendToken("VAttributeValue", start, end)
-        return {
+        return literal && {
             parent: null,
             type: literal.type,
             range: literal.range,
@@ -349,7 +349,10 @@ class HTMLTransformer {
             result.value = directive
                 ? this.createVExpressionContainer(i, end, quoteSize)
                 : this.createVAttributeValue(i, end, value)
-            result.value.parent = result
+
+            if (result.value != null) {
+                result.value.parent = result
+            }
         }
 
         return result