Skip to content

Commit 30ef348

Browse files
committed
feat: enable TLS between TrustyAI and MariaDB
Enable and enforce TLS for MariaDB connections Add fixture to copy mariadb-ca certificate to trustyai-ca-cert secret.
1 parent 346edbc commit 30ef348

File tree

1 file changed

+18
-1
lines changed

1 file changed

+18
-1
lines changed

tests/model_explainability/trustyai_service/conftest.py

Lines changed: 18 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -67,6 +67,7 @@ def trustyai_service_with_db_storage(
6767
cluster_monitoring_config: ConfigMap,
6868
user_workload_monitoring_config: ConfigMap,
6969
mariadb: MariaDB,
70+
configure_trustyai_db_ca_secret: None,
7071
) -> Generator[TrustyAIService, Any, Any]:
7172
with TrustyAIService(
7273
client=admin_client,
@@ -275,12 +276,28 @@ def mariadb(
275276
mariadb_dict["spec"]["replicas"] = 1
276277
mariadb_dict["spec"]["galera"]["enabled"] = False
277278
mariadb_dict["spec"]["metrics"]["enabled"] = False
279+
mariadb_dict["spec"]["tls"] = {"enabled": True, "required": True}
278280

279281
password_secret_key_ref = {"generate": False, "key": "databasePassword", "name": DB_CREDENTIALS_SECRET_NAME}
280282

281283
mariadb_dict["spec"]["rootPasswordSecretKeyRef"] = password_secret_key_ref
282284
mariadb_dict["spec"]["passwordSecretKeyRef"] = password_secret_key_ref
283-
284285
with MariaDB(kind_dict=mariadb_dict) as mariadb:
285286
wait_for_mariadb_pods(client=admin_client, mariadb=mariadb)
286287
yield mariadb
288+
289+
290+
@pytest.fixture(scope="class")
291+
def configure_trustyai_db_ca_secret(
292+
admin_client: DynamicClient, model_namespace: Namespace, mariadb: MariaDB
293+
) -> Generator[None, Any, None]:
294+
mariadb_ca_secret = Secret(
295+
client=admin_client, name=f"{mariadb.name}-ca", namespace=model_namespace.name, ensure_exists=True
296+
)
297+
with Secret(
298+
client=admin_client,
299+
name=f"{TRUSTYAI_SERVICE_NAME}-db-ca",
300+
namespace=model_namespace.name,
301+
data_dict={"ca.crt": mariadb_ca_secret.instance.data["ca.crt"]},
302+
):
303+
yield

0 commit comments

Comments
 (0)