chore: provide setter method for application context

pact-foundation · Dec 2, 2022 · ec43c4b · ec43c4b
1 parent fe63ee1
commit ec43c4b
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 4 deletions.
diff --git a/lib/pact_broker/api.rb b/lib/pact_broker/api.rb
@@ -1,4 +1,5 @@
 require "webmachine/adapters/rack_mapped"
+require "webmachine/application_monkey_patch"
 require "pact_broker/db/models"
 require "pact_broker/api/resources"
 require "pact_broker/api/decorators"
@@ -140,10 +141,7 @@ def self.build_api(application_context = PactBroker::ApplicationContext.default_
       end
     end
 
-    # naughty, but better than setting each route manually
-    pact_api.routes.each do | route |
-      route.instance_variable_get(:@bindings)[:application_context] = application_context
-    end
+    pact_api.application_context = application_context
 
     pact_api.configure do |config|
       config.adapter = :RackMapped

diff --git a/lib/webmachine/application_monkey_patch.rb b/lib/webmachine/application_monkey_patch.rb
@@ -0,0 +1,10 @@
+require "webmachine/application"
+
+class Webmachine::Application
+  def application_context= application_context
+    # naughty, but better than setting each route manually
+    routes.each do | route |
+      route.instance_variable_get(:@bindings)[:application_context] = application_context
+    end
+  end
+end