Fix for Stephanevg#122

Methods with undefined return types should be treated like void methods
vikingtiger · Oct 20, 2019 · c20fbf1 · c20fbf1
1 parent a097c49
commit c20fbf1
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/Code/Functions/Public/Write-CUPesterTest.ps1 b/Code/Functions/Public/Write-CUPesterTest.ps1
@@ -375,7 +375,7 @@ Function Write-CUPesterTest {
 
                 [void]$sb.AppendLine($visibility)
 
-                If ($Method.ReturnType -eq '[void]' -or $Null -eq $Method.ReturnType) {
+                If (!($Method.ReturnType) -or $Method.ReturnType -eq '[void]') {
                     [void]$sb.AppendLine("It '[$($Class.Name)] --> $($Method.Name)$($Signature) Should not return anything (voided)' {")
                 }
                 else {
@@ -422,12 +422,12 @@ Function Write-CUPesterTest {
 
                 [void]$sb.AppendLine("# -- Assert")
                 [void]$sb.AppendLine("")
-                If ($Method.ReturnType -eq '[void]' -or $Null -eq $Method.ReturnType) {
-                    [void]$sb.AppendLine("$MethodCall" + '| should be $null')
+                If (!($Method.ReturnType) -or $Method.ReturnType -eq '[void]') {
+                    [void]$sb.AppendLine("$MethodCall" + '| Should -Be $null')
                 }
                 else {
 
-                    [void]$sb.AppendLine("($MethodCall).GetType().Name | should be $ReturnType")
+                    [void]$sb.AppendLine("($MethodCall).GetType().Name | Should -Be $ReturnType")
                 }