From 1b018f783951e2b8197f758b90826c8580d6711f Mon Sep 17 00:00:00 2001 From: Pedro Rodrigues Date: Wed, 23 Jul 2025 00:57:51 +0100 Subject: [PATCH] fix(free-tier): starter workspace tests --- singlestoredb/management/workspace.py | 9 +++++---- singlestoredb/tests/test_management.py | 1 + 2 files changed, 6 insertions(+), 4 deletions(-) diff --git a/singlestoredb/management/workspace.py b/singlestoredb/management/workspace.py index 3b6777bb8..99458872e 100644 --- a/singlestoredb/management/workspace.py +++ b/singlestoredb/management/workspace.py @@ -1866,7 +1866,7 @@ def create_starter_workspace( database_name : str Name of the database for the starter workspace workspace_group : dict[str, str] - Workspace group input (dict with keys: 'cell_id') + Workspace group input (dict with keys: 'cell_id' and 'name'). Returns ------- @@ -1875,15 +1875,16 @@ def create_starter_workspace( if not workspace_group or not isinstance(workspace_group, dict): raise ValueError( 'workspace_group must be a dict with keys: ' - "'cell_id'", + "'cell_id' and 'name'", ) - if set(workspace_group.keys()) != {'cell_id'}: - raise ValueError("workspace_group must contain only 'cell_id'") + if set(workspace_group.keys()) != {'cell_id', 'name'}: + raise ValueError("workspace_group must contain only 'cell_id' and 'name'") payload = { 'name': name, 'databaseName': database_name, 'workspaceGroup': { + 'name': workspace_group['name'], 'cellID': workspace_group['cell_id'], }, } diff --git a/singlestoredb/tests/test_management.py b/singlestoredb/tests/test_management.py index a2dc2d94f..3292c44fa 100755 --- a/singlestoredb/tests/test_management.py +++ b/singlestoredb/tests/test_management.py @@ -399,6 +399,7 @@ def setUpClass(cls): f'starter-ws-test-{name}', database_name=f'starter_db_{name}', workspace_group={ + 'name': f'starter-wg-test-{name}', 'cell_id': random.choice(us_regions).id, }, )