AddressBookTestRevamped.kt

@RunWith(JMockit::class)
class AddressBookTestKt {

    @Injectable
    private lateinit var addressBookRepository: AddressBookRepository
    @Tested
    private lateinit var addressBook: AddressBook

    @Test
    fun retrieveAddressByIdTest() {

        expectations {
            // Instruct mocks
            addressBookRepository.findById(1)
            result = Address(1)
        }

        val address = addressBook.getAddressById(1)
        Assert.assertNotNull(address)

        verifications {
            // Verify behaviour
            addressBookRepository.findById(1)
            times = 1
        }
    }
}