test_lib/broker_config.rb in message-driver-0.4.0 vs test_lib/broker_config.rb in message-driver-0.5.0

@@ -1,29 +1,45 @@
 class BrokerConfig
   class << self
     def config
       adapter_file = File.expand_path('../../.adapter_under_test', __FILE__)
       @adapter, @version = (ENV['ADAPTER'] || (File.exist?(adapter_file) ? File.read(adapter_file).chomp : '')).split(':')
+      vhost = ENV['VHOST'] || 'message-driver-test'
       case @adapter
       when 'bunny'
         {
           adapter: :bunny,
-          vhost: 'message-driver-test',
+          vhost: vhost,
           continuation_timeout: 10000
         }
       when 'in_memory'
-        {adapter: :in_memory}
+        { adapter: :in_memory }
       when 'stomp'
         {
           adapter: :stomp,
-          vhost: 'message-driver-test',
-          hosts: [{host: 'localhost', login: 'guest', passcode: 'guest'}],
+          vhost: vhost,
+          hosts: [{ host: 'localhost', login: 'guest', passcode: 'guest' }],
           reliable: false,
           max_reconnect_attempts: 1
         }
       else
-        {adapter: :in_memory}
+        { adapter: :in_memory }
       end
+    end
+
+    def provider
+      case current_adapter
+      when :bunny, :stomp
+        :rabbitmq
+      when :in_memory
+        :in_memory
+      else
+        current_adapter
+      end
+    end
+
+    def setup_provider
+      require_relative "provider/#{provider}"
     end
 
     def all_adapters
       %w(in_memory bunny stomp)
     end