Merge pull request #436 from atlanhq/fix-integration-tests

Updated integration test to align to new tenant setup

Author: Aryamanz29

tests/integration/purpose_test.py

@@ -28,10 +28,11 @@
 
 MODULE_NAME = TestId.make_unique("Purpose")
 PERSONA_NAME = "Data Assets"
-DB_NAME = "RAW"
-TABLE_NAME = "PACKAGETYPES"
-COLUMN_NAME = "PACKAGETYPENAME"
-SCHEMA_NAME = "WIDEWORLDIMPORTERS_WAREHOUSE"
+DB_NAME = "ANALYTICS"
+TABLE_NAME = "STG_STATE_PROVINCES"
+COLUMN_NAME = "LATEST_RECORDED_POPULATION"
+SCHEMA_NAME = "WIDE_WORLD_IMPORTERS"
+REDACTED_NUMBER = "0000000"
 API_TOKEN_NAME = MODULE_NAME
 
 
@@ -178,7 +179,7 @@ def test_find_purpose_by_name(
         count = 0
         # TODO: replace with exponential back-off and jitter
         while not result and count < 10:
-            time.sleep(2)
+            time.sleep(5)
             result = client.asset.find_purposes_by_name(MODULE_NAME)
             count += 1
     assert result
@@ -269,11 +270,9 @@ def test_run_query_without_policy(client: AtlanClient, assign_tag_to_asset, quer
     assert response.rows
     assert len(response.rows) > 1
     row = response.rows[0]
-    assert row
-    assert len(row) == 7
-    assert row[2]
+    assert row and len(row) == 10
     # Ensure it is NOT redacted
-    assert row[2].startswith("Xx") is False
+    assert row[6] and row[6] != REDACTED_NUMBER
 
 
 def test_token_permissions(client: AtlanClient, token):
@@ -317,8 +316,6 @@ def test_run_query_with_policy(assign_tag_to_asset, token, query):
         assert response.rows
         assert len(response.rows) > 1
         row = response.rows[0]
-        assert row
-        assert len(row) == 7
-        assert row[2]
+        assert row and len(row) == 10
         # Ensure it IS redacted
-        assert row[2].startswith("Xx")
+        assert row[6] and row[6] == REDACTED_NUMBER

tests/integration/test_task_client.py

@@ -15,16 +15,16 @@
 MODULE_NAME = TestId.make_unique("TaskClient")
 TAG_NAME = MODULE_NAME
 
-DB_NAME = "RAW"
+DB_NAME = "WIDE_WORLD_IMPORTERS"
 TABLE_NAME = "PACKAGETYPES"
 COLUMN_NAME = "PACKAGETYPENAME"
-SCHEMA_NAME = "WIDEWORLDIMPORTERS_WAREHOUSE"
+SCHEMA_NAME = "BRONZE_WAREHOUSE"
 
 
 @pytest.fixture(scope="module")
 def snowflake_conn(client: AtlanClient):
     return client.asset.find_connections_by_name(
-        "development", AtlanConnectorType.SNOWFLAKE
+        "production", AtlanConnectorType.SNOWFLAKE
     )[0]
 
 

tests/integration/test_workflow_client.py

@@ -280,16 +280,6 @@ def test_get_all_credentials(client: AtlanClient):
     ), "Expected at least one record but found none"
 
 
-def test_get_all_credentials_with_limit_and_offset(client: AtlanClient):
-    limit = 5
-    offset = 2
-    credentials = client.credentials.get_all(limit=limit, offset=offset)
-    assert credentials.records is not None, "Expected records but found None"
-    assert (
-        len(credentials.records or []) <= limit
-    ), f"Expected at most {limit} records, got {len(credentials.records or [])}"
-
-
 def test_get_all_credentials_with_filter_limit_offset(client: AtlanClient):
     filter_criteria = {"connectorType": "jdbc"}
     limit = 1