refactor(test): use local test resources (aallam#190)

Author: aallam

build.gradle.kts

@@ -2,6 +2,9 @@ import com.diffplug.gradle.spotless.SpotlessExtension
 import org.gradle.api.tasks.testing.logging.TestExceptionFormat
 import org.gradle.api.tasks.testing.logging.TestLogEvent.*
 import org.jetbrains.dokka.gradle.DokkaMultiModuleTask
+import org.jetbrains.kotlin.gradle.targets.js.testing.KotlinJsTest
+import org.jetbrains.kotlin.gradle.targets.jvm.tasks.KotlinJvmTest
+import org.jetbrains.kotlin.gradle.targets.native.tasks.KotlinNativeTest
 
 @Suppress("DSL_SCOPE_VIOLATION")
 plugins {
@@ -30,6 +33,19 @@ subprojects {
             showStandardStreams = false
         }
     }
+
+    tasks.withType<KotlinJvmTest>().configureEach {
+        environment("LIB_ROOT", rootDir)
+    }
+
+    tasks.withType<KotlinNativeTest>().configureEach {
+        environment("SIMCTL_CHILD_LIB_ROOT", rootDir)
+        environment("LIB_ROOT", rootDir)
+    }
+
+    tasks.withType<KotlinJsTest>().configureEach {
+        environment("LIB_ROOT", rootDir.toString())
+    }
 }
 
 tasks.withType<DokkaMultiModuleTask>() {

openai-client/src/commonTest/kotlin/com/aallam/openai/client/TestAudio.kt

@@ -4,24 +4,18 @@ import com.aallam.openai.api.audio.transcriptionRequest
 import com.aallam.openai.api.audio.translationRequest
 import com.aallam.openai.api.file.FileSource
 import com.aallam.openai.api.model.ModelId
-import com.aallam.openai.client.internal.asSource
-import io.ktor.client.*
-import io.ktor.client.call.*
-import io.ktor.client.request.*
+import com.aallam.openai.client.internal.TestFileSystem
+import com.aallam.openai.client.internal.testFilePath
 import kotlin.test.Test
 import kotlin.test.assertEquals
 import kotlin.test.assertTrue
 
 class TestAudio : TestOpenAI() {
 
-    private val httpClient = HttpClient()
-
     @Test
     fun transcription() = test {
-        val speedTalkingUrl = "https://github.com/aallam/sample-data/raw/main/openai/audio/micro-machines.wav"
-        val audioBytes: ByteArray = httpClient.get(speedTalkingUrl).body()
         val request = transcriptionRequest {
-            audio = FileSource(name = "micro-machines.wav", source = audioBytes.asSource())
+            audio = FileSource(path = testFilePath("audio/micro-machines.wav"), fileSystem = TestFileSystem)
             model = ModelId("whisper-1")
         }
 
@@ -37,10 +31,8 @@ class TestAudio : TestOpenAI() {
 
     @Test
     fun transcriptionText() = test {
-        val speedTalkingUrl = "https://github.com/aallam/sample-data/raw/main/openai/audio/micro-machines.wav"
-        val audioBytes: ByteArray = httpClient.get(speedTalkingUrl).body()
         val request = transcriptionRequest {
-            audio = FileSource(name = "micro-machines.wav", source = audioBytes.asSource())
+            audio = FileSource(path = testFilePath("audio/micro-machines.wav"), fileSystem = TestFileSystem)
             model = ModelId("whisper-1")
             responseFormat = "text"
         }
@@ -50,10 +42,8 @@ class TestAudio : TestOpenAI() {
 
     @Test
     fun transcriptionJsonVerbose() = test {
-        val speedTalkingUrl = "https://github.com/aallam/sample-data/raw/main/openai/audio/micro-machines.wav"
-        val audioBytes: ByteArray = httpClient.get(speedTalkingUrl).body()
         val request = transcriptionRequest {
-            audio = FileSource(name = "micro-machines.wav", source = audioBytes.asSource())
+            audio = FileSource(path = testFilePath("audio/micro-machines.wav"), fileSystem = TestFileSystem)
             model = ModelId("whisper-1")
             responseFormat = "verbose_json"
         }
@@ -66,10 +56,8 @@ class TestAudio : TestOpenAI() {
 
     @Test
     fun translation() = test {
-        val multilingualUrl = "https://github.com/aallam/sample-data/raw/main/openai/audio/multilingual.wav"
-        val audioBytes: ByteArray = httpClient.get(multilingualUrl).body()
         val request = translationRequest {
-            audio = FileSource(name = "multilingual.wav", source = audioBytes.asSource())
+            audio = FileSource(path = testFilePath("audio/multilingual.wav"), fileSystem = TestFileSystem)
             model = ModelId("whisper-1")
         }
         val translation = openAI.translation(request)
@@ -84,10 +72,8 @@ class TestAudio : TestOpenAI() {
 
     @Test
     fun translationText() = test {
-        val multilingualUrl = "https://github.com/aallam/sample-data/raw/main/openai/audio/multilingual.wav"
-        val audioBytes: ByteArray = httpClient.get(multilingualUrl).body()
         val request = translationRequest {
-            audio = FileSource(name = "multilingual.wav", source = audioBytes.asSource())
+            audio = FileSource(path = testFilePath("audio/multilingual.wav"), fileSystem = TestFileSystem)
             model = ModelId("whisper-1")
             responseFormat = "text"
         }
@@ -97,10 +83,8 @@ class TestAudio : TestOpenAI() {
 
     @Test
     fun translationJsonVerbose() = test {
-        val multilingualUrl = "https://github.com/aallam/sample-data/raw/main/openai/audio/multilingual.wav"
-        val audioBytes: ByteArray = httpClient.get(multilingualUrl).body()
         val request = translationRequest {
-            audio = FileSource(name = "multilingual.wav", source = audioBytes.asSource())
+            audio = FileSource(path = testFilePath("audio/multilingual.wav"), fileSystem = TestFileSystem)
             model = ModelId("whisper-1")
             responseFormat = "verbose_json"
         }

openai-client/src/commonTest/kotlin/com/aallam/openai/client/TestImages.kt

@@ -5,18 +5,13 @@ import com.aallam.openai.api.image.ImageSize
 import com.aallam.openai.api.image.imageCreation
 import com.aallam.openai.api.image.imageEdit
 import com.aallam.openai.api.image.imageVariation
-import com.aallam.openai.client.internal.asSource
-import io.ktor.client.*
-import io.ktor.client.call.*
-import io.ktor.client.request.*
+import com.aallam.openai.client.internal.TestFileSystem
+import com.aallam.openai.client.internal.testFilePath
 import kotlin.test.Test
 import kotlin.test.assertTrue
-import kotlin.time.Duration.Companion.minutes
 
 class TestImages : TestOpenAI() {
 
-    private val httpClient = HttpClient()
-
     @Test
     fun imageCreationURL() = test {
         val request = imageCreation {
@@ -42,11 +37,9 @@ class TestImages : TestOpenAI() {
 
     @Test
     fun imageEditURL() = test {
-        val imageBytes: ByteArray = httpClient.get(PoolImage).body()
-        val maskBytes: ByteArray = httpClient.get(PoolMaskImage).body()
         val request = imageEdit {
-            image = FileSource(name = "pool.png", source = imageBytes.asSource())
-            mask = FileSource(name = "poolmask.png", source = maskBytes.asSource())
+            image = FileSource(path = testFilePath("image/pool.png"), fileSystem = TestFileSystem)
+            mask = FileSource(path = testFilePath("image/poolmask.png"), fileSystem = TestFileSystem)
             prompt = "a sunlit indoor lounge area with a pool containing a flamingo"
             n = 1
             size = ImageSize.is1024x1024
@@ -57,11 +50,9 @@ class TestImages : TestOpenAI() {
 
     @Test
     fun imageEditJSON() = test {
-        val imageBytes: ByteArray = httpClient.get(PoolImage).body()
-        val maskBytes: ByteArray = httpClient.get(PoolMaskImage).body()
         val request = imageEdit {
-            image = FileSource(name = "pool.png", source = imageBytes.asSource())
-            mask = FileSource(name = "poolmask.png", source = maskBytes.asSource())
+            image = FileSource(path = testFilePath("image/pool.png"), fileSystem = TestFileSystem)
+            mask = FileSource(path = testFilePath("image/poolmask.png"), fileSystem = TestFileSystem)
             prompt = "a sunlit indoor lounge area with a pool containing a flamingo"
             n = 1
             size = ImageSize.is1024x1024
@@ -72,9 +63,8 @@ class TestImages : TestOpenAI() {
 
     @Test
     fun imageVariationURL() = test {
-        val imageBytes: ByteArray = httpClient.get(PetsImage).body()
         val request = imageVariation {
-            image = FileSource("pets.png", imageBytes.asSource())
+            image = FileSource(path = testFilePath("image/pets.png"), fileSystem = TestFileSystem)
             n = 1
             size = ImageSize.is1024x1024
         }
@@ -84,21 +74,12 @@ class TestImages : TestOpenAI() {
 
     @Test
     fun imageVariationJSON() = test {
-        val imageBytes: ByteArray = httpClient.get(PetsImage).body()
         val request = imageVariation {
-            image = FileSource("pets.png", imageBytes.asSource())
+            image = FileSource(path = testFilePath("image/pets.png"), fileSystem = TestFileSystem)
             n = 1
             size = ImageSize.is1024x1024
         }
         val response = openAI.imageJSON(request)
         assertTrue { response.isNotEmpty() }
     }
-
-
-    companion object {
-        private val Path = "https://raw.githubusercontent.com/aallam/sample-data/main/openai/image"
-        private val PoolImage = "$Path/pool.png"
-        private val PoolMaskImage = "$Path/poolmask.png"
-        private val PetsImage = "$Path/pets.png"
-    }
 }

openai-client/src/commonTest/kotlin/com/aallam/openai/client/TestOpenAI.kt

@@ -8,6 +8,8 @@ import com.aallam.openai.client.internal.env
 import com.aallam.openai.client.internal.http.HttpTransport
 import kotlinx.coroutines.test.TestScope
 import kotlinx.coroutines.test.runTest
+import okio.Path.Companion.toPath
+import okio.Source
 import kotlin.time.Duration.Companion.minutes
 
 internal val token: String

openai-client/src/commonTest/kotlin/com/aallam/openai/client/internal/Resource.kt

@@ -0,0 +1,21 @@
+package com.aallam.openai.client.internal
+
+import okio.FileSystem
+import okio.Path
+import okio.Path.Companion.toPath
+
+/**
+ * File system to access test files.
+ */
+internal expect val TestFileSystem: FileSystem
+
+/**
+ * Get [Path] of a given [fileName] test file.
+ */
+fun testFilePath(fileName: String): Path = libRoot / "openai-client/src/commonTest/resources" / fileName
+
+/**
+ * Get the library lib root.
+ */
+private val libRoot
+    get() = env("LIB_ROOT")?.toPath() ?: error("Can't find `LIB_ROOT` environment variable")

openai-client/src/commonTest/kotlin/com/aallam/openai/client/internal/Source.kt

@@ -8,10 +8,3 @@ internal fun String.asSource(): Source {
     buffer.writeUtf8(this)
     return buffer
 }
-
-
-internal fun ByteArray.asSource(): Source {
-    val buffer = Buffer()
-    buffer.write(this)
-    return buffer
-}

openai-client/src/commonTest/resources/audio/micro-machines.wav

@@ -1,5 +1,6 @@
 package com.aallam.openai.client.internal
 
 internal actual fun env(name: String): String? {
-    return js("process.env[name]").unsafeCast<String?>()
+    //return js("process.env[name]").unsafeCast<String?>()
+    return js("globalThis.process.env[name]") as String?
 }

openai-client/src/commonTest/resources/audio/multilingual.wav

@@ -0,0 +1,8 @@
+package com.aallam.openai.client.internal
+
+import okio.FileSystem
+import okio.NodeJsFileSystem
+import okio.Path.Companion.toPath
+import okio.Source
+
+internal actual val TestFileSystem: FileSystem = NodeJsFileSystem

openai-client/src/commonTest/resources/image/pets.png

@@ -0,0 +1,5 @@
+package com.aallam.openai.client.internal
+
+import okio.FileSystem
+
+internal actual val TestFileSystem: FileSystem = FileSystem.SYSTEM

openai-client/src/commonTest/resources/image/pool.png

@@ -0,0 +1,5 @@
+package com.aallam.openai.client.internal
+
+import okio.FileSystem
+
+internal actual val TestFileSystem: FileSystem = FileSystem.SYSTEM