Skip to content

Commit 3d17823

Browse files
authored
feat: enable TLS between TrustyAI and MariaDB (opendatahub-io#190)
Enable and enforce TLS for MariaDB connections Add fixture to copy mariadb-ca certificate to trustyai-ca-cert secret.
1 parent b3bb1c3 commit 3d17823

1 file changed

Lines changed: 18 additions & 1 deletion

File tree

tests/model_explainability/trustyai_service/conftest.py

Lines changed: 18 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -66,6 +66,7 @@ def trustyai_service_with_db_storage(
6666
cluster_monitoring_config: ConfigMap,
6767
user_workload_monitoring_config: ConfigMap,
6868
mariadb: MariaDB,
69+
trustyai_db_ca_secret: None,
6970
) -> Generator[TrustyAIService, Any, Any]:
7071
with TrustyAIService(
7172
client=admin_client,
@@ -226,12 +227,28 @@ def mariadb(
226227
mariadb_dict["spec"]["replicas"] = 1
227228
mariadb_dict["spec"]["galera"]["enabled"] = False
228229
mariadb_dict["spec"]["metrics"]["enabled"] = False
230+
mariadb_dict["spec"]["tls"] = {"enabled": True, "required": True}
229231

230232
password_secret_key_ref = {"generate": False, "key": "databasePassword", "name": DB_CREDENTIALS_SECRET_NAME}
231233

232234
mariadb_dict["spec"]["rootPasswordSecretKeyRef"] = password_secret_key_ref
233235
mariadb_dict["spec"]["passwordSecretKeyRef"] = password_secret_key_ref
234-
235236
with MariaDB(kind_dict=mariadb_dict) as mariadb:
236237
wait_for_mariadb_pods(client=admin_client, mariadb=mariadb)
237238
yield mariadb
239+
240+
241+
@pytest.fixture(scope="class")
242+
def trustyai_db_ca_secret(
243+
admin_client: DynamicClient, model_namespace: Namespace, mariadb: MariaDB
244+
) -> Generator[None, Any, None]:
245+
mariadb_ca_secret = Secret(
246+
client=admin_client, name=f"{mariadb.name}-ca", namespace=model_namespace.name, ensure_exists=True
247+
)
248+
with Secret(
249+
client=admin_client,
250+
name=f"{TRUSTYAI_SERVICE_NAME}-db-ca",
251+
namespace=model_namespace.name,
252+
data_dict={"ca.crt": mariadb_ca_secret.instance.data["ca.crt"]},
253+
):
254+
yield

0 commit comments

Comments
 (0)