I would vote to write 2 separate tests, one for each of the situations.