Added Mockito timeout on test involving asynchronous code execution