Added missing unit test, fixed test

ad-si · Mar 14, 2013 · 7582e28 · 7582e28
1 parent 4d4dc25
commit 7582e28
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/test/standard_features.js b/test/standard_features.js
@@ -48,7 +48,8 @@ TestCase('Standard', {
 	'test callback function': function () {
 		/*:DOC gen = <div></div>*/
 
-		var bar, element
+		var bar = false,
+			element = false
 
 		function foo(el) {
 			bar = true
@@ -81,5 +82,14 @@ TestCase('Standard', {
 
 		assertEquals('Should return element', document.getElementById('foo'), fragment.foo)
 		assertEquals('Should return element', document.getElementById('bar'), fragment.bar)
+	},
+
+	'test return of marked elements ': function () {
+		/*:DOC += <div id="gen"></div>*/
+
+		var fragment = DOMinate([document.getElementById('gen'), ['a$foo'], ['p$bar']])
+
+		assertEquals('Should return element', document.getElementsByTagName('a')[0], fragment.foo)
+		assertEquals('Should return element', document.getElementsByTagName('p')[0], fragment.bar)
 	}
 })