feat(v3): support set up and tear down of multiple provider states

mejuri-inc · Oct 1, 2018 · cbad0be · cbad0be
1 parent 6207520
commit cbad0be
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 6 deletions.
diff --git a/lib/pact/provider/rspec.rb b/lib/pact/provider/rspec.rb
@@ -84,15 +84,15 @@ def describe_interaction interaction, options
             before do | example |
               interaction_context.run_once :before do
                 Pact.configuration.logger.info "Running example '#{Pact::RSpec.full_description(example)}'"
-                set_up_provider_state interaction.provider_state, options[:consumer]
+                set_up_provider_states interaction.provider_states, options[:consumer]
                 replay_interaction interaction, options[:request_customizer]
                 interaction_context.last_response = last_response
               end
             end
 
             after do
               interaction_context.run_once :after do
-                tear_down_provider_state interaction.provider_state, options[:consumer]
+                tear_down_provider_states interaction.provider_states, options[:consumer]
               end
             end
 

diff --git a/lib/pact/provider/test_methods.rb b/lib/pact/provider/test_methods.rb
@@ -35,12 +35,16 @@ def parse_body_from_response rack_response
         end
       end
 
-      def set_up_provider_state provider_state_name, consumer, options = {}
-        Pact.configuration.provider_state_set_up.call(provider_state_name, consumer, options)
+      def set_up_provider_states provider_states, consumer, options = {}
+        provider_states.each do | provider_state |
+          Pact.configuration.provider_state_set_up.call(provider_state.name, consumer, options.merge(params: provider_state.params))
+        end
       end
 
-      def tear_down_provider_state provider_state_name, consumer, options = {}
-        Pact.configuration.provider_state_tear_down.call(provider_state_name, consumer, options)
+      def tear_down_provider_states provider_states, consumer, options = {}
+        provider_states.reverse.each do | provider_state |
+          Pact.configuration.provider_state_tear_down.call(provider_state.name, consumer, options.merge(params: provider_state.params))
+        end
       end
 
       def set_metadata example, key, value