From 41e99a6123d7fa32a349dd688eb12e73061b6283 Mon Sep 17 00:00:00 2001
From: Irvine Sunday <40403681+irvinesunday@users.noreply.github.com>
Date: Tue, 14 Nov 2023 11:07:28 +0300
Subject: [PATCH] Adds referenceability for attributes nav. prop (#482)

---
 transforms/csdl/preprocess_csdl.xsl             | 9 +++++++++
 transforms/csdl/preprocess_csdl_test_input.xml  | 3 +++
 transforms/csdl/preprocess_csdl_test_output.xml | 9 +++++++++
 3 files changed, 21 insertions(+)

diff --git a/transforms/csdl/preprocess_csdl.xsl b/transforms/csdl/preprocess_csdl.xsl
index ddc6cfe9..419798cb 100644
--- a/transforms/csdl/preprocess_csdl.xsl
+++ b/transforms/csdl/preprocess_csdl.xsl
@@ -354,6 +354,15 @@
             </xsl:call-template>
         </xsl:copy>
     </xsl:template>
+
+    <xsl:template match="edm:Schema[@Namespace='microsoft.graph']/edm:ComplexType[@Name='onAttributeCollectionExternalUsersSelfServiceSignUp']/edm:NavigationProperty[@Name='attributes']">
+        <xsl:copy>
+            <xsl:copy-of select="@* | node()" />
+            <xsl:call-template name="ReferenceableRestrictionsTemplate">
+                <xsl:with-param name="referenceable">true</xsl:with-param>
+            </xsl:call-template>
+        </xsl:copy>
+    </xsl:template>
     
     <xsl:template match="edm:Schema[@Namespace='microsoft.graph']/edm:EntityType[@Name='user']/edm:NavigationProperty[@Name='createdObjects']|
                          edm:Schema[@Namespace='microsoft.graph']/edm:EntityType[@Name='servicePrincipal']/edm:NavigationProperty[@Name='createdObjects']">
diff --git a/transforms/csdl/preprocess_csdl_test_input.xml b/transforms/csdl/preprocess_csdl_test_input.xml
index 9ce26291..f24083f7 100644
--- a/transforms/csdl/preprocess_csdl_test_input.xml
+++ b/transforms/csdl/preprocess_csdl_test_input.xml
@@ -102,6 +102,9 @@
                 <Property Name="onAttributeCollection" Type="graph.onAttributeCollectionHandler" />
                 <Property Name="onAuthenticationMethodLoadStart" Type="graph.onAuthenticationMethodLoadStartHandler" />
             </EntityType>
+            <ComplexType Name="onAttributeCollectionExternalUsersSelfServiceSignUp" BaseType="graph.onAttributeCollectionHandler">
+                <NavigationProperty Name="attributes" Type="Collection(graph.identityUserFlowAttribute)" />
+            </ComplexType>
             <ComplexType Name="onAuthenticationMethodLoadStartExternalUsersSelfServiceSignUp" BaseType="graph.onAuthenticationMethodLoadStartHandler">
                 <NavigationProperty Name="identityProviders" Type="Collection(graph.identityProviderBase)" />
             </ComplexType>
diff --git a/transforms/csdl/preprocess_csdl_test_output.xml b/transforms/csdl/preprocess_csdl_test_output.xml
index 4e5aeb5a..ecf84f85 100644
--- a/transforms/csdl/preprocess_csdl_test_output.xml
+++ b/transforms/csdl/preprocess_csdl_test_output.xml
@@ -513,6 +513,15 @@
           </Annotation>
         </Property>
       </EntityType>
+      <ComplexType Name="onAttributeCollectionExternalUsersSelfServiceSignUp" BaseType="graph.onAttributeCollectionHandler">
+        <NavigationProperty Name="attributes" Type="Collection(graph.identityUserFlowAttribute)">
+          <Annotation Term="Org.OData.Capabilities.V1.NavigationRestrictions">
+            <Record>
+              <PropertyValue Property="Referenceable" Bool="true" />
+            </Record>
+          </Annotation>
+        </NavigationProperty>
+      </ComplexType>
       <ComplexType Name="onAuthenticationMethodLoadStartExternalUsersSelfServiceSignUp" BaseType="graph.onAuthenticationMethodLoadStartHandler">
         <NavigationProperty Name="identityProviders" Type="Collection(graph.identityProviderBase)">
           <Annotation Term="Org.OData.Capabilities.V1.NavigationRestrictions">