Completed unit tests using Mockito.