cleaning up

Author: d-led

lib/actor_simulation.ex

@@ -121,12 +121,12 @@ defmodule ActorSimulation do
   def add_actor(simulation, name, opts \\ []) do
     actor_def = Definition.new(name, opts)
 
-    # Inject trace collector if tracing is enabled
+    # Inject trace collector if tracing is enabled and stats tracking
     actor_opts =
       if simulation.trace_enabled do
-        [trace_collector: self()]
+        [trace_collector: self(), stats_enabled: true]
       else
-        []
+        [stats_enabled: true]
       end
 
     {:ok, pid} = Actor.start_link(actor_def, simulation.clock, actor_opts)

lib/actor_simulation/actor.ex

@@ -24,11 +24,13 @@ defmodule ActorSimulation.Actor do
 
   def start_link(definition, clock, opts \\ []) do
     # Use test-local virtual clock injection instead of global
-    # Also support trace collector injection
+    # Also support trace collector injection and stats tracking
     trace_collector = Keyword.get(opts, :trace_collector)
+    stats_enabled = Keyword.get(opts, :stats_enabled, false)
 
     VirtualTimeGenServer.start_link(__MODULE__, {definition, trace_collector},
-      virtual_clock: clock
+      virtual_clock: clock,
+      stats_enabled: stats_enabled
     )
   end