/*
 * Copyright 2019-2022 Mamoe Technologies and contributors.
 *
 * 此源代码的使用受 GNU AFFERO GENERAL PUBLIC LICENSE version 3 许可证的约束, 可以在以下链接找到该许可证.
 * Use of this source code is governed by the GNU AGPLv3 license that can be found through the following link.
 *
 * https://github.com/mamoe/mirai/blob/dev/LICENSE
 */


package net.mamoe.mirai.internal.test

import kotlinx.coroutines.test.TestScope
import kotlinx.coroutines.test.runTest
import kotlinx.coroutines.withTimeout
import kotlin.coroutines.CoroutineContext
import kotlin.coroutines.EmptyCoroutineContext
import kotlin.time.Duration

fun runBlockingUnit(
    context: CoroutineContext = EmptyCoroutineContext,
    block: suspend TestScope.() -> Unit
) = runTest(context) { block() }

fun runBlockingUnit(
    context: CoroutineContext = EmptyCoroutineContext,
    timeout: Duration,
    block: suspend TestScope.() -> Unit
) {
    runTest(context, dispatchTimeoutMs = timeout.inWholeMilliseconds) {
        withTimeout(timeout) {
            block()
        }
    }
}