fix a bug in patch where Subprocess() was missing in `Macro::Stra…

…tegy`.
trailblazer · Nov 15, 2023 · ac960d5 · ac960d5 · contentfree · Dec 29, 2023
1 parent fef3161
commit ac960d5
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 1 deletion.
diff --git a/lib/trailblazer/macro/strategy.rb b/lib/trailblazer/macro/strategy.rb
@@ -4,7 +4,7 @@ module Macro
     class Strategy # We want to look like a real {Linear::Strategy}.
       class << self
         extend Forwardable
-        def_delegators :block_activity, :step, :pass, :fail # TODO: add all DSL::Helper
+        def_delegators :block_activity, :step, :pass, :fail, :Subprocess # TODO: add all DSL::Helper
       end
 
       # This makes {Wrap} look like {block_activity}.

diff --git a/test/docs/wrap_test.rb b/test/docs/wrap_test.rb
@@ -661,6 +661,19 @@ class MyValidation < Trailblazer::Activity::Railway
 
     #@ the patched version only runs {mock_validation}.
     assert_invoke patched_activity, seq: %{[:mock_validation]}
+
+    # Patch Wrap/Subprocess/step, put step after {validate}
+    patched_activity = Trailblazer::Activity::DSL::Linear::Patch.call(
+      activity,
+      ["Wrap/WrapUnitTest::HandleUnsafeProcess", :validation],
+      -> { step mock_validation }
+    )
+
+    #@ the original activity with Wrap is unchanged.
+    assert_invoke activity, seq: %{[:validate]}
+
+    #@ the patched version only runs {mock_validation}.
+    assert_invoke patched_activity, seq: %{[:validate, :mock_validation]}
   end
 end