test(bigquery): add bq_memory_limited defaults to existing mocks

The new bq_memory_limited warning path in get_df_payload iterates all
cache mocks in existing tests. Those pre-existing tests don't set
bq_memory_limited on their MagicMock/MockCache instances, causing
TypeErrors on f-string format or AttributeErrors. Set the field to
False/0 in the three impacted tests so they exercise the no-warning
code path.
This commit is contained in:
Claude Code
2026-04-18 16:13:26 -07:00
parent 889f23e412
commit 05a9b47f1a

View File

@@ -928,6 +928,7 @@ def test_get_df_payload_validates_before_cache_key_generation():
mock_cache.query = "SELECT * FROM table"
mock_cache.error_message = None
mock_cache.status = "success"
mock_cache.bq_memory_limited = False
mock_cache_manager.get.return_value = mock_cache
# Call get_df_payload
@@ -1303,6 +1304,7 @@ def test_force_cached_normalizes_totals_query_row_limit():
cache.is_cached = True
cache.sql_rowcount = len(df)
cache.cache_dttm = "2024-01-01T00:00:00"
cache.bq_memory_limited = False
return cache
mock_cache_manager.get.side_effect = cache_get
@@ -1359,6 +1361,8 @@ def test_get_df_payload_invalidates_cache_missing_applied_filter_columns():
self.applied_template_filters = []
self.rejected_filter_columns = []
self.annotation_data = {}
self.bq_memory_limited = False
self.bq_memory_limited_row_count = 0
self.set_query_result = MagicMock()
mock_cache = MockCache()