mitosheet: fix snowflake pytests for new schema

mito-ds · Aug 12, 2024 · 81175dc · 81175dc
1 parent 9296342
commit 81175dc
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/mitosheet/mitosheet/tests/api/test_get_available_snowflake_options_and_defaults.py b/mitosheet/mitosheet/tests/api/test_get_available_snowflake_options_and_defaults.py
@@ -49,7 +49,7 @@ def test_integration_success():
         'type': 'success',    
         'config_options': {
                 'roles': ['NO_PYTEST_TABLE_ACCESS', 'READONLY'],
-                'warehouses': ['COMPUTE_WH'],    
+                'warehouses': ['COMPUTE_WH', 'SYSTEM$STREAMLIT_NOTEBOOK_WH'],    
                 'databases': ['PYTESTDATABASE', 'SNOWFLAKE', 'SNOWFLAKE_SAMPLE_DATA'],    
                 'schemas': ['INFORMATION_SCHEMA', 'PYTESTSCHEMA'],
                 'tables_and_views': ['COLUMNHEADER_TEST', 'NOROWS', 'SIMPLE_PYTEST_TABLE', 'TYPETEST', 'TYPETEST_SIMPLE', 'YOUR_TABLE_NAME', 'SIMPLE_PYTEST_TABLE_VIEW'],
@@ -114,15 +114,15 @@ def test_switch_roles_updates_defaults():
         'type': 'success',    
         'config_options': {
                 'roles': ['NO_PYTEST_TABLE_ACCESS', 'READONLY'],
-                'warehouses': [],    
+                'warehouses': ['SYSTEM$STREAMLIT_NOTEBOOK_WH'],    
                 'databases': ['SNOWFLAKE', 'SNOWFLAKE_SAMPLE_DATA'],    
                 'schemas': ['ALERT', 'CORE', 'CORTEX', 'IMAGES', 'INFORMATION_SCHEMA', 'ML', 'NOTIFICATION'],
                 'tables_and_views': [],
                 'columns': []
         },
         'default_values': {
             'role': 'NO_PYTEST_TABLE_ACCESS', 
-            'warehouse': None,
+            'warehouse': 'SYSTEM$STREAMLIT_NOTEBOOK_WH',
             'database': 'SNOWFLAKE',
             'schema': 'ALERT',
             'table_or_view': None,