BookingTestsWithBeforeMethod.kt

class SetupTeardownWithDataProvider {
    private lateinit var vehicleType: VehicleType
    private lateinit var orderId: String
    private val booking = Booking()

    @DataProvider
    fun getVehicleTypes(): MutableIterator<Array<Any>> {
        val vehicles = arrayListOf<Array<Any>>()

        vehicles.add(arrayOf(VehicleType.CAR))
        vehicles.add(arrayOf(VehicleType.CAR_XL))

        return vehicles.iterator()
    }

    @BeforeMethod
    fun givenBookingIsCreated(vehicleType: VehicleType) {
        orderId = booking.makeBooking(vehicleType)
    }

    @Test(dataProvider = "getVehicleTypes")
    fun testBookingCreation(vehicleType: VehicleType) {
        val isBookingCreated = booking.doesBookingExists(vehicleType, orderId)
        Assert.assertNotNull(isBookingCreated)
    }

    @Test(dataProvider = "getVehicleTypes")
    fun testBookingCancellation(vehicleType: VehicleType) {
        val isBookingCreated = booking.doesBookingExists(vehicleType, orderId)
        Assert.assertNotNull(isBookingCreated)

        booking.cancelBooking(vehicleType, orderId)
        val isBookingFound = booking.doesBookingExists(vehicleType, orderId)

        Assert.assertNotNull(isBookingCreated)
        Assert.assertFalse(isBookingFound!!)
    }
}