Mock based unit test working.