PXBF-2039-phpstan-fixes: fix undefined method getEntity

GSA · Jan 24, 2025 · d37e906 · d37e906
1 parent 090dd56
commit d37e906
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/...benefit_finder/modules/usagov_benefit_finder_content/usagov_benefit_finder_content.module b/...benefit_finder/modules/usagov_benefit_finder_content/usagov_benefit_finder_content.module
@@ -6,7 +6,9 @@
  */
 
 use Drupal\Core\Database\Database;
+use Drupal\Core\Entity\EntityForm;
 use Drupal\Core\Form\FormStateInterface;
+use Drupal\node\NodeInterface;
 use Drupal\node\Entity\Node;
 
 /**
@@ -482,8 +484,16 @@ function usagov_benefit_finder_content_form_node_bears_life_event_form_edit_form
 function _usagov_benefit_finder_content_check_life_event_form_criteria_depth(array &$form, FormStateInterface $form_state) {
   $error_flag = FALSE;
   $criteria_depths = [];
+  $life_event_form_node = NULL;
+
+  $form_object = $form_state->getFormObject();
+  if ($form_object instanceof EntityForm) {
+    $entity = $form_object->getEntity();
+    if ($entity instanceof NodeInterface) {
+      $life_event_form_node = $entity;
+    }
+  }
 
-  $life_event_form_node = $form_state->getFormObject()->getEntity();
   $sections = $life_event_form_node->get('field_b_sections_elg_criteria')->referencedEntities();
 
   foreach ($sections as $section) {