Skip to content

Commit 00cdf24

Browse files
authored
fix(free-tier): starter workspace creation tests (#78)
* fix starter workspace creation tests * remove no manager test for regions
1 parent 7a3a5ae commit 00cdf24

File tree

1 file changed

+4
-17
lines changed

1 file changed

+4
-17
lines changed

singlestoredb/tests/test_management.py

Lines changed: 4 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,6 @@
1414
from singlestoredb.management.job import Status
1515
from singlestoredb.management.job import TargetType
1616
from singlestoredb.management.region import Region
17-
from singlestoredb.management.region import RegionManager
1817
from singlestoredb.management.utils import NamedList
1918

2019

@@ -390,7 +389,9 @@ def password(self):
390389
def setUpClass(cls):
391390
cls.manager = s2.manage_workspaces()
392391

393-
us_regions = [x for x in cls.manager.regions if 'US' in x.name]
392+
shared_tier_regions = [
393+
x for x in cls.manager.shared_tier_regions if 'US' in x.name
394+
]
394395
cls.password = secrets.token_urlsafe(20) + '-x&$'
395396

396397
name = clean_name(secrets.token_urlsafe(20)[:20])
@@ -400,7 +401,7 @@ def setUpClass(cls):
400401
database_name=f'starter_db_{name}',
401402
workspace_group={
402403
'name': f'starter-wg-test-{name}',
403-
'cell_id': random.choice(us_regions).id,
404+
'cell_id': random.choice(shared_tier_regions).id,
404405
},
405406
)
406407

@@ -1564,17 +1565,3 @@ def test_str_repr(self):
15641565

15651566
# Test __repr__
15661567
assert repr(region) == str(region)
1567-
1568-
def test_no_manager(self):
1569-
"""Test behavior when manager is not available."""
1570-
regions = self.manager.list_regions()
1571-
if not regions:
1572-
self.skipTest('No regions available for testing')
1573-
1574-
region = regions[0]
1575-
region._manager = None
1576-
1577-
# Verify from_dict class method
1578-
with self.assertRaises(s2.ManagementError) as cm:
1579-
RegionManager.list_shared_tier_regions(None)
1580-
assert 'No workspace manager' in str(cm.exception)

0 commit comments

Comments
 (0)