Proposed fix for location issues

Author: edyoung

Tests/PSGetUpdateModule.Tests.ps1

@@ -91,31 +91,59 @@ Describe UpdateModuleFromAlternateRepo -Tags 'BVT' {
         PSGetTestUtils\Uninstall-Module ContosoClient
     }
 
-    It "Check situation" {
+    It "Check that removing a slash from a repo doesn't break update" {
         $withSlash = "https://www.poshtestgallery.com/api/v2/"
         $noSlash = "https://www.poshtestgallery.com/api/v2"
-        Write-Host (Get-PSRepository | Out-String)
+        #Write-Host (Get-PSRepository | Out-String)
         (Get-PSRepository PSGallery).SourceLocation | Should Be $withSlash
 
         Install-Module ContosoServer -RequiredVersion 1.0
         (Get-InstalledModule ContosoServer).RepositorySourceLocation | Should Be $withSlash
-        Write-Host (Get-InstalledModule ContosoServer -AllVersions | Format-List | Out-String)
+        #Write-Host (Get-InstalledModule ContosoServer -AllVersions | Format-List | Out-String)
 
         # now update where PSGallery Source Location is
         Set-PSGallerySourceLocation -Location $noSlash
-        Write-Host (Get-PSRepository | Out-String)
+        #Write-Host (Get-PSRepository | Out-String)
         (Get-PSRepository PSGallery).SourceLocation | Should Be $noSlash
 
         # reload powershellget to force-update cached repository info
         Import-Module PowerShellGet -Force
 
         # now try and update module isntalled using other SourceLocation
         Update-Module ContosoServer -RequiredVersion 2.0 -ErrorAction Stop
-        Write-Host (Get-InstalledModule ContosoServer -AllVersions | Format-List | Out-String)
+        #Write-Host (Get-InstalledModule ContosoServer -AllVersions | Format-List | Out-String)
         (Get-InstalledModule ContosoServer).RepositorySourceLocation | Should Be $noSlash
 
         (Get-InstalledModule ContosoServer).Version | Should Be 2.0
+    }
+
+    It "Check that adding a slash to a repo doesn't break update" {
+        $withSlash = "https://www.poshtestgallery.com/api/v2/"
+        $noSlash = "https://www.poshtestgallery.com/api/v2"
+        #Write-Host (Get-PSRepository | Out-String)
+
+        Set-PSGallerySourceLocation -Location $noSlash
+
+        (Get-PSRepository PSGallery).SourceLocation | Should Be $noSlash
+
+        Install-Module ContosoServer -RequiredVersion 1.0
+        (Get-InstalledModule ContosoServer).RepositorySourceLocation | Should Be $noSlash
+        #Write-Host (Get-InstalledModule ContosoServer -AllVersions | Format-List | Out-String)
 
+        # now update where PSGallery Source Location is
+        Set-PSGallerySourceLocation -Location $withSlash
+        #Write-Host (Get-PSRepository | Out-String)
+        (Get-PSRepository PSGallery).SourceLocation | Should Be $withSlash
+
+        # reload powershellget to force-update cached repository info
+        Import-Module PowerShellGet -Force
+
+        # now try and update module isntalled using other SourceLocation
+        Update-Module ContosoServer -RequiredVersion 2.0 -ErrorAction Stop
+        #Write-Host (Get-InstalledModule ContosoServer -AllVersions | Format-List | Out-String)
+        (Get-InstalledModule ContosoServer).RepositorySourceLocation | Should Be $withSlash
+
+        (Get-InstalledModule ContosoServer).Version | Should Be 2.0
     }
 }
 

src/PowerShellGet/private/functions/Get-SourceName.ps1

@@ -1,25 +1,22 @@
-function Get-SourceName
-{
+function Get-SourceName {
     [CmdletBinding()]
     [OutputType("string")]
     Param
     (
-        [Parameter(Mandatory=$true)]
+        [Parameter(Mandatory = $true)]
         [ValidateNotNullOrEmpty()]
         [string]
         $Location
     )
 
     Set-ModuleSourcesVariable
 
-    foreach($psModuleSource in $script:PSGetModuleSources.Values)
-    {
-        if(($psModuleSource.Name -eq $Location) -or
-           ($psModuleSource.SourceLocation -eq $Location) -or
-           ((Get-Member -InputObject $psModuleSource -Name $script:ScriptSourceLocation) -and
-           ($psModuleSource.ScriptSourceLocation -eq $Location)))
-        {
+    foreach ($psModuleSource in $script:PSGetModuleSources.Values) {
+        if (($psModuleSource.Name -eq $Location) -or
+            (Test-EquivalentLocation -LocationA $psModuleSource.SourceLocation -LocationB $Location) -or
+            ((Get-Member -InputObject $psModuleSource -Name $script:ScriptSourceLocation) -and
+                (Test-EquivalentLocation -LocationA $psModuleSource.ScriptSourceLocation -LocationB $Location))) {
             return $psModuleSource.Name
         }
     }
-}
+}

src/PowerShellGet/private/functions/Test-EquivalentLocation.ps1

@@ -0,0 +1,18 @@
+
+# Compare 2 strings, ignoring any trailing slashes or backslashes.
+# This is not exactly the same as URL or path equivalence but it should work in practice
+function Test-EquivalentLocation {
+    [CmdletBinding()]
+    [OutputType("bool")]
+    param(
+        [Parameter(Mandatory = $false)]
+        [string]$LocationA,
+
+        [Parameter(Mandatory = $false)]
+        [string]$LocationB
+    )
+
+    $LocationA = $LocationA.TrimEnd("\/")
+    $LocationB = $LocationB.TrimEnd("\/")
+    return $LocationA -eq $LocationB
+}