shortcut method to listen on end event

dodo · Dec 19, 2011 · ec7874c · ec7874c
1 parent c8d62bb
commit ec7874c
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/src/template.coffee b/src/template.coffee
@@ -185,6 +185,12 @@ class Template extends EventEmitter
     end: =>
         @xml.end arguments...
 
+    ready: (callback) =>
+        if @xml.closed is yes
+            callback()
+        else
+            @xml.once('end', callback)
+
 
 # exports
 

diff --git a/src/test/template.coffee b/src/test/template.coffee
@@ -25,7 +25,7 @@ module.exports =
                         @end()
 
 
-        xml.on 'end', æ.done
+        xml.ready æ.done
         xml.on 'data', (tag) -> æ.equal results.shift(), tag
         results = [
             '<!DOCTYPE html>'
@@ -56,7 +56,7 @@ module.exports =
                     @$title "holla"
                 @$body ->
                     @$p "hello world"
-        xml.on 'end', æ.done
+        xml.ready æ.done
         xml.on 'data', (tag) -> æ.equal results.shift(), tag
         results = [
             '<!DOCTYPE html>\n'
@@ -95,7 +95,7 @@ module.exports =
             title:'test'
             content:'..'
 
-        xml.on 'end', æ.done
+        xml.ready æ.done
         xml.on 'data', (tag) -> æ.equal results.shift(), tag
         results = [
             '<html>'