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 java.io.File
import kotlin.script.experimental.api.ResultWithDiagnostics
import kotlin.script.experimental.dependencies.ExternalDependenciesResolver
import kotlin.script.experimental.dependencies.RepositoryCoordinates
import kotlin.script.experimental.dependencies.maven.MavenDependenciesResolver
import kotlin.test.assertTrue

class ResolverTests {
    private val logger = testLoggerFactory.getLogger(ResolverTests::class.java)

    private fun ExternalDependenciesResolver.doResolve(artifact: String): List {
        testRepositories.forEach {
            addRepository(RepositoryCoordinates(it.coordinates), 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")
        logger.debug("Downloaded files: ${files.count()}")
        files.forEach {
            logger.debug(it.toString())
        }
    }
}