spec/spec_helper.rb in browsermob-proxy-0.0.7 vs spec/spec_helper.rb in browsermob-proxy-0.0.8

@@ -1,18 +1,27 @@
 require 'browsermob/proxy'
 require 'selenium-webdriver'
 require 'browsermob/proxy/webdriver_listener'
+require 'rack'
 
 RestClient.log = STDOUT
 
 module BrowserMob
   module Proxy
     module SpecHelper
+      def self.httpd
+        @httpd ||= HttpServer.new(Rack::File.new(fixture_dir))
+      end
+
+      def self.fixture_dir
+        @fixture_dir ||= File.join(File.expand_path("../", __FILE__), "fixtures")
+      end
+
       def server
         $_bm_server ||= Server.new(
           File.join(home, "bin", "browsermob-proxy"),
-          :port => Selenium::WebDriver::PortProber.random,
+          :port => Selenium::WebDriver::PortProber.above(3000),
           :log => true
         ).start
       end
 
       def new_proxy
@@ -26,14 +35,37 @@
       def fixture(name)
         File.read(fixture_path(name))
       end
 
       def url_for(page)
-        "file://#{fixture_path page}"
+        SpecHelper.httpd.url_for(page)
       end
 
       def fixture_path(name)
-        File.join(File.expand_path("../", __FILE__), "fixtures", name)
+        File.join(SpecHelper.fixture_dir, name)
+      end
+
+      class HttpServer
+        def initialize(app)
+          @port = Selenium::WebDriver::PortProber.above(3000)
+
+          pid = fork do
+            Rack::Server.new(:app => app, :Port => @port).start
+          end
+
+          at_exit { Process.kill 'TERM', pid }
+
+          poller = Selenium::WebDriver::SocketPoller.new("localhost", @port, 10)
+
+          unless poller.connected?
+            raise "unable to start web server in 5 seconds"
+          end
+        end
+
+        def url_for(page)
+          # use 0.0.0.0 to avoid default no-proxy rules on localhost
+          "http://0.0.0.0:#{@port}/#{page}"
+        end
       end
 
     end
   end
 end