Improve unit tests using mock.