From c5efd1ef362f7fade7e14705c61bab3cc716b712 Mon Sep 17 00:00:00 2001
From: Matthew Phillips <matthew@matthewphillips.info>
Date: Mon, 22 Oct 2018 16:21:16 -0400
Subject: [PATCH] Make work when documentElement is removed

This makes can-component work when the documentElement is removed. Fixes
---
 can-component.js                 |  4 +++-
 test/component-in-stache-test.js | 27 +++++++++++++++++++++++++++
 2 files changed, 30 insertions(+), 1 deletion(-)

diff --git a/can-component.js b/can-component.js
index 8c0fe9a..9344122 100644
--- a/can-component.js
+++ b/can-component.js
@@ -488,7 +488,9 @@ var Component = Construct.extend(
 				});
 			} else {
 				var removalDisposal = domMutate.onNodeRemoval(el, function () {
-					if (!el.ownerDocument.documentElement.contains(el)) {
+					var doc = el.ownerDocument;
+					var rootNode = doc.contains ? doc : doc.documentElement;
+					if (!rootNode.contains(el)) {
 						removalDisposal();
 						callTeardownFunctions();
 					}
diff --git a/test/component-in-stache-test.js b/test/component-in-stache-test.js
index 3876182..ff1e185 100644
--- a/test/component-in-stache-test.js
+++ b/test/component-in-stache-test.js
@@ -8,6 +8,8 @@ var QUnit = require("steal-qunit");
 var viewModel = require("can-view-model");
 var canDev = require("can-log/dev/dev");
 var testHelpers = require("can-test-helpers");
+var domMutate = require("can-dom-mutate");
+var globals = require("can-globals");
 
 QUnit.module("can-component can be rendered by can-stache");
 
@@ -117,3 +119,28 @@ QUnit.test("Component can be removed from the page", 3, function(){
 	prop.set(4);
 	QUnit.equal(frag.firstChild.getElementsByTagName("to-be-removed")[0].innerHTML, "4");
 });
+
+QUnit.test("Cleans up itself on the documentElement removal", function() {
+	Component.extend({
+		tag: "ssr-cleanup",
+		view: "hello world",
+		ViewModel: {}
+	});
+
+	var doc = document.implementation.createHTMLDocument("Test");
+	var realDoc = globals.getKeyValue("document");
+	globals.setKeyValue("document", doc);
+
+	var frag = stache("<ssr-cleanup />")({});
+	doc.body.appendChild(frag);
+
+	domMutate.onNodeRemoval(doc.body.firstChild, function() {
+		globals.setKeyValue("document", realDoc);
+		QUnit.ok(true, "Called back without throwing");
+		start();
+	})
+
+	stop();
+
+	doc.removeChild(doc.documentElement);
+});