U409-026: Fix previous_part for protected bodies.

ada/ast.py

@@ -2505,7 +2505,8 @@ def protected_previous_part():
         Return the ProtectedDecl corresponding to this node.
         """
         return Entity.defining_name.env_elements.at(0)._.match(
-            lambda prot_decl=T.ProtectedTypeDecl: prot_decl,
+            lambda prot_type=T.ProtectedTypeDecl: prot_type,
+            lambda prot_decl=T.SingleProtectedDecl: prot_decl,
             lambda _: No(T.BasicDecl.entity)
         )
 

testsuite/tests/navigation/protected_body/prot.adb

@@ -0,0 +1,15 @@
+package body Prot is
+   protected body P is
+      entry E (I : in out Integer) when Flag is
+      begin
+         I   := Var + I;
+         Var := I;           --  FLAG
+      end E;
+
+      procedure P (I : Integer) is
+      begin
+         Flag := I > 0;
+      end P;
+   end P;
+   --% node.p_decl_part()
+end Prot;

testsuite/tests/navigation/protected_body/prot.ads

@@ -0,0 +1,11 @@
+package Prot is
+   Var : Integer;
+
+   protected P is
+      entry E (I : in out Integer);
+      procedure P (I : Integer);
+   private
+      Flag : Boolean;
+   end P;
+
+end Prot;

testsuite/tests/navigation/protected_body/test.out

@@ -0,0 +1,5 @@
+Eval 'node.p_decl_part()' on node <ProtectedBody ["P"] prot.adb:2:4-13:10>
+Result: <SingleProtectedDecl ["P"] prot.ads:4:4-9:10>
+
+
+

testsuite/tests/navigation/protected_body/test.yaml

@@ -0,0 +1,2 @@
+driver: inline-playground
+input_sources: [prot.adb, prot.ads]

user_manual/changes/U409-026.yaml

@@ -0,0 +1,6 @@
+type: bugfix
+title: Fix navigation of protected bodies
+description: |
+    This change fixes a bug where calling ``P_Decl_Part`` on a protected body
+    which corresponding declaration was a SingleProtectedDecl would fail.
+date: 2021-04-09