update baseurl for airbyte workers

canonical · Oct 31, 2024 · 0a61b51 · 0a61b51
1 parent ea2b638
commit 0a61b51
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/charm_helpers.py b/src/charm_helpers.py
@@ -143,7 +143,7 @@ def create_env(model_name, app_name, container_name, config, state):
     }
 
     # https://github.com/airbytehq/airbyte/issues/29506#issuecomment-1775148609
-    if container_name == "airbyte-workload-launcher":
+    if container_name in ["airbyte-workload-launcher", "airbyte-workers"]:
         env.update(
             {
                 "INTERNAL_API_HOST": f"http://{app_name}:{INTERNAL_API_PORT}",

diff --git a/tests/unit/test_charm.py b/tests/unit/test_charm.py
@@ -417,7 +417,7 @@ def create_plan(container_name, storage_type):
         },
     }
 
-    if container_name == "airbyte-workload-launcher":
+    if container_name in ["airbyte-workload-launcher", "airbyte-workers"]:
         want_plan["services"][container_name]["environment"].update(
             {"INTERNAL_API_HOST": "http://airbyte-k8s:8001", "WORKLOAD_API_HOST": "http://airbyte-k8s:8007"}
         )