feat: move testing side effects to the context manager

testing/src/scenario/_ops_main_mock.py

@@ -299,7 +299,10 @@ def _emit(self, event: ops.EventBase):
         ):
             # Dump/undump the event to ensure any custom attributes are (re)set by restore().
             event.restore(event.snapshot())
-            self._context.emitted_events.append(event)
+            # In the future, we will only store this on the manager.
+            self._context._emitted_events.append(event)
+            if self._context._manager is not None:
+                self._context._manager.emitted_events.append(event)
 
         return super()._emit(event)
 
@@ -320,6 +323,9 @@ def _reemit(self, single_event_path: str | None = None):
 
             # Dump/undump the event to ensure any custom attributes are (re)set by restore().
             event.restore(event.snapshot())
-            self._context.emitted_events.append(event)
+            # In the future, we will only store this on the manager.
+            self._context._emitted_events.append(event)
+            if self._context._manager is not None:
+                self._context._manager.emitted_events.append(event)
 
         return super()._reemit(single_event_path)

testing/src/scenario/_runtime.py

@@ -270,7 +270,7 @@ def exec(
         self,
         state: State,
         event: _Event,
-        context: Context,
+        context: Context[CharmType],
     ):
         """Runs an event with this state as initial state on a charm.
 
@@ -341,7 +341,10 @@ def exec(
             finally:
                 if ops:
                     ops.destroy()
-                    context.trace_data.extend(ops.trace_data)
+                    # In the future, we will only store this on the manager.
+                    context._trace_data.extend(ops.trace_data)
+                    if context._manager is not None:
+                        context._manager.trace_data.extend(ops.trace_data)
                 for key in tuple(os.environ):
                     if key not in previous_env:
                         del os.environ[key]