Property Base Test

MapStatePropertyGenerator.kt

package com.txusballesteros.mytaxi.core.domain.repository

import com.pholser.junit.quickcheck.generator.GenerationStatus
import com.pholser.junit.quickcheck.generator.Generator
import com.pholser.junit.quickcheck.random.SourceOfRandomness
import com.txusballesteros.mytaxi.core.domain.model.MapState

class MapStatePropertyGenerator: Generator<MapState>(MapState::class.java) {
    companion object {
        private val MINIMUM_VALUE = 0
        private val MAXIMUM_VALUE = MapState.values().size.dec()
    }

    override fun generate(random: SourceOfRandomness, status: GenerationStatus): MapState {
        val current = random.nextInt(MINIMUM_VALUE, MAXIMUM_VALUE)
        return MapState.values()[current]
    }
}

MapStatePropertyTest.kt

package com.txusballesteros.mytaxi.core.domain.repository

import com.nhaarman.mockito_kotlin.*
import com.pholser.junit.quickcheck.From
import com.pholser.junit.quickcheck.Property
import com.pholser.junit.quickcheck.runner.JUnitQuickcheck
import com.txusballesteros.mytaxi.core.domain.model.MapState
import com.txusballesteros.mytaxi.core.domain.repository.stream.MapStateStream
import com.txusballesteros.mytaxi.core.testing.PropertyUnitTest
import org.junit.runner.RunWith

@RunWith(JUnitQuickcheck::class)
class MapStateRepositoryPropertyTest: PropertyUnitTest() {
    private val streamMock: MapStateStream = mock()
    private lateinit var repository: MapStateRepository

    override fun onPrepareTest() {
        repository = MapStateRepository(streamMock)
    }

    @Property
    fun shouldRequestNewBooking(@From(MapStatePropertyGenerator::class) mapState: MapState) {
        whenever(streamMock.valueOrNull()).thenReturn(mapState)

        repository.newBookingRequested()

        if (mapState == MapState.RADAR) {
            verify(streamMock).offer(eq(MapState.OFFER))
        } else {
            verify(streamMock, never()).offer(any())
        }
    }

    @Property
    fun shouldRequestCashPayment(@From(MapStatePropertyGenerator::class) mapState: MapState) {
        whenever(streamMock.valueOrNull()).thenReturn(mapState)

        repository.cashPaymentRequested()

        if (mapState == MapState.OFFER) {
            verify(streamMock).offer(eq(MapState.PAYMENT))
        } else {
            verify(streamMock, never()).offer(any())
        }
    }

    @Property
    fun shouldCancelBookingRequest(@From(MapStatePropertyGenerator::class) mapState: MapState) {
        whenever(streamMock.valueOrNull()).thenReturn(mapState)

        repository.cancelBookingRequest()

        verify(streamMock).offer(eq(MapState.RADAR))
    }

    @Property
    fun shouldCancelCashPaymentRequest(@From(MapStatePropertyGenerator::class) mapState: MapState) {
        whenever(streamMock.valueOrNull()).thenReturn(mapState)

        repository.cancelCashPayment()

        verify(streamMock).offer(eq(MapState.RADAR))
    }
}