Replacing setUp methods with @InjectMocks.
Cleaning up unused imports and some formatting.