Move EventCatcher to tests.utils for use in unit and functional t…

…ests

tests/functional/configs/test_warn_error_options.py

@@ -6,7 +6,7 @@
 from dbt.events.types import DeprecatedModel
 from dbt.tests.util import update_config_file
 from dbt_common.events.base_types import EventLevel
-from tests.functional.utils import EventCatcher
+from tests.utils import EventCatcher
 
 ModelsDictSpec = Dict[str, Union[str, "ModelsDictSpec"]]
 

tests/functional/manifest_validations/test_check_for_spaces_in_model_names.py

@@ -10,7 +10,7 @@
 )
 from dbt.tests.util import update_config_file
 from dbt_common.events.base_types import EventLevel
-from tests.functional.utils import EventCatcher
+from tests.utils import EventCatcher
 
 
 class TestSpacesInModelNamesHappyPath:

tests/unit/events/test_logging.py

@@ -8,7 +8,7 @@
 from dbt_common.events.base_types import BaseEvent
 from dbt_common.events.event_manager_client import get_event_manager
 from dbt_common.events.logger import LoggerConfig
-from tests.functional.utils import EventCatcher
+from tests.utils import EventCatcher
 
 
 class TestSetupEventLogger:

tests/utils.py

@@ -0,0 +1,17 @@
+from dataclasses import dataclass, field
+from typing import List
+
+from dbt_common.events.base_types import BaseEvent, EventMsg
+
+
+@dataclass
+class EventCatcher:
+    event_to_catch: BaseEvent
+    caught_events: List[EventMsg] = field(default_factory=list)
+
+    def catch(self, event: EventMsg):
+        if event.info.name == self.event_to_catch.__name__:
+            self.caught_events.append(event)
+
+    def flush(self) -> None:
+        self.caught_events = []