lib/pact/messages.rb in pact-messages-0.1.0 vs lib/pact/messages.rb in pact-messages-0.1.1

@@ -1,6 +1,8 @@
 require 'pact'
+require 'pact/messages/consumer'
+require 'pact/messages/message_finder'
 
 module Pact
   module Messages
     module_function
 
@@ -11,16 +13,17 @@
     def build_mock_service(mock_service_name, &block)
       Pact::Messages::Consumer::MockServiceFactory.build(mock_service_name, &block)
     end
 
     def get_message_contract(provider_name, consumer_name, provider_state = nil)
-      Pact::Messages::ContractRepository.get_message_contract(provider_name, consumer_name, provider_state)
+      Pact::Messages::MessageFinder.get_message_contract(provider_name, consumer_name, provider_state)
     end
 
     def get_message_sample(provider_name, consumer_name, provider_state = nil)
-      Pact::Messages::ContractRepository.get_message_sample(provider_name, consumer_name, provider_state)
+      Pact::Messages::MessageFinder.get_message_sample(provider_name, consumer_name, provider_state)
     end
+
+    def pact_broker_url=(url)
+      Pact::Messages::MessageFinder.pact_broker_url = url
+    end
   end
 end
-
-require 'pact/messages/consumer'
-require 'pact/messages/contract_repository'