From d52af5599a2bdc72f8223c777dd1d90c5483a2fb Mon Sep 17 00:00:00 2001
From: Tiferet Gazit <tiferet@github.com>
Date: Sat, 20 Nov 2021 02:28:53 +0000
Subject: [PATCH] Allow the neighborhood to be the entire enclosing function

---
 .../experimental/adaptivethreatmodeling/EndpointFeatures.qll   | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/javascript/ql/experimental/adaptivethreatmodeling/lib/experimental/adaptivethreatmodeling/EndpointFeatures.qll b/javascript/ql/experimental/adaptivethreatmodeling/lib/experimental/adaptivethreatmodeling/EndpointFeatures.qll
index 0bb2cf66940..a821ea67363 100644
--- a/javascript/ql/experimental/adaptivethreatmodeling/lib/experimental/adaptivethreatmodeling/EndpointFeatures.qll
+++ b/javascript/ql/experimental/adaptivethreatmodeling/lib/experimental/adaptivethreatmodeling/EndpointFeatures.qll
@@ -185,8 +185,7 @@ module NeighborhoodBodies {
    */
   Raw::AstNode getNeighborhoodAstNode(Raw::AstNode node) {
     if
-      // `node` will always have a parent as we start at and endpoint
-      node.getParentNode() = getOutermostEnclosingFunction(node) or
+      node = getOutermostEnclosingFunction(node) or
       getNumDescendents(node.getParentNode()) > maxNumDescendants()
     then result = node
     else result = getNeighborhoodAstNode(node.getParentNode())