Use wrapper WorkloadApiClient (#9845)

Author: jdpgrailsdev

airbyte-api/build.gradle

@@ -297,6 +297,8 @@ dependencies {
     compileOnly libs.v3.swagger.annotations
     kapt libs.v3.swagger.annotations
 
+    implementation platform(libs.micronaut.bom)
+    implementation libs.bundles.micronaut
     implementation libs.commons.io
     implementation libs.failsafe.okhttp
     implementation libs.guava
@@ -315,10 +317,8 @@ dependencies {
     testImplementation libs.bundles.junit
     testImplementation libs.assertj.core
     testImplementation libs.junit.pioneer
-
-    implementation platform(libs.micronaut.bom)
-    implementation libs.bundles.micronaut
-
+    testImplementation libs.mockk
+    testImplementation libs.kotlin.test.runner.junit5
 }
 
 sourceSets {

airbyte-api/src/main/kotlin/WorkloadApiClient.kt

@@ -32,12 +32,14 @@ import okhttp3.Response
  */
 @SuppressWarnings("Parameter")
 class WorkloadApiClient {
+  var workloadApi: WorkloadApi
+
   @JvmOverloads
   constructor(
     basePath: String,
     policy: RetryPolicy<Response> = RetryPolicy.ofDefaults(),
     httpClient: OkHttpClient = OkHttpClient(),
   ) {
-    val workloadApi = WorkloadApi()
+    workloadApi = WorkloadApi(basePath = basePath, client = httpClient, policy = policy)
   }
 }

airbyte-api/src/test/kotlin/io/airbyte/api/client/WorkloadApiClientTest.kt

@@ -0,0 +1,28 @@
+/*
+ * Copyright (c) 2023 Airbyte, Inc., all rights reserved.
+ */
+
+package io.airbyte.api.client
+
+import dev.failsafe.RetryPolicy
+import io.mockk.mockk
+import okhttp3.OkHttpClient
+import okhttp3.Response
+import org.junit.jupiter.api.Assertions.assertEquals
+import org.junit.jupiter.api.Assertions.assertNotNull
+import org.junit.jupiter.api.Test
+
+class WorkloadApiClientTest {
+  @Test
+  fun `test that the workload API client creates the underlying Workload API object with the provided configuration`() {
+    val basePath = "base-path"
+    val client: OkHttpClient = mockk()
+    val policy: RetryPolicy<Response> = mockk()
+
+    val workloadApiClient = WorkloadApiClient(basePath, policy, client)
+    assertNotNull(workloadApiClient.workloadApi)
+    assertEquals(client, workloadApiClient.workloadApi.client)
+    assertEquals(policy, workloadApiClient.workloadApi.policy)
+    assertEquals(basePath, workloadApiClient.workloadApi.baseUrl)
+  }
+}

airbyte-commons-worker/src/main/kotlin/io/airbyte/workers/config/WorkloadApiClientFactory.kt

@@ -5,6 +5,7 @@
 package io.airbyte.workers.config
 
 import dev.failsafe.RetryPolicy
+import io.airbyte.api.client.WorkloadApiClient
 import io.airbyte.workload.api.client2.generated.WorkloadApi
 import io.github.oshai.kotlinlogging.KotlinLogging
 import io.micronaut.context.annotation.Factory
@@ -56,6 +57,6 @@ class WorkloadApiClientFactory {
         .withMaxRetries(maxRetries)
         .build()
 
-    return WorkloadApi("$internalApiScheme://$workloadApiBasePath", okHttpClient, retryPolicy)
+    return WorkloadApiClient("$internalApiScheme://$workloadApiBasePath", retryPolicy, okHttpClient).workloadApi
   }
 }