Mock tests beta application

test/module/Common-Functions.ps1

@@ -1,5 +1,10 @@
 $psVersion = $global:PSVersionTable.PSVersion
-$entraVersion = (Get-module Microsoft.Graph.Entra | select version).Version.ToString()
+if($null -ne (Get-Module -Name Microsoft.Graph.Entra.Beta)){
+    $entraVersion = (Get-module Microsoft.Graph.Entra.Beta | select version).Version.ToString() 
+}
+if($null -ne (Get-Module -Name Microsoft.Graph.Entra)){
+    $entraVersion = (Get-module Microsoft.Graph.Entra | select version).Version.ToString() 
+}
 
 function Get-Parameters{
     param(

test/module/Entra/Add-EntraApplicationOwner.Tests.ps1

@@ -0,0 +1,38 @@
+BeforeAll {  
+    if((Get-Module -Name Microsoft.Graph.Entra) -eq $null){
+        Import-Module Microsoft.Graph.Entra      
+    }
+    Import-Module (Join-Path $psscriptroot "..\Common-Functions.ps1") -Force
+
+    Mock -CommandName New-MgApplicationOwnerByRef -MockWith {} -ModuleName Microsoft.Graph.Entra
+}
+
+Describe "Add-EntraApplicationOwner" {
+    Context "Test for Add-EntraApplicationOwner" { 
+        It "Should return empty object"{
+            $result = Add-EntraApplicationOwner -ObjectId "c81d387e-d431-43b4-b12e-f07cbb35b771" -RefObjectId "412be9d1-1460-4061-8eed-cca203fcb215"
+            $result | Should -BeNullOrEmpty           
+
+            Should -Invoke -CommandName New-MgApplicationOwnerByRef -ModuleName Microsoft.Graph.Entra -Times 1
+        }       
+        It "Should fail when parameters are empty" {
+            { Add-EntraApplicationOwner -ObjectId "" -RefObjectId "" } | Should -Throw "Cannot bind argument to parameter*"
+        }
+        It "Should contain ApplicationId in parameters when passed ObjectId to it" {              
+            Mock -CommandName New-MgApplicationOwnerByRef -MockWith {$args} -ModuleName Microsoft.Graph.Entra
+
+            $result = Add-EntraApplicationOwner -ObjectId "c81d387e-d431-43b4-b12e-f07cbb35b771" -RefObjectId "412be9d1-1460-4061-8eed-cca203fcb215"
+            $params = Get-Parameters -data $result
+            $params.ApplicationId | Should -Be "c81d387e-d431-43b4-b12e-f07cbb35b771"
+        }
+        It "Should contain 'User-Agent' header" {
+            Mock -CommandName New-MgApplicationOwnerByRef -MockWith {$args} -ModuleName Microsoft.Graph.Entra
+
+            $userAgentHeaderValue = "PowerShell/$psVersion EntraPowershell/$entraVersion Add-EntraApplicationOwner"
+
+            $result = Add-EntraApplicationOwner -ObjectId "c81d387e-d431-43b4-b12e-f07cbb35b771" -RefObjectId "412be9d1-1460-4061-8eed-cca203fcb215"
+            $params = Get-Parameters -data $result
+            $params.Headers["User-Agent"] | Should -Be $userAgentHeaderValue
+        }
+    }
+}

test/module/Entra/Add-EntraMSAdministrativeUnitMember.Tests.ps1

@@ -0,0 +1,38 @@
+BeforeAll {  
+    if((Get-Module -Name Microsoft.Graph.Entra) -eq $null){
+        Import-Module Microsoft.Graph.Entra      
+    }
+    Import-Module (Join-Path $psscriptroot "..\Common-Functions.ps1") -Force
+
+    Mock -CommandName New-MgDirectoryAdministrativeUnitMemberByRef -MockWith {} -ModuleName Microsoft.Graph.Entra
+}
+
+Describe "Add-EntraMSAdministrativeUnitMember" {
+    Context "Test for Add-EntraMSAdministrativeUnitMember" { 
+        It "Should return empty object"{
+            $result = Add-EntraMSAdministrativeUnitMember -Id "f306a126-cf2e-439d-b20f-95ce4bcb7ffa" -RefObjectId "d6873b36-81d6-4c5e-bec0-9e3ca2c86846"
+            $result | Should -BeNullOrEmpty           
+
+            Should -Invoke -CommandName New-MgDirectoryAdministrativeUnitMemberByRef -ModuleName Microsoft.Graph.Entra -Times 1
+        }       
+        It "Should fail when parameters are empty" {
+            { Add-EntraMSAdministrativeUnitMember -Id "" -RefObjectId "" } | Should -Throw "Cannot bind argument to parameter*"
+        }
+        It "Should contain AdministrativeUnitId in parameters when passed Id to it" {              
+            Mock -CommandName New-MgDirectoryAdministrativeUnitMemberByRef -MockWith {$args} -ModuleName Microsoft.Graph.Entra
+
+            $result = Add-EntraMSAdministrativeUnitMember -Id "f306a126-cf2e-439d-b20f-95ce4bcb7ffa" -RefObjectId "d6873b36-81d6-4c5e-bec0-9e3ca2c86846"
+            $params = Get-Parameters -data $result
+            $params.AdministrativeUnitId | Should -Be "f306a126-cf2e-439d-b20f-95ce4bcb7ffa"
+        }
+        It "Should contain 'User-Agent' header" {
+            Mock -CommandName New-MgDirectoryAdministrativeUnitMemberByRef -MockWith {$args} -ModuleName Microsoft.Graph.Entra
+
+            $userAgentHeaderValue = "PowerShell/$psVersion EntraPowershell/$entraVersion Add-EntraMSAdministrativeUnitMember"
+
+            $result = Add-EntraMSAdministrativeUnitMember -Id "f306a126-cf2e-439d-b20f-95ce4bcb7ffa" -RefObjectId "d6873b36-81d6-4c5e-bec0-9e3ca2c86846"
+            $params = Get-Parameters -data $result
+            $params.Headers["User-Agent"] | Should -Be $userAgentHeaderValue
+        }
+    }
+}

test/module/Entra/Add-EntraMSApplicationOwner.Tests.ps1

@@ -0,0 +1,45 @@
+BeforeAll {  
+    if((Get-Module -Name Microsoft.Graph.Entra) -eq $null){
+        Import-Module Microsoft.Graph.Entra      
+    }
+    Import-Module (Join-Path $psscriptroot "..\Common-Functions.ps1") -Force
+
+    Mock -CommandName New-MgApplicationOwnerByRef -MockWith {} -ModuleName Microsoft.Graph.Entra
+}
+
+Describe "Add-EntraMSApplicationOwner" {
+    Context "Test for Add-EntraMSApplicationOwner" { 
+        It "Should return empty object"{
+            $result = Add-EntraMSApplicationOwner -ObjectId "c81d387e-d431-43b4-b12e-f07cbb35b771" -RefObjectId "412be9d1-1460-4061-8eed-cca203fcb215"
+            $result | Should -BeNullOrEmpty           
+
+            Should -Invoke -CommandName New-MgApplicationOwnerByRef -ModuleName Microsoft.Graph.Entra -Times 1
+        }       
+        It "Should fail when parameters are empty" {
+            { Add-EntraMSApplicationOwner -ObjectId "" -RefObjectId "" } | Should -Throw "Cannot bind argument to parameter*"
+        }
+        It "Should contain ApplicationId in parameters when passed ObjectId to it" {              
+            Mock -CommandName New-MgApplicationOwnerByRef -MockWith {$args} -ModuleName Microsoft.Graph.Entra
+
+            $result = Add-EntraMSApplicationOwner -ObjectId "c81d387e-d431-43b4-b12e-f07cbb35b771" -RefObjectId "412be9d1-1460-4061-8eed-cca203fcb215"
+            $params = Get-Parameters -data $result
+            $params.ApplicationId | Should -Be "c81d387e-d431-43b4-b12e-f07cbb35b771"
+        }
+        It "Should contain OdataId in parameters when passed RefObjectId to it" {              
+            Mock -CommandName New-MgApplicationOwnerByRef -MockWith {$args} -ModuleName Microsoft.Graph.Entra
+
+            $result = Add-EntraMSApplicationOwner -ObjectId "c81d387e-d431-43b4-b12e-f07cbb35b771" -RefObjectId "412be9d1-1460-4061-8eed-cca203fcb215"
+            $params = Get-Parameters -data $result
+            $params.OdataId | Should -Be "https://graph.microsoft.com/v1.0/directoryObjects/412be9d1-1460-4061-8eed-cca203fcb215"
+        }
+        It "Should contain 'User-Agent' header" {
+            Mock -CommandName New-MgApplicationOwnerByRef -MockWith {$args} -ModuleName Microsoft.Graph.Entra
+
+            $userAgentHeaderValue = "PowerShell/$psVersion EntraPowershell/$entraVersion Add-EntraMSApplicationOwner"
+
+            $result = Add-EntraMSApplicationOwner -ObjectId "c81d387e-d431-43b4-b12e-f07cbb35b771" -RefObjectId "412be9d1-1460-4061-8eed-cca203fcb215"
+            $params = Get-Parameters -data $result
+            $params.Headers["User-Agent"] | Should -Be $userAgentHeaderValue
+        }
+    }
+}

test/module/Entra/Add-EntraMSScopedRoleMembership.Tests.ps1

@@ -0,0 +1,54 @@
+BeforeAll {  
+    if((Get-Module -Name Microsoft.Graph.Entra) -eq $null){
+        Import-Module Microsoft.Graph.Entra      
+    }
+    Import-Module (Join-Path $psscriptroot "..\Common-Functions.ps1") -Force
+
+
+    $scriptblock = {
+        # Write-Host "Mocking New-MgDirectoryAdministrativeUnitScopedRoleMember with parameters: $($args | ConvertTo-Json -Depth 3)"
+        return @(
+            [PSCustomObject]@{
+                "Id"                    = "412be9d1-1460-4061-8eed-cca203fcb215"
+                "AdministrativeUnitId"  = "c9ab56cc-e349-4237-856e-cab03157a91e"
+                "RoleId"                = "526b7173-5a6e-49dc-88ec-b677a9093709"
+            }
+        )
+    }
+
+    Mock -CommandName New-MgDirectoryAdministrativeUnitScopedRoleMember -MockWith $scriptblock -ModuleName Microsoft.Graph.Entra
+
+    $RoleMember = New-Object -TypeName Microsoft.Open.MSGraph.Model.MsRolememberinfo
+    $RoleMember.Id = "96d4258f-9187-4e6c-b031-724e789b9323"
+}
+
+Describe "Add-EntraMSScopedRoleMembership" {
+    Context "Test for Add-EntraMSScopedRoleMembership" { 
+        It "Should not return empty object"{
+            $result = Add-EntraMSScopedRoleMembership -Id "c9ab56cc-e349-4237-856e-cab03157a91e" -RoleId "526b7173-5a6e-49dc-88ec-b677a9093709" -RoleMemberInfo $RoleMember
+            $result | Should -Not -BeNullOrEmpty 
+            $result.Id | should -Be @('412be9d1-1460-4061-8eed-cca203fcb215')          
+
+            Should -Invoke -CommandName New-MgDirectoryAdministrativeUnitScopedRoleMember -ModuleName Microsoft.Graph.Entra -Times 1
+        }       
+        It "Should fail when parameters are empty" {
+            { Add-EntraMSScopedRoleMembership -Id "" -RoleId "" } | Should -Throw "Cannot bind argument to parameter*"
+        }
+        It "Should contain AdministrativeUnitId in parameters when passed Id to it" {              
+            Mock -CommandName New-MgDirectoryAdministrativeUnitScopedRoleMember -MockWith {$args} -ModuleName Microsoft.Graph.Entra
+
+            $result = Add-EntraMSScopedRoleMembership -Id "c9ab56cc-e349-4237-856e-cab03157a91e" -RoleId "526b7173-5a6e-49dc-88ec-b677a9093709" -RoleMemberInfo $RoleMember
+            $params = Get-Parameters -data $result
+            $params.AdministrativeUnitId | Should -Be "c9ab56cc-e349-4237-856e-cab03157a91e"
+        }
+        It "Should contain 'User-Agent' header" {
+            Mock -CommandName New-MgDirectoryAdministrativeUnitScopedRoleMember -MockWith {$args} -ModuleName Microsoft.Graph.Entra
+
+            $userAgentHeaderValue = "PowerShell/$psVersion EntraPowershell/$entraVersion Add-EntraMSScopedRoleMembership"
+
+            $result = Add-EntraMSScopedRoleMembership -Id "c9ab56cc-e349-4237-856e-cab03157a91e" -RoleId "526b7173-5a6e-49dc-88ec-b677a9093709" -RoleMemberInfo $RoleMember
+            $params = Get-Parameters -data $result
+            $params.Headers["User-Agent"] | Should -Be $userAgentHeaderValue
+        }
+    }
+}