X-Git-Url: https://git.mdrn.pl/fnpeditor.git/blobdiff_plain/0a765eadd374d46fc8f4ebe88f9f97a4d8fae622..058d6f3448085c6f1d513691f8f2d5a12ae1d10a:/src/smartxml/smartxml.test.js

diff --git a/src/smartxml/smartxml.test.js b/src/smartxml/smartxml.test.js
index bfe06ad..3175ae9 100644
--- a/src/smartxml/smartxml.test.js
+++ b/src/smartxml/smartxml.test.js
@@ -836,15 +836,9 @@ describe('smartxml', function() {
             textNode = doc.root.contents()[0];
             extension = {};
             
-            expect(function() {
-                elementNode.testTransformation();
-            }).to.throw(Error);
-            expect(function() {
-                textNode.testTransformation();
-            }).to.throw(Error);
-            expect(function() {
-                doc.testTransformation();
-            }).to.throw(Error);
+            expect(elementNode.testTransformation).to.be.undefined;
+            expect(textNode.testTransformation).to.be.undefined;
+            expect(doc.testTransformation).to.be.undefined;
             
             expect(doc.testMethod).to.be.undefined;
             expect(elementNode.testMethod).to.be.undefined;
@@ -909,10 +903,24 @@ describe('smartxml', function() {
         });
 
         it('allows adding transformation to a DocumentNode', function() {
-            extension = {documentNode: {transformations: {
-                testTransformation: function() { return this; },
-                testTransformation2: {impl: function() { return this;}}
-            }}};
+            extension = {
+                documentNode: {
+                    transformations: {
+                        testTransformation: function() { return this; },
+                        testTransformation2: {impl: function() { return this;}}
+                    }
+                },
+                textNode: {
+                    transformations: {
+                        textTestTransformation: function() { return this; }
+                    }
+                },
+                elementNode: {
+                    transformations: {
+                        elementTestTransformation: function() { return this; }
+                    }
+                }
+            };
             
             doc.registerExtension(extension);
 
@@ -924,6 +932,12 @@ describe('smartxml', function() {
             expect(elementNode.testTransformation2().sameNode(elementNode)).to.equal(true, '2');
             expect(textNode.testTransformation().sameNode(textNode)).to.equal(true, '3');
             expect(textNode.testTransformation2().sameNode(textNode)).to.equal(true, '4');
+
+            expect(elementNode.elementTestTransformation().sameNode(elementNode)).to.be.true;
+            expect(elementNode.textTestTransformation).to.be.undefined;
+        
+            expect(textNode.textTestTransformation().sameNode(textNode)).to.be.true;
+            expect(textNode.elementTestTransfomation).to.be.undefined; 
         });
     });