Refactoring in WLXMLDocument.registerExtension

[fnpeditor.git] / src / wlxml / extensions / list / list.test.js

diff --git a/src/wlxml/extensions/list/list.test.js b/src/wlxml/extensions/list/list.test.js
index 1efa795..10a6e91 100644 (file)
--- a/src/wlxml/extensions/list/list.test.js
+++ b/src/wlxml/extensions/list/list.test.js
@@ -6,11 +6,13 @@ var chai = require('libs/chai'),
     wlxml = require('wlxml/wlxml'),
     expect = chai.expect,
     $ = require('libs/jquery'),
-    lists = require('wlxml/extensions/list/list');
+    listsExtension = require('wlxml/extensions/list/list');
 
 
 var getDocumentFromXML = function(xml, options) {
-    return wlxml.WLXMLDocumentFromXML(xml, options || {});
+    var doc = wlxml.WLXMLDocumentFromXML(xml, options || {});
+    doc.registerExtension(listsExtension);
+    return doc;
 };
 
 var removeEmptyTextNodes = function(xml) {