Use TestNG instead of JUnit

Also include dependency for mockito.