kotlin-fork/libraries/stdlib/test/time/MeasureTimeTest.kt

/*
 * Copyright 2010-2021 JetBrains s.r.o. and Kotlin Programming Language contributors.
 * Use of this source code is governed by the Apache 2.0 license that can be found in the license/LICENSE.txt file.
 */

@file:OptIn(ExperimentalTime::class)
package test.time

import kotlin.random.Random
import kotlin.test.*
import kotlin.time.*

class MeasureTimeTest {

    private fun longRunningCalc(): String = buildString {
        repeat(10) {
            while (Random.nextDouble() >= 0.001);
            append(('a'..'z').random())
        }
    }

    @Test
    fun measureTimeOfCalc() {
        val someResult: String

        val elapsed = measureTime {
            someResult = longRunningCalc()
        }

        println("elapsed: $elapsed")

        assertEquals(10, someResult.length)
        assertTrue(elapsed > Duration.ZERO)
    }

    @Test
    fun measureTimeAndResult() {
        val someResult: String

        val measured: TimedValue<String> = measureTimedValue { longRunningCalc().also { someResult = it } }
        println("measured: $measured")

        val (result, elapsed) = measured

        assertEquals(someResult, result)
        assertTrue(elapsed > Duration.ZERO)
    }


    @Test
    fun measureTimeTestClock() {
        val timeSource = TestTimeSource()
        val expectedNs = Random.nextLong(1_000_000_000L)
        val elapsed = timeSource.measureTime {
            timeSource += Duration.nanoseconds(expectedNs)
        }

        assertEquals(Duration.nanoseconds(expectedNs), elapsed)

        val expectedResult: Long

        val (result, elapsed2) = timeSource.measureTimedValue {
            timeSource += Duration.nanoseconds(expectedNs)
            expectedResult = expectedNs
            expectedNs
        }

        assertEquals(expectedResult, result)
        assertEquals(Duration.nanoseconds(result), elapsed2)
    }
}