From 306a93117668fed893e2e6d88b025364ffa9e81f Mon Sep 17 00:00:00 2001
From: Rich-Harris <richard.a.harris@gmail.com>
Date: Sun, 4 Dec 2016 21:09:06 -0500
Subject: [PATCH] test update while yield fragment is torn down, and tweak code
 for consistency

---
 test/compiler/component-yield-if/Widget.html | 22 ++++++++++----------
 test/compiler/component-yield-if/_config.js  | 19 +++++++++++++----
 2 files changed, 26 insertions(+), 15 deletions(-)

diff --git a/test/compiler/component-yield-if/Widget.html b/test/compiler/component-yield-if/Widget.html
index 7dbf91afb5..a2c8d49daa 100644
--- a/test/compiler/component-yield-if/Widget.html
+++ b/test/compiler/component-yield-if/Widget.html
@@ -1,15 +1,15 @@
 <p>
-  {{#if show}}
-    {{yield}}
-  {{/if}}
+	{{#if show}}
+		{{yield}}
+	{{/if}}
 </p>
-<script>
-  export default {
-    data(){
-      return {
-        show: false
-      }
-    }
-  }
 
+<script>
+	export default {
+		data () {
+			return {
+				show: false
+			}
+		}
+	};
 </script>
diff --git a/test/compiler/component-yield-if/_config.js b/test/compiler/component-yield-if/_config.js
index fad768230d..a428e4522d 100644
--- a/test/compiler/component-yield-if/_config.js
+++ b/test/compiler/component-yield-if/_config.js
@@ -1,13 +1,24 @@
 export default {
 	html: '<div><p></p></div>',
+
 	test ( assert, component, target ) {
 		const widget = component.refs.widget;
+
 		assert.equal( widget.get( 'show' ), false );
+
 		widget.set({show: true});
-		assert.equal( target.innerHTML, '<div><p>Hello<!--yield--><!--#if show--></p></div>' );
+		assert.htmlEqual( target.innerHTML, '<div><p>Hello</p></div>' );
+
 		component.set({data: 'World'});
-		assert.equal( target.innerHTML, '<div><p>World<!--yield--><!--#if show--></p></div>' );
+		assert.htmlEqual( target.innerHTML, '<div><p>World</p></div>' );
+
 		widget.set({show: false});
-		assert.equal( target.innerHTML, '<div><p><!--#if show--></p></div>' );
+		assert.htmlEqual( target.innerHTML, '<div><p></p></div>' );
+
+		component.set({data: 'Goodbye'});
+		assert.htmlEqual( target.innerHTML, '<div><p></p></div>' );
+
+		widget.set({show: true});
+		assert.htmlEqual( target.innerHTML, '<div><p>Goodbye</p></div>' );
 	}
-}
+};