Made updates as per PR review

Author: timayabi2020

src/Authentication/Authentication/custom/common/GraphUri.ps1

@@ -103,14 +103,21 @@ function GraphUri_RemoveNamespaceFromActionFunction {
     $ActionFunctionFQNPattern = "\/Microsoft.Graph.(.*)$"
 
     $NewUri = $Uri
-    # Remove FQN in action/function names.
+    # Remove FQN in paths.
     if ($Uri -match $ActionFunctionFQNPattern) {
         $MatchedUriSegment = $Matches.0
+        $SegmentBuilder = ""
         # Trim nested namespace segments.
-        $NestedNamespaceSegments = $Matches.1 -split "\."
-        # Remove trailing '()' from functions.
-        $LastSegment = $NestedNamespaceSegments[-1] -replace "\(\)", ""
-        $NewUri = $Uri -replace [Regex]::Escape($MatchedUriSegment), "/$LastSegment"
+        $NestedNamespaceSegments = $Matches.1 -split "/"
+        foreach($Segment in $NestedNamespaceSegments){
+            # Remove microsoft.graph prefix and trailing '()' from functions.
+            $Segment = $segment.Replace("microsoft.graph.","").Replace("()", "")
+            # Get resource object name from segment if it exists. e.g get 'updateAudience' from windowsUpdates.updateAudience
+            $ResourceObj = $Segment.Split(".")
+            $Segment = $ResourceObj.Count -gt 1 ? $ResourceObj[$ResourceObj.Count-1] : $Segment       
+            $SegmentBuilder += "/$Segment"
+        }
+        $NewUri = $Uri -replace [Regex]::Escape($MatchedUriSegment), $SegmentBuilder
     }
 
     return $NewUri

src/Authentication/Authentication/test/Find-MgGraphCommand.Tests.ps1

@@ -138,7 +138,7 @@ Describe "Find-MgGraphCommand Command" {
                 $MgCommand.Command | Should -Be @("Get-MgReportSharePointActivityUserCount", "Get-MgBetaReportSharePointActivityUserCount")
             } | Should -Not -Throw    
         }
-        It 'Should find commands for uri woth /me segments' {
+        It 'Should find commands for uri with /me segments' {
             {
                 $MgCommand = Find-MgGraphCommand -Uri "/me/events/"
                 $MgCommand | Should -HaveCount 4
@@ -150,7 +150,7 @@ Describe "Find-MgGraphCommand Command" {
                 $MgCommand.Command | Select-Object -Unique | Should -BeIn @("New-MgUserEvent", "Get-MgUserEvent", "New-MgBetaUserEvent", "Get-MgBetaUserEvent")
             } | Should -Not -Throw
         }
-        It 'Should find commands for uri woth /me segments' {
+        It 'Should find commands for uri with /me segments' {
             {
                 $MgCommand = Find-MgGraphCommand -Uri "https://graph.microsoft.com/v1.0/me/events/"
                 $MgCommand | Should -HaveCount 4
@@ -162,6 +162,18 @@ Describe "Find-MgGraphCommand Command" {
                 $MgCommand.Command | Select-Object -Unique | Should -BeIn @("New-MgUserEvent", "Get-MgUserEvent", "New-MgBetaUserEvent", "Get-MgBetaUserEvent")
             } | Should -Not -Throw
         }
+        It 'Should find commands for uri with Microsoft.Graph prefix in nested segments' {
+            {
+                $MgCommand = Find-MgGraphCommand -Uri "/identity/authenticationEventsFlows/{authenticationEventsFlow-id}/microsoft.graph.externalUsersSelfServiceSignUpEventsFlow/onAuthenticationMethodLoadStart/microsoft.graph.onAuthenticationMethodLoadStartExternalUsersSelfServiceSignUp/identityProviders"
+                $MgCommand | Should -HaveCount 1
+                $MgCommand.Command | Select-Object -Unique | should -HaveCount 1
+                $MgCommand.Method | Select-Object -Unique | should -HaveCount 1
+                $MgCommand.APIVersion | Select-Object -Unique | should -HaveCount 1
+                $MgCommand.Variants | Select-Object -Unique | should -HaveCount 1
+                $MgCommand.URI | Select-Object -Unique | Should -Be "/identity/authenticationEventsFlows/{authenticationEventsFlow-id}/externalUsersSelfServiceSignUpEventsFlow/onAuthenticationMethodLoadStart/onAuthenticationMethodLoadStartExternalUsersSelfServiceSignUp/identityProviders"
+                $MgCommand.Command | Select-Object -Unique | Should -BeIn @("Get-MgBetaIdentityAuthenticationEventFlowAsOnAuthenticationMethodLoadStartExternalUserSelfServiceSignUpIdentityProvider")
+            } | Should -Not -Throw
+        }
     }
 
     Context "FindByCommand" {

tools/PostGeneration/NewCommandMetadata.ps1

@@ -64,10 +64,13 @@ $ApiVersion | ForEach-Object {
                 $SegmentBuilder = ""
                 # Trim nested namespace segments.
                 $NestedNamespaceSegments = $Matches.1 -split "/"
-                # Remove microsoft.graph prefix and remove trailing '()' from functions.
-                foreach($segment in $NestedNamespaceSegments){
-                    $segment = $segment.Replace("microsoft.graph.","").Replace("()", "")
-                    $SegmentBuilder += "/$segment"
+                foreach($Segment in $NestedNamespaceSegments){
+                    # Remove microsoft.graph prefix and trailing '()' from functions.
+                    $Segment = $segment.Replace("microsoft.graph.","").Replace("()", "")
+                    # Get resource object name from segment if it exists. e.g get 'updateAudience' from windowsUpdates.updateAudience
+                    $ResourceObj = $Segment.Split(".")
+                    $Segment = $ResourceObj.Count -gt 1 ? $ResourceObj[$ResourceObj.Count-1] : $Segment       
+                    $SegmentBuilder += "/$Segment"
                 }
                 $Uri = $Uri -replace [Regex]::Escape($MatchedUriSegment), $SegmentBuilder
             }