fix(e2e): remove remaining hardcoded DEMOBUILD refs in test harnesses

Follow-up to the rotating-demo-database change: the data-grading fallback,
geo-scope check, and settings_test row check still pointed at DEMOBUILD.
All now derive the database from the model TML (via check_snowflake_schema's
new database field) or get_demo_database().

Co-Authored-By: Claude Fable 5 <noreply@anthropic.com>

tests/e2e_quality.py

@@ -1074,7 +1074,8 @@ def run_ai_grading(run_context: dict, company: str, vertical: str, line: str, fu
             db = db or related_db
             schema = schema or related_schema
         if (not db or not schema) and schema_override:
-            db, schema = "DEMOBUILD", schema_override
+            from snowflake_auth import get_demo_database
+            db, schema = get_demo_database(), schema_override
             print(f"    ℹ️  Using schema resolved from model: {schema_override}")
         sample = get_snowflake_sample(db, schema) if db and schema else "Could not determine db/schema"
         print("    🤖 Grading data quality...")
@@ -1213,16 +1214,17 @@ def verify_group1_settings(result: dict) -> dict:
             conn   = get_snowflake_connection()
             cursor = conn.cursor()
             schema = sf["schema"]
+            database = sf["database"]
             # Look for a column named COUNTRY, REGION, or STATE
-            cursor.execute(f'SHOW TABLES IN SCHEMA DEMOBUILD."{schema}"')
+            cursor.execute(f'SHOW TABLES IN SCHEMA "{database}"."{schema}"')
             tables = [row[1] for row in cursor.fetchall()]
             foreign_found = False
             checked = False
             for tname in tables:
-                cursor.execute(f'SHOW COLUMNS IN TABLE DEMOBUILD."{schema}"."{tname}"')
+                cursor.execute(f'SHOW COLUMNS IN TABLE "{database}"."{schema}"."{tname}"')
                 cols = [row[2].upper() for row in cursor.fetchall()]
                 if "COUNTRY" in cols:
-                    cursor.execute(f'SELECT DISTINCT "COUNTRY" FROM DEMOBUILD."{schema}"."{tname}" LIMIT 20')
+                    cursor.execute(f'SELECT DISTINCT "COUNTRY" FROM "{database}"."{schema}"."{tname}" LIMIT 20')
                     countries = [row[0] for row in cursor.fetchall() if row[0]]
                     non_us = [c for c in countries if c not in ("USA", "US", "United States", "United States of America")]
                     foreign_found = len(non_us) > 0
@@ -1634,7 +1636,8 @@ def check_snowflake_schema(company: str, start_time: float, schema_override: str
 
         cursor.close()
         conn.close()
-        return {"found": True, "schema": schema, "tables": table_info, "total_rows": total_rows}
+        return {"found": True, "database": database, "schema": schema,
+                "tables": table_info, "total_rows": total_rows}
 
     except Exception as e:
         return {"found": False, "error": str(e)}

tests/settings_test.py

@@ -88,14 +88,15 @@ def write_setting(key: str, value: str):
 def check_snowflake_rows(schema_name: str) -> dict:
     """Return {table: row_count} for every table in the schema."""
     try:
-        from snowflake_auth import get_snowflake_connection
+        from snowflake_auth import get_snowflake_connection, get_demo_database
         conn   = get_snowflake_connection()
         cursor = conn.cursor()
-        cursor.execute(f'SHOW TABLES IN SCHEMA DEMOBUILD."{schema_name}"')
+        database = get_demo_database()  # demos live in the rotating <base>_<YYYY_MM> DB
+        cursor.execute(f'SHOW TABLES IN SCHEMA "{database}"."{schema_name}"')
         tables = [row[1] for row in cursor.fetchall()]
         counts = {}
         for t in tables:
-            cursor.execute(f'SELECT COUNT(*) FROM DEMOBUILD."{schema_name}"."{t}"')
+            cursor.execute(f'SELECT COUNT(*) FROM "{database}"."{schema_name}"."{t}"')
             counts[t] = cursor.fetchone()[0]
         cursor.close()
         conn.close()