diff --git a/tests/integration/test_charms.py b/tests/integration/test_charms.py index 7e1b1f20..e59da3a2 100644 --- a/tests/integration/test_charms.py +++ b/tests/integration/test_charms.py @@ -20,6 +20,14 @@ UI_APP_NAME = UI_METADATA["name"] DB_APP_NAME = DB_METADATA["name"] +MYSQL_K8S = "mysql-k8s" +MYSQL_K8S_CHANNEL = "8.0/edge" +MYSQL_K8S_TRUST = True + +KUBEFLOW_PROFILES = "kubeflow-profiles" +KUBEFLOW_PROFILES_CHANNEL = "latest/edge" +KUBEFLOW_PROFILES_TRUST = True + logger = logging.getLogger(__name__) @@ -47,9 +55,11 @@ async def test_deploy_katib_charms(ops_test: OpsTest): # Deploy katib-db await ops_test.model.deploy( - "charmed-osm-mariadb-k8s", + MYSQL_K8S, + channel=MYSQL_K8S_CHANNEL, application_name="katib-db", config={"database": "katib"}, + trust=MYSQL_K8S_TRUST, ) # Relate to katib-db @@ -63,9 +73,9 @@ async def test_deploy_katib_charms(ops_test: OpsTest): # Deploy charms responsible for CRDs creation await ops_test.model.deploy( - entity_url="kubeflow-profiles", - channel="latest/edge", - trust=True, + entity_url=KUBEFLOW_PROFILES, + channel=KUBEFLOW_PROFILES_CHANNEL, + trust=KUBEFLOW_PROFILES_TRUST, ) # Wait for everything to deploy diff --git a/tests/integration/test_katib_experiments.py b/tests/integration/test_katib_experiments.py index 05e5200e..935bed32 100644 --- a/tests/integration/test_katib_experiments.py +++ b/tests/integration/test_katib_experiments.py @@ -27,7 +27,9 @@ kind="Profile", plural="profiles", ) -TRAINING_CHARM = "training-operator" +TRAINING_OPERATOR = "training-operator" +TRAINING_OPERATOR_CHANNEL = "latest/edge" +TRAINING_OPERATOR_TRUST = True @pytest.fixture(scope="module") @@ -42,12 +44,12 @@ def lightkube_client() -> lightkube.Client: async def training_operator(ops_test: OpsTest): """Deploy training-operator charm, and wait until it's active.""" await ops_test.model.deploy( - entity_url=TRAINING_CHARM, - channel="latest/edge", - trust=True, + entity_url=TRAINING_OPERATOR, + channel=TRAINING_OPERATOR_CHANNEL, + trust=TRAINING_OPERATOR_TRUST, ) await ops_test.model.wait_for_idle( - apps=[TRAINING_CHARM], status="active", raise_on_blocked=False, timeout=60 * 5 + apps=[TRAINING_OPERATOR], status="active", raise_on_blocked=False, timeout=60 * 5 )