Fix mocking commands with single quote in names

BOM is required for PowerShell to properly recognize UTF-8 encoding.
jeffchulg · Feb 18, 2016 · 7dcc6e5 · 7dcc6e5
1 parent 6b51cc5
commit 7dcc6e5
Showing 1 changed file with 10 additions and 3 deletions.
diff --git a/Functions/Mock.ps1 b/Functions/Mock.ps1
@@ -1,4 +1,4 @@
-function Mock {
+function Mock {
 
 <#
 .SYNOPSIS
@@ -273,9 +273,16 @@ about_Mocking
             }
         }
 
+        $EscapeSingleQuotedStringContent =
+            if ($global:PSVersionTable.PSVersion.Major -ge 5) {
+                { [System.Management.Automation.Language.CodeGeneration]::EscapeSingleQuotedStringContent($args[0]) }
+            } else {
+                { $args[0] -replace "['‘’‚‛]", '$&$&' }
+            }
+
         $newContent = & $SafeCommands['Get-Content'] function:\MockPrototype
-        $newContent = $newContent -replace '#FUNCTIONNAME#', $CommandName
-        $newContent = $newContent -replace '#MODULENAME#', $ModuleName
+        $newContent = $newContent -replace '#FUNCTIONNAME#', (& $EscapeSingleQuotedStringContent $CommandName)
+        $newContent = $newContent -replace '#MODULENAME#', (& $EscapeSingleQuotedStringContent $ModuleName)
 
         $canCaptureArgs = 'true'
         if ($contextInfo.Command.CommandType -eq 'Cmdlet' -or