X-Git-Url: https://git.mdrn.pl/fnpeditor.git/blobdiff_plain/de48f6605b47f51523959d40cd6ba53e1b404b3f..73e8795a3d19b7221f34353799d39808d57a4d05:/src/wlxml/wlxml.test.js

diff --git a/src/wlxml/wlxml.test.js b/src/wlxml/wlxml.test.js
index bb5101f..3d1df03 100644
--- a/src/wlxml/wlxml.test.js
+++ b/src/wlxml/wlxml.test.js
@@ -258,10 +258,11 @@ describe('WLXMLDocument', function() {
             doc = getDocumentFromXML('<section>Alice<div class="test_class"></div></section>');
             elementNode = doc.root;
             textNode = doc.root.contents()[0];
+            testClassNode = doc.root.contents('.test_class');
             extension = {};
             
+            expect(testClassNode.object).to.be.undefined;
 
-            // spr+ a expect dotyczacy object api?
         });
 
         it('allows adding method to an ElementNode of specific class', function() {
@@ -280,8 +281,8 @@ describe('WLXMLDocument', function() {
             }}}};
             doc.registerExtension(extension);
             testClassNode = doc.root.contents()[1];
-            expect(testClassNode.object.transform('testTransformation').sameNode(testClassNode)).to.equal(true, '1');
-            expect(testClassNode.object.transform('testTransformation2').sameNode(testClassNode)).to.equal(true, '1');
+            expect(testClassNode.object.testTransformation().sameNode(testClassNode)).to.equal(true, '1');
+            expect(testClassNode.object.testTransformation2().sameNode(testClassNode)).to.equal(true, '1');
         });