fix(has-block-type): add call with single block type

volkmann-design-code · Sep 30, 2024 · f651f91 · f651f91
1 parent bf9fed8
commit f651f91
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/src/LayoutHelpers.php b/src/LayoutHelpers.php
@@ -6,11 +6,12 @@
 
 class LayoutHelpers {
 
-    static function hasBlockType(Layout $layout, array $blockTypes): bool
+    static function hasBlockType(Layout $layout, array|string $blockTypes): bool
     {
+        $blockTypesArray = is_array($blockTypes) ? $blockTypes : [$blockTypes];
         foreach ($layout->columns() as $column) {
             foreach ($column->blocks() as $block) {
-                if (in_array($block->type(), $blockTypes)) {
+                if (in_array($block->type(), $blockTypesArray)) {
                     return true;
                 }
             }

diff --git a/tests/LayoutHelpersTest.php b/tests/LayoutHelpersTest.php
@@ -25,7 +25,9 @@ public function testHasBlockType(): void
             ]
         ]);
 
+        $this->assertTrue(LayoutHelpers::hasBlockType($layout, 'heading'));
         $this->assertTrue(LayoutHelpers::hasBlockType($layout, ['heading']));
+        $this->assertFalse(LayoutHelpers::hasBlockType($layout, 'image'));
         $this->assertFalse(LayoutHelpers::hasBlockType($layout, ['image']));
     }