From c57106661da6c5a39a771137e082de0b5967cfe9 Mon Sep 17 00:00:00 2001
From: Steven Orvell <sorvell@google.com>
Date: Mon, 3 Apr 2017 18:40:21 -0700
Subject: [PATCH] Adds tests from https://github.com/Polymer/polymer/pull/4099.
 The other changes from the PR are no longer needed.

---
 test/unit/polymer.element.html | 52 ++++++++++++++++++++++++++++++++++
 1 file changed, 52 insertions(+)

diff --git a/test/unit/polymer.element.html b/test/unit/polymer.element.html
index e6273517e1..02afd0d6be 100644
--- a/test/unit/polymer.element.html
+++ b/test/unit/polymer.element.html
@@ -288,6 +288,33 @@ <h1>Sub template</h1>
   });
   </script>
 
+  <dom-module id="sub-extended-template">
+    <script>
+    HTMLImports.whenReady(function() {
+      // Setting up property bind listeners in both base and sub class
+      // is required to catch regression bug
+      class ElementWithBindListeners extends window.MyElement {
+        static get is() { return 'element-with-bind-listeners'; }
+      }
+      class SubExtendedTemplate extends ElementWithBindListeners {
+        static get is() { return 'sub-extended-template'; }
+        static get template() {
+          if (!this.hasOwnProperty('_template')) {
+            const baseClass = Object.getPrototypeOf(this.prototype).constructor;
+            this._template = baseClass.template.cloneNode(true);
+            const newElement = document.createElement('div');
+            newElement.innerHTML = '<div id="subContent" imaginary-prop$="{{prop}}""></div>';
+            this._template.content.appendChild(newElement.firstChild);
+          }
+          return this._template;
+        }
+      }
+      customElements.define(ElementWithBindListeners.is, ElementWithBindListeners);
+      customElements.define(SubExtendedTemplate.is, SubExtendedTemplate);
+    })
+    </script>
+  </dom-module>
+
   <test-fixture id="my-element-attr">
     <template>
       <my-element prop="attr"></my-element>
@@ -545,6 +572,31 @@ <h1>Sub template</h1>
     })
   });
 
+  suite('extend sub-class template', function() {
+    var el, baseEl;
+    setup(function() {
+      el = document.createElement('sub-extended-template');
+      document.body.appendChild(el);
+      baseEl = document.createElement('element-with-bind-listeners');
+      document.body.appendChild(baseEl);
+    });
+    teardown(function() {
+      document.body.removeChild(el);
+      document.body.removeChild(baseEl);
+    });
+    test('has original template', function() {
+      assert.equal(el.prop, 'base');
+      assert.equal(el.$.content.textContent, el.prop);
+    });
+    test('has new element', function() {
+      assert.ok(el.$.subContent);
+      assert.equal(el.$.subContent.getAttribute('imaginary-prop'), 'base');
+    });
+    test('base doesnt have new element', function() {
+      assert.notOk(baseEl.$.subContent);
+    });
+  });
+
   </script>
 </body>
 </html>