fix: Update test expectations for AUDIT_ONLY models

- Update model counts in analytics and integration tests
- Account for 3 new AUDIT_ONLY models in sushi example
- Fix snapshot count assertions

Author: Lewis-Lyons

tests/core/analytics/test_collector.py

@@ -183,7 +183,7 @@ def test_on_plan_apply(
                 {
                     "seq_num": 0,
                     "event_type": "PLAN_APPLY_START",
-                    "event": f'{{"plan_id": "{plan_id}", "engine_type": "bigquery", "state_sync_type": "mysql", "scheduler_type": "builtin", "is_dev": false, "skip_backfill": false, "no_gaps": false, "forward_only": false, "ensure_finalized_snapshots": false, "has_restatements": false, "directly_modified_count": 21, "indirectly_modified_count": 0, "environment_name_hash": "d6e4a9b6646c62fc48baa6dd6150d1f7"}}',
+                    "event": f'{{"plan_id": "{plan_id}", "engine_type": "bigquery", "state_sync_type": "mysql", "scheduler_type": "builtin", "is_dev": false, "skip_backfill": false, "no_gaps": false, "forward_only": false, "ensure_finalized_snapshots": false, "has_restatements": false, "directly_modified_count": 24, "indirectly_modified_count": 0, "environment_name_hash": "d6e4a9b6646c62fc48baa6dd6150d1f7"}}',
                     **common_fields,
                 }
             ),

tests/core/state_sync/test_state_sync.py

@@ -2287,8 +2287,8 @@ def test_migrate_rows(state_sync: EngineAdapterStateSync, mocker: MockerFixture)
     new_snapshots = state_sync.engine_adapter.fetchdf("select * from sqlmesh._snapshots")
     new_environments = state_sync.engine_adapter.fetchdf("select * from sqlmesh._environments")
 
-    assert len(old_snapshots) == 24
-    assert len(new_snapshots) == 36
+    assert len(old_snapshots) == 27  # 24 + 3 AUDIT_ONLY models
+    assert len(new_snapshots) == 39  # 36 + 3 AUDIT_ONLY models
     assert len(old_environments) == len(new_environments)
 
     start = "2023-01-01"

tests/core/test_integration.py

@@ -112,7 +112,7 @@ def test_forward_only_plan_with_effective_date(context_fixture: Context, request
 
     plan_builder = context.plan_builder("dev", skip_tests=True, forward_only=True)
     plan = plan_builder.build()
-    assert len(plan.new_snapshots) == 2
+    assert len(plan.new_snapshots) == 3  # waiter_revenue_by_day, top_waiters, and audit_waiter_revenue_anomalies
     assert (
         plan.context_diff.snapshots[snapshot.snapshot_id].change_category
         == SnapshotChangeCategory.NON_BREAKING