Make ucws tests skipped when DATABRICKS_ACCOUNT_ID is present (#448)

edwardfeng-db · web-flow · commit 5fc77b0d7b2d · 2023-11-17T10:41:32.000Z
## Changes
&lt;!-- Summary of your changes that are easy to understand --&gt;
- Added `skip_for_env` function and make all tests with `ucws` skip when
`DATABRICKS_ACCOUNT_ID` is present

## Tests
&lt;!-- 
How is this tested? Please see the checklist below and also describe any
other relevant tests
--&gt;
Verified that the ucws tests got skipped when running `make dev
integration`
- [ ] `make test` run locally
- [ ] `make fmt` applied
- [ ] relevant integration tests applied
diff --git a/tests/integration/conftest.py b/tests/integration/conftest.py
@@ -71,6 +71,8 @@ def w(env_or_skip) -> WorkspaceClient:
 def ucws(env_or_skip) -> WorkspaceClient:
     _load_debug_env_if_runs_from_ide('ucws')
     env_or_skip("CLOUD_ENV")
+    if 'DATABRICKS_ACCOUNT_ID' in os.environ:
+        pytest.skip("Skipping workspace test on account level")
     if 'TEST_METASTORE_ID' not in os.environ:
         pytest.skip("not in Unity Catalog Workspace test env")
     return WorkspaceClient()
