org.jetbrains.kotlinx.jupyter.test.ResolverTests.kt Maven / Gradle / Ivy

Go to download
package org.jetbrains.kotlinx.jupyter.test

import kotlinx.coroutines.runBlocking
import org.junit.jupiter.api.Test
import org.slf4j.Logger
import org.slf4j.LoggerFactory
import java.io.File
import kotlin.script.experimental.api.ResultWithDiagnostics
import kotlin.script.experimental.dependencies.ExternalDependenciesResolver
import kotlin.script.experimental.dependencies.maven.MavenDependenciesResolver
import kotlin.test.assertTrue

class ResolverTests {
    private val log: Logger by lazy { LoggerFactory.getLogger("resolver") }

    private fun ExternalDependenciesResolver.doResolve(artifact: String): List {
        testRepositories.forEach { addRepository(it, ExternalDependenciesResolver.Options.Empty, sourceCodeLocation = null) }
        assertTrue(acceptsArtifact(artifact))
        val result = runBlocking { resolve(artifact, ExternalDependenciesResolver.Options.Empty, sourceCodeLocation = null) }
        assertTrue(result is ResultWithDiagnostics.Success)
        return result.value
    }

    @Test
    fun resolveSparkMlLibTest() {
        val files = MavenDependenciesResolver().doResolve("org.apache.spark:spark-mllib_2.11:2.4.4")
        log.debug("Downloaded files: ${files.count()}")
        files.forEach {
            log.debug(it.toString())
        }
    }
}