From 868e820ee622fcad7d9782de2e66b2cbd55ad679 Mon Sep 17 00:00:00 2001
From: "Odd B. Henriksen" <odd@digicon.no>
Date: Sun, 20 Oct 2019 23:21:59 +0200
Subject: [PATCH] Fixes #122 Methods with undefined return type should be
 treated like void methods.

---
 Code/Functions/Public/Write-CUPesterTest.ps1 | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/Code/Functions/Public/Write-CUPesterTest.ps1 b/Code/Functions/Public/Write-CUPesterTest.ps1
index f037171..ae21e4f 100644
--- 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,7 +422,7 @@ Function Write-CUPesterTest {
                 
                 [void]$sb.AppendLine("# -- Assert")
                 [void]$sb.AppendLine("")
-                If ($Method.ReturnType -eq '[void]' -or $Null -eq $Method.ReturnType) {
+                If (!($Method.ReturnType) -or $Method.ReturnType -eq '[void]') {
                     [void]$sb.AppendLine("$MethodCall" + '| Should -Be $null')
                 }
                 else {