Mock test update

Mock tests now mock Write-Host outside of a Context block.