X-Git-Url: https://git.mdrn.pl/fnpeditor.git/blobdiff_plain/b729636a8bfe248aa9890a6f3bef56cbbab885fd..fd6329db57fc3cc1cf2305da8d0127de4f2198e2:/src/wlxml/wlxml.test.js?ds=sidebyside

diff --git a/src/wlxml/wlxml.test.js b/src/wlxml/wlxml.test.js
index e2c11fa..3d1df03 100644
--- a/src/wlxml/wlxml.test.js
+++ b/src/wlxml/wlxml.test.js
@@ -258,67 +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 = {};
             
-            console.log('A');
-            expect(function() {
-                elementNode.transform('testTransformation');
-            }).to.throw(Error);
-            console.log('B');
-            expect(function() {
-                textNode.transform('testTransformation');
-            }).to.throw(Error);
-            console.log('C');
-            expect(function() {
-                doc.transform('testTransformation');
-            }).to.throw(Error);
-            expect(doc.testMethod).to.be.undefined;
-            expect(elementNode.testMethod).to.be.undefined;
-            expect(textNode.testMethod).to.be.undefined;
-        });
-
-        it('allows adding method to a document', function() {
-            extension = {document: {methods: {
-                testMethod: function() { return this; }
-            }}};
-
-            doc.registerExtension(extension);
-            expect(doc.testMethod()).to.equal(doc, 'context is set to a document instance');
-        });
-
-        it('allows adding transformation to a document', function() {
-            extension = {document: {transformations: {
-                testTransformation: function() { return this; },
-                testTransformation2: {impl: function() { return this;}}
-            }}};
-
-            doc.registerExtension(extension);
-            expect(doc.transform('testTransformation')).to.equal(doc, 'context is set to a document instance');
-            expect(doc.transform('testTransformation2')).to.equal(doc, 'context is set to a document instance');
-        });
-
-        it('allows adding method to a DocumentNode instance', function() {
-            extension = {documentNode: {methods: {
-                testMethod: function() { return this; }    
-            }}};
-
-            doc.registerExtension(extension);
-            expect(elementNode.testMethod().sameNode(elementNode)).to.equal(true, 'context is set to a node instance');
-            expect(textNode.testMethod().sameNode(textNode)).to.equal(true, 'context is set to a node instance');
-        });
-
-        it('allows adding transformation to a DocumentNode', function() {
-            extension = {documentNode: {transformations: {
-                testTransformation: function() { return this; },
-                testTransformation2: {impl: function() { return this;}}
-            }}};
-            
-            doc.registerExtension(extension);
+            expect(testClassNode.object).to.be.undefined;
 
-            expect(elementNode.transform('testTransformation').sameNode(elementNode)).to.equal(true, '1');
-            expect(elementNode.transform('testTransformation2').sameNode(elementNode)).to.equal(true, '2');
-            expect(textNode.transform('testTransformation').sameNode(textNode)).to.equal(true, '3');
-            expect(textNode.transform('testTransformation2').sameNode(textNode)).to.equal(true, '4');
         });
 
         it('allows adding method to an ElementNode of specific class', function() {
@@ -337,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');
         });