lib/page_models/configuration.rb in pagemodels-0.1.1 vs lib/page_models/configuration.rb in pagemodels-0.1.2

@@ -2,14 +2,21 @@
 
 module PageModels  
   class Configuration
     include Singleton
     attr_writer :driver
+    attr_accessor :base_url
     
     def initialize
+      reset!
+    end
+    
+    def reset!
       @driver = :capybara
+      @base_url = "http://localhost:3000"
       @frameworks = []
+      @driver_instance = nil
     end
         
     def integrate(framework)
       @frameworks << framework
     end
@@ -17,14 +24,18 @@
     def integrate!      
       @frameworks.each { |framework| require "page_models/integration/#{framework}" }
     end
     
     def driver
-      case @driver
-        when :capybara
-          Capybara.current_session
-        else
-          raise ConfigurationError.new("No driver configured.")
+      @driver_instance ||= begin
+        case @driver
+          when :capybara
+            Capybara.current_session
+          when :celerity
+            Celerity::Browser.new
+          else
+            Watir::Browser.new(@driver)
+        end
       end
     end
   end
 end
\ No newline at end of file