Skip to content

Commit a915c31

Browse files
Improved cleanup logic
1 parent c8d19e8 commit a915c31

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

model_armor/snippets/snippets_test.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -387,6 +387,7 @@ def floor_settings_project_id(project_id: str) -> Generator[str, None, None]:
387387
)
388388
except GoogleAPIError:
389389
print("Floor settings not set or not authorized to set floor settings")
390+
pytest.fail("Failed to cleanup floor settings")
390391

391392

392393
@pytest.fixture()
@@ -415,6 +416,7 @@ def floor_setting_organization_id(
415416
print(
416417
"Floor settings not set or not authorized to set floor settings for organization"
417418
)
419+
pytest.fail("Failed to cleanup floor settings")
418420

419421

420422
@pytest.fixture()
@@ -441,6 +443,7 @@ def floor_setting_folder_id(folder_id: str) -> Generator[str, None, None]:
441443
print(
442444
"Floor settings not set or not authorized to set floor settings for folder"
443445
)
446+
pytest.fail("Failed to cleanup floor settings")
444447

445448

446449
def test_create_template(

0 commit comments

Comments
 (0)