fix last test

PostHog · Sep 5, 2024 · ccae49f · ccae49f
1 parent 1c1560c
commit ccae49f
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/posthog/hogql_queries/insights/funnels/test/test_funnel.py b/posthog/hogql_queries/insights/funnels/test/test_funnel.py
@@ -4,7 +4,6 @@
 
 from django.test import override_settings
 from freezegun import freeze_time
-from rest_framework.exceptions import ValidationError
 
 from posthog.api.instance_settings import get_instance_setting
 from posthog.clickhouse.client.execute import sync_execute
@@ -1901,7 +1900,7 @@ def test_funnel_exclusions_invalid_params(self):
             }
 
             query = cast(FunnelsQuery, filter_to_query(filters))
-            self.assertRaises(ValidationError, lambda: FunnelsQueryRunner(query=query, team=self.team).calculate())
+            self.assertRaises(ExposedHogQLError, lambda: FunnelsQueryRunner(query=query, team=self.team).calculate())
 
             filters = {
                 **filters,
@@ -1915,7 +1914,7 @@ def test_funnel_exclusions_invalid_params(self):
                 ],
             }
             query = cast(FunnelsQuery, filter_to_query(filters))
-            self.assertRaises(ValidationError, lambda: FunnelsQueryRunner(query=query, team=self.team).calculate())
+            self.assertRaises(ExposedHogQLError, lambda: FunnelsQueryRunner(query=query, team=self.team).calculate())
 
             filters = {
                 **filters,
@@ -1929,7 +1928,7 @@ def test_funnel_exclusions_invalid_params(self):
                 ],
             }
             query = cast(FunnelsQuery, filter_to_query(filters))
-            self.assertRaises(ValidationError, lambda: FunnelsQueryRunner(query=query, team=self.team).calculate())
+            self.assertRaises(ExposedHogQLError, lambda: FunnelsQueryRunner(query=query, team=self.team).calculate())
 
             filters = {
                 **filters,