Surface replay? on workflow contexts

lib/temporal/testing/local_workflow_context.rb

@@ -24,6 +24,11 @@ def logger
         Temporal.logger
       end
 
+      def replay?
+        # We never replay in a test context
+        false
+      end
+
       def headers
         metadata.headers
       end

lib/temporal/workflow/context.rb

@@ -25,10 +25,14 @@ def initialize(state_manager, dispatcher, metadata)
 
       def logger
         @logger ||= ReplayAwareLogger.new(Temporal.logger)
-        @logger.replay = state_manager.replay?
+        @logger.replay = replay?
         @logger
       end
 
+      def replay?
+        @state_manager.replay?
+      end
+
       def headers
         metadata.headers
       end

spec/unit/lib/temporal/testing/local_workflow_context_spec.rb

@@ -98,4 +98,10 @@ def execute
       expect(result).to eq('ok')
     end
   end
+
+  describe '#replay?' do
+    it 'always false' do
+      expect(workflow_context.replay?).to be false
+    end
+  end
 end

spec/unit/lib/temporal/workflow/context_spec.rb

@@ -0,0 +1,20 @@
+require 'temporal/workflow/context'
+
+describe Temporal::Workflow::Context do
+  let(:state_manager) { instance_double('Temporal::Workflow::StateManager') }
+  let(:workflow_context) do
+    Temporal::Workflow::Context.new(
+      state_manager,
+      nil,
+      nil
+    )
+  end
+  describe '#replay' do
+    it 'gets value from state_manager' do
+      allow(state_manager).to receive(:replay?).and_return true
+
+      expect(workflow_context.replay?).to be true
+      expect(state_manager).to have_received(:replay?)
+    end
+  end
+end