Add test for snapshot loading

Author: crericha

tests/dbt/test_model.py

@@ -733,6 +733,18 @@ def test_load_microbatch_with_ref_no_filter(
     )
 
 
+@pytest.mark.slow
+def test_load_multiple_snapshots_defined_in_same_file(sushi_test_dbt_context: Context) -> None:
+    context = sushi_test_dbt_context
+    assert context.get_model("snapshots.items_snapshot")
+    assert context.get_model("snapshots.items_check_snapshot")
+
+    # Make sure cache works too
+    context.load()
+    assert context.get_model("snapshots.items_snapshot")
+    assert context.get_model("snapshots.items_check_snapshot")
+
+
 @pytest.mark.slow
 def test_dbt_jinja_macro_undefined_variable_error(create_empty_project):
     project_dir, model_dir = create_empty_project()