Only add layout attribute to AMP components

Author: pierlon

includes/sanitizers/class-amp-layout-sanitizer.php

@@ -20,8 +20,11 @@ class AMP_Layout_Sanitizer extends AMP_Base_Sanitizer {
 	public function sanitize() {
 		$xpath = new DOMXPath( $this->dom );
 
-		// Elements with the `layout` attribute will be validated by `AMP_Tag_And_Attribute_Sanitizer`.
-		$nodes = $xpath->query( '//*[ not( @layout ) and ( @data-amp-layout or @width or @height or @style ) ]' );
+		/**
+		 * Sanitize AMP nodes to be AMP compatible. Elements with the `layout` attribute will be validated by
+		 * `AMP_Tag_And_Attribute_Sanitizer`.
+		 */
+		$nodes = $xpath->query( '//*[ starts-with( name(), "amp-" ) and not( @layout ) and ( @data-amp-layout or @width or @height or @style ) ]' );
 
 		foreach ( $nodes as $node ) {
 			/**

tests/php/test-amp-layout-sanitizer.php

@@ -19,6 +19,11 @@ class AMP_Layout_Sanitizer_Test extends WP_UnitTestCase {
 	 */
 	public function get_body_data() {
 		return [
+			'non_amp_component'                           => [
+				'<svg height="10%" width="10%"></svg>',
+				'<svg height="10%" width="10%"></svg>',
+			],
+
 			'no_width_or_height'                          => [
 				'<amp-img src="foo.jpg" data-amp-layout="fill"></amp-img>',
 				'<amp-img src="foo.jpg" layout="fill"></amp-img>',