X-Git-Url: https://git.mdrn.pl/fnpeditor.git/blobdiff_plain/cb5f162babe577ede5a113b49d072b51a97ecaad..58ba604b1c414d5484d8165c38662050809655aa:/src/wlxml/wlxml.test.js

diff --git a/src/wlxml/wlxml.test.js b/src/wlxml/wlxml.test.js
index 516b47c..c37b21f 100644
--- a/src/wlxml/wlxml.test.js
+++ b/src/wlxml/wlxml.test.js
@@ -262,7 +262,7 @@ describe('WLXMLDocument', function() {
         var doc, extension, elementNode, textNode, testClassNode;
 
         beforeEach(function() {
-            doc = getDocumentFromXML('<section>Alice<div class="test_class"></div></section>');
+            doc = getDocumentFromXML('<section>Alice<div class="test_class"></div><div class="test_class.a"></div></section>');
             elementNode = doc.root;
             textNode = doc.root.contents()[0];
             testClassNode = doc.root.contents('.test_class');
@@ -292,9 +292,25 @@ describe('WLXMLDocument', function() {
             expect(testClassNode.object.testTransformation2().sameNode(testClassNode)).to.equal(true, '1');
         });
 
-
+        it('added methods are inherited by nodes with subclasses', function() {
+            extension = {wlxmlClass: {test_class: {methods: {
+                testMethod: function() { return this; }
+            }}}};
+            doc.registerExtension(extension);
+            testClassNode = doc.root.contents()[2];
+            expect(testClassNode.object.testMethod().sameNode(testClassNode)).to.equal(true);
+        });
+        it('added transformations are inherited by nodes with subclasses', function() {
+            extension = {wlxmlClass: {test_class: {transformations: {
+                testTransformation: function() { return this; },
+                testTransformation2: {impl: function() { return this; }}
+            }}}};
+            doc.registerExtension(extension);
+            testClassNode = doc.root.contents()[2];
+            expect(testClassNode.object.testTransformation().sameNode(testClassNode)).to.equal(true, '1');
+            expect(testClassNode.object.testTransformation2().sameNode(testClassNode)).to.equal(true, '2');
+        });
     });
-
 });
 
 });
\ No newline at end of file