diff --git a/tests/src/Unit/DomHelperTraitTest.php b/tests/src/Unit/DomHelperTraitTest.php
index f547abb..6445969 100644
--- a/tests/src/Unit/DomHelperTraitTest.php
+++ b/tests/src/Unit/DomHelperTraitTest.php
@@ -48,16 +48,41 @@ class DomHelperTraitTest extends UnitTestCase {
 
   /**
    * Tests DomHelperTrait::setNodeContent().
+   *
+   * @dataProvider providerTestSetNodeContent
+   */
+  public function testSetNodeContent($content, $expected_output) {
+    $this->setNodeContent($this->node, $content);
+    $this->assertEquals(Html::serialize($this->document), $expected_output);
+  }
+
+  /**
+   * @return array
+   * @see ::testSetNodeContent()
    */
-  public function testSetNodeContent() {
-    $this->setNodeContent($this->node, '<div><replacement replaced="true" /></div>');
-    $this->assertEquals(Html::serialize($this->document), '<outer><test foo="bar" namespace:foo="bar"><div><replacement replaced="true"></replacement></div></test></outer>');
-    // Test replacing with an empty value.
-    $this->setNodeContent($this->node, '');
-    $this->assertEquals(Html::serialize($this->document), '<outer><test foo="bar" namespace:foo="bar"></test></outer>');
-    // Test replacing again with a non-empty value.
-    $this->setNodeContent($this->node, '<div></div>');
-    $this->assertEquals(Html::serialize($this->document), '<outer><test foo="bar" namespace:foo="bar"><div></div></test></outer>');
+  public function providerTestSetNodeContent() {
+    return [
+      'empty' => [
+        '',
+        '<outer><test foo="bar" namespace:foo="bar"></test></outer>',
+      ],
+      'single node without children' => [
+        '<div></div>',
+        '<outer><test foo="bar" namespace:foo="bar"><div></div></test></outer>',
+      ],
+      'single node with children' => [
+        '<div><replacement replaced="true" /></div>',
+        '<outer><test foo="bar" namespace:foo="bar"><div><replacement replaced="true"></replacement></div></test></outer>',
+      ],
+      'multiple nodes' => [
+        '<p>first</p><p>second</p>',
+        '<outer><test foo="bar" namespace:foo="bar"><p>first</p><p>second</p></test></outer>',
+      ],
+      'multiple nodes, with a text node, comment node and element node' => [
+        'Second <!-- comment --> <p>third</p>',
+        '<outer><test foo="bar" namespace:foo="bar">Second <!-- comment --> <p>third</p></test></outer>',
+      ]
+    ];
   }
 
   /**