fix(ast): trim whitespace in tag end

Closes #3
edge-js · Oct 16, 2017 · 2ff8b36 · 2ff8b36
1 parent 5bc9889
commit 2ff8b36
Show file tree

Hide file tree

Showing 2 changed files with 25 additions and 1 deletion.
diff --git a/src/Ast/index.js b/src/Ast/index.js
@@ -149,7 +149,7 @@ class Ast {
   _tokenForTag (line, tag, args, index, selfClosing) {
     return {
       tag,
-      args: args ? args.replace(/\)$/, '') : undefined,
+      args: args ? args.trim().replace(/\)$/, '') : undefined,
       selfClosing,
       childs: [],
       body: line,

diff --git a/test/unit/template.spec.js b/test/unit/template.spec.js
@@ -253,6 +253,30 @@ test.group('Template Compiler', (group) => {
     const inlineOutput = template.compileString(inlineStatement)
     assert.equal(output, inlineOutput)
   })
+
+  test('trim component name white space', (assert) => {
+    const statement = `
+    <html>
+      <body>
+        @include('content') 
+      </body>
+    </html>
+    `
+    const template = new Template(this.tags, {})
+    const output = template.compileString(statement)
+
+    assert.equal(output, dedent`
+    return (function templateFn () {
+      let out = new String()
+      out += \`<html>\\n\`
+      out += \`      <body>\\n\`
+      out += \`\${this.runTimeRender('content')}\`
+      out += \`      </body>\\n\`
+      out += \`    </html>\\n\`
+      return out
+    }).bind(this)()
+    `)
+  })
 })
 
 test.group('Template Runner', () => {