Removed unnecessary code
Using Mockito annotations to use a more concise, orthoginal style for setting up mock behavior
Making assertion more clear - that the test would fail if the function threw an exception.