Unit Testing When Then Return Example