Add tests

Author: adityapatwardhan

src/code/ContainerRegistryServerAPICalls.cs

@@ -1698,9 +1698,11 @@ private Hashtable[] FindPackagesWithVersionHelper(string packageName, VersionTyp
 
         private string PrependMARPrefix(string packageName)
         {
+            string prefix = string.IsNullOrEmpty(InternalHooks.MARPrefix) ? PSRepositoryInfo.MARPrefix : InternalHooks.MARPrefix;
+
             // If the repostitory is MAR and its not a wildcard search, we need to prefix the package name with MAR prefix.
             string updatedPackageName = Repository.IsMARRepository() && packageName.Trim() != "*"
-                                            ? string.Concat(PSRepositoryInfo.MARPrefix, packageName)
+                                            ? string.Concat(prefix, packageName)
                                             : packageName;
 
             return updatedPackageName;

src/code/InternalHooks.cs

@@ -2,6 +2,7 @@
 // Licensed under the MIT License.
 
 using System.Reflection;
+using Azure.Core;
 
 namespace Microsoft.PowerShell.PSResourceGet.UtilClasses
 {
@@ -15,6 +16,8 @@ public class InternalHooks
 
         internal static string AllowedUri;
 
+        internal static string MARPrefix;
+
         public static void SetTestHook(string property, object value)
         {
             var fieldInfo = typeof(InternalHooks).GetField(property, BindingFlags.Static | BindingFlags.NonPublic);

test/FindPSResourceTests/FindPSResourceContainerRegistryServer.Tests.ps1

@@ -228,3 +228,21 @@ Describe 'Test HTTP Find-PSResource for ACR Server Protocol' -tags 'CI' {
         $res.Type.ToString() | Should -Be "Script"
     }
 }
+
+Describe 'Test Find-PSResource for MAR Repository' -tags 'CI' {
+    BeforeAll {
+        [Microsoft.PowerShell.PSResourceGet.UtilClasses.InternalHooks]::SetTestHook("MARPrefix", "azure-powershell/");
+        Register-PSResourceRepository -Name "MAR" -Uri "https://mcr.microsoft.com" -ApiVersion "ContainerRegistry"
+    }
+
+    AfterAll {
+        [Microsoft.PowerShell.PSResourceGet.UtilClasses.InternalHooks]::SetTestHook("MARPrefix", $null);
+        Unregister-PSResourceRepository -Name "MAR"
+    }
+
+    It "Should find resource given specific Name, Version null" {
+        $res = Find-PSResource -Name "Az.Accounts" -Repository "MAR"
+        $res.Name | Should -Be "Az.Accounts"
+        $res.Version | Should -Be "3.0.4"
+    }
+}

test/InstallPSResourceTests/InstallPSResourceContainerRegistryServer.Tests.ps1

@@ -137,7 +137,7 @@ Describe 'Test Install-PSResource for ACR scenarios' -tags 'CI' {
 
     It "Install resource with a dependency (should install both parent and dependency)" {
         Install-PSResource -Name $testModuleParentName -Repository $ACRRepoName -TrustRepository
-        
+
         $parentPkg = Get-InstalledPSResource $testModuleParentName
         $parentPkg.Name | Should -Be $testModuleParentName
         $parentPkg.Version | Should -Be "1.0.0"
@@ -307,3 +307,28 @@ Describe 'Test Install-PSResource for V3Server scenarios' -tags 'ManualValidatio
         Set-PSResourceRepository PoshTestGallery -Trusted
     }
 }
+
+Describe 'Test Install-PSResource for MAR Repository' -tags 'CI' {
+    BeforeAll {
+        [Microsoft.PowerShell.PSResourceGet.UtilClasses.InternalHooks]::SetTestHook("MARPrefix", "azure-powershell/");
+        Register-PSResourceRepository -Name "MAR" -Uri "https://mcr.microsoft.com" -ApiVersion "ContainerRegistry"
+    }
+
+    AfterAll {
+        [Microsoft.PowerShell.PSResourceGet.UtilClasses.InternalHooks]::SetTestHook("MARPrefix", $null);
+        Unregister-PSResourceRepository -Name "MAR"
+    }
+
+    It "Should find resource given specific Name, Version null" {
+        try {
+            $pkg = Install-PSResource -Name "Az.Accounts" -Repository "MAR" -PassThru -TrustRepository -Reinstall
+            $pkg.Name | Should -Be "Az.Accounts"
+            $pkg.Version | Should -Be "3.0.4"
+        }
+        finally {
+            if ($pkg) {
+                Uninstall-PSResource -Name "Az.Accounts" -Version "3.0.4"
+            }
+        }
+    }
+}

test/PublishPSResourceTests/PublishPSResourceContainerRegistryServer.Tests.ps1

@@ -512,3 +512,24 @@ Describe "Test Publish-PSResource" -tags 'CI' {
         $results[0].Version | Should -Be $version
     }
 }
+
+Describe 'Test Publish-PSResource for MAR Repository' -tags 'CI' {
+    BeforeAll {
+        [Microsoft.PowerShell.PSResourceGet.UtilClasses.InternalHooks]::SetTestHook("MARPrefix", "azure-powershell/");
+        Register-PSResourceRepository -Name "MAR" -Uri "https://mcr.microsoft.com" -ApiVersion "ContainerRegistry"
+    }
+
+    AfterAll {
+        [Microsoft.PowerShell.PSResourceGet.UtilClasses.InternalHooks]::SetTestHook("MARPrefix", $null);
+        Unregister-PSResourceRepository -Name "MAR"
+    }
+
+    It "Should find resource given specific Name, Version null" {
+        $fileName = "NonExistent.psd1"
+        $modulePath = New-Item -Path "$TestDrive\NonExistent" -ItemType Directory -Force
+        $psd1Path = Join-Path -Path $modulePath -ChildPath $fileName
+        New-ModuleManifest -Path $psd1Path -ModuleVersion "1.0.0" -Description "NonExistent module"
+
+        { Publish-PSResource -Path $modulePath -Repository "MAR" -ErrorAction Stop } | Should -Throw
+    }
+}