polonium 0.1.1 → 0.2.0

data/lib/polonium/element.rb

@@ -9,16 +9,15 @@ module Polonium
     end
 
     def assert_element_present(params={})
-      driver.wait_for_is_element_present(locator, params)
+      driver.assert_element_present(locator, params)
     end
 
     def assert_element_not_present(params={})
-      driver.wait_for_is_element_not_present(locator, params)
+      driver.assert_element_not_present(locator, params)
     end
 
     def assert_value(expected_value)
-      assert_element_present
-      wait_for do |configuration|
+      wait_for_element do |configuration|
         actual_value = driver.get_value(locator)
         configuration.message = "Expected '#{locator}' to be '#{expected_value}' but was '#{actual_value}'"
         has_value? expected_value, actual_value
@@ -26,18 +25,25 @@ module Polonium
     end
 
     def assert_attribute(expected_name, expected_value)
-      assert_element_present
       attr_locator = "#{locator}@#{expected_name}"
-      wait_for do |configuration|
+      wait_for_element do |configuration|
         actual = driver.get_attribute(attr_locator)  #todo: actual value
         configuration.message = "Expected attribute '#{attr_locator}' to be '#{expected_value}' but was '#{actual}'"
         values_match? actual, expected_value
       end
     end
 
+    def assert_attribute_does_not_contain(attribute_name, illegal_value)
+      attr_locator = "#{locator}@#{attribute_name}"
+      wait_for_element do |configuration|
+        actual = driver.get_attribute(attr_locator)  #todo: actual value
+        configuration.message = "Expected attribute '#{attr_locator}' to not contain '#{illegal_value}' but was '#{actual}'"
+        !actual.match(illegal_value)
+      end
+    end
+
     def assert_selected(expected_value)
-      assert_element_present
-      wait_for do |configuration|
+      wait_for_element do |configuration|
         actual = driver.get_selected_label(locator)
         configuration.message = "Expected '#{locator}' to be selected with '#{expected_value}' but was '#{actual}"
         values_match? actual, expected_value
@@ -45,42 +51,37 @@ module Polonium
     end
 
     def assert_visible(options={})
-      assert_element_present
       options = {
         :message => "Expected '#{locator}' to be visible, but it wasn't"
       }.merge(options)
-      wait_for(options) do
+      wait_for_element(options) do
         driver.is_visible(locator)
       end
     end
 
     def assert_not_visible(options={})
-      assert_element_present
       options = {
         :message => "Expected '#{locator}' to be hidden, but it wasn't"
       }.merge(options)
-      wait_for(options) do
+      wait_for_element(options) do
         !driver.is_visible(locator)
       end
     end
 
     def assert_checked
-      assert_element_present
-      wait_for(:message => "Expected '#{locator}' to be checked") do
+      wait_for_element(:message => "Expected '#{locator}' to be checked") do
         driver.is_checked(locator)
       end
     end
 
     def assert_not_checked
-      assert_element_present
-      wait_for(:message => "Expected '#{locator}' to not be checked") do
+      wait_for_element(:message => "Expected '#{locator}' to not be checked") do
         !driver.is_checked(locator)
       end
     end
 
     def assert_text(expected_text, options={})
-      assert_element_present
-      wait_for(options) do |configuration|
+      wait_for_element(options) do |configuration|
         actual = driver.get_text(locator)
         configuration.message = "Expected text '#{expected_text}' to be full contents of #{locator} but was '#{actual}')"
         values_match? actual, expected_text
@@ -89,34 +90,32 @@ module Polonium
 
     def assert_contains(expected_text, options={})
       return assert_contains_in_order(*expected_text) if expected_text.is_a? Array
-      assert_element_present
-      options = {
-        :message => "#{locator} should contain #{expected_text}"
-      }.merge(options)
-      wait_for(options) do
-        contains?(expected_text)
+      wait_for_element(options) do |configuration|
+        if contains?(expected_text)
+          true
+        else
+          configuration.message = "#{locator} should contain #{expected_text}"
+          false
+        end
       end
     end
 
     def assert_does_not_contain(expected_text, options={})
-      assert_element_present
-      wait_for(options) do
+      wait_for_element(options) do
         !contains?(expected_text)
       end
     end
 
     def assert_next_sibling(expected_sibling_id, options = {})
-      assert_element_present
       eval_js = "this.page().findElement('#{locator}').nextSibling.id"
-      wait_for(:message => "id '#{locator}' should be next to '#{expected_sibling_id}'") do
+      wait_for_element(:message => "id '#{locator}' should be next to '#{expected_sibling_id}'") do
         actual_sibling_id = driver.get_eval(eval_js)
         expected_sibling_id == actual_sibling_id
       end
     end
 
     def assert_contains_in_order(*text_fragments)
-      assert_element_present
-      wait_for do |configuration|
+      wait_for_element do |configuration|
         success = false
 
         html = driver.get_text(locator)
@@ -139,7 +138,26 @@ module Polonium
         success
       end
     end
+    
+    def assert_number_of_children(expected_number)
+      eval_js = "this.page().findElement('#{locator}').childNodes.length"
+      wait_for_element(:message => "id '#{locator}' should contain exactly #{expected_number} children") do
+        actual_number = driver.get_eval(eval_js)
+        expected_number == actual_number.to_i
+      end
+    end
 
+    def click
+      driver.click locator
+    end
+
+    def type(text)
+      driver.type(locator, text)
+    end
+
+    def select(option_locator)
+      driver.select(locator, option_locator)
+    end
 
     def is_present?
       driver.is_element_present(locator)
@@ -169,6 +187,18 @@ module Polonium
     end
 
     protected
+    def wait_for_element(options={})
+      wait_for(options) do |configuration|
+        if is_present?
+          configuration.message = ""
+          yield(configuration)
+        else
+          configuration.message = "Expected element '#{locator}' to be present, but it was not"
+          false
+        end
+      end
+    end
+
     def method_missing(method_name, *args, &blk)
       if driver.respond_to?(method_name)
         driver_args = [locator] + args

data/lib/polonium/page.rb

@@ -2,7 +2,18 @@ module Polonium
   class Page
     include WaitFor, ValuesMatch
     attr_reader :driver
-    PAGE_LOADED_COMMAND = "this.browserbot.getDocument().body ? true : false"
+    PAGE_LOADED_COMMAND = <<-JS
+      (function(selenium) {
+        BrowserBot.prototype.bodyText = function() {
+            if (!this.getDocument().body) {
+                return "";
+            }
+            return getText(this.getDocument().body);
+        };
+        selenium.browserbot.bodyText = BrowserBot.prototype.bodyText;
+        return selenium.browserbot.getDocument().body ? true : false;
+      })(this);
+    JS
 
     def initialize(driver)
       @driver = driver
@@ -91,4 +102,4 @@ module Polonium
       end
     end
   end
-end
+end

data/lib/polonium/server_runners/external_server_runner.rb

@@ -0,0 +1,20 @@
+module Polonium
+  module ServerRunners
+    class ExternalServerRunner < ServerRunner
+      protected
+      def start_server
+        stop_server
+        system("cd #{configuration.rails_root}; script/server -e #{configuration.rails_env} -p #{configuration.internal_app_server_port} -c #{configuration.rails_root}")
+      rescue Exception => e
+        puts e.message
+        puts e.backtrace
+        raise e
+      end
+
+      def stop_server
+        cmd = "ps ax | grep 'script/server -e #{configuration.rails_env}' | sed /grep/d | awk '{print $1}' | xargs kill -9 2>/dev/null"
+        system(cmd)
+      end
+    end
+  end
+end

data/lib/polonium/server_runners/mongrel_server_runner.rb

@@ -0,0 +1,63 @@
+module Polonium
+  module ServerRunners
+    class MongrelServerRunner < ServerRunner
+      def start
+        mongrel_configurator = create_mongrel_configurator
+        initialize_server(mongrel_configurator)
+
+        Thread.start do
+          start_server(mongrel_configurator)
+        end
+        @started = true
+      end
+
+      protected
+      def start_server(mongrel_configurator)
+        mongrel_configurator.run
+        mongrel_configurator.log "Mongrel running at #{configuration.internal_app_server_host}:#{configuration.internal_app_server_port}"
+        mongrel_configurator.join
+      end
+
+      def initialize_server(config)
+        configuration = self.configuration
+        config.listener do |*args|
+          mongrel = (args.first || self)
+          mongrel.log "Starting Rails in environment #{defaults[:environment]} ..."
+          mongrel.uri "/", :handler => mongrel.rails
+          mongrel.log "Rails loaded."
+
+          mongrel.log "Loading any Rails specific GemPlugins"
+          mongrel.load_plugins
+          configuration.app_server_initialization.call(mongrel)
+        end
+      end
+
+      def stop_server
+      end
+
+      def create_mongrel_configurator #:nodoc:
+        dir = File.dirname(__FILE__)
+        require 'mongrel/rails'
+        settings = {
+          :host => configuration.internal_app_server_host,
+          :port => configuration.internal_app_server_port,
+          :cwd => configuration.rails_root,
+          :log_file => "#{configuration.rails_root}/log/mongrel.log",
+          :pid_file => "#{configuration.rails_root}/log/mongrel.pid",
+          :environment => configuration.rails_env,
+          :docroot => "#{configuration.rails_root}/public",
+          :mime_map => nil,
+          :daemon => false,
+          :debug => false,
+          :includes => ["mongrel"],
+          :config_script => nil
+        }
+
+        configurator = Mongrel::Rails::RailsConfigurator.new(settings) do
+          log "Starting Mongrel in #{defaults[:environment]} mode at #{defaults[:host]}:#{defaults[:port]}"
+        end
+        configurator
+      end
+    end
+  end
+end

data/lib/polonium/server_runners/server_runner.rb

@@ -0,0 +1,36 @@
+module Polonium
+  module ServerRunners
+    class ServerRunner
+      attr_reader :configuration
+      def initialize(configuration)
+        @configuration = configuration
+        @started = false
+      end
+
+      def start
+        Thread.start do
+          start_server
+        end
+        @started = true
+      end
+
+      def stop
+        stop_server
+        @started = false
+      end
+
+      def started?
+        @started
+      end
+
+      protected
+      def start_server
+        raise NotImplementedError.new("this is abstract!")
+      end
+
+      def stop_server
+        raise NotImplementedError.new("this is abstract!")
+      end
+    end
+  end
+end

data/lib/polonium/server_runners/webrick_server_runner.rb

@@ -0,0 +1,49 @@
+module Polonium
+  module ServerRunners
+    class WebrickServerRunner < ServerRunner
+      attr_accessor :server
+
+      def initialize(configuration)
+        require 'webrick_server'
+        super
+      end
+
+      protected
+      def start_server
+        @server = create_webrick_server
+        mount_parameters = {
+          :port            => configuration.internal_app_server_port,
+          :ip              => configuration.internal_app_server_host,
+          :environment     => configuration.rails_env.dup,
+          :server_root     => configuration.server_root,
+          :server_type     => WEBrick::SimpleServer,
+          :charset         => "UTF-8",
+          :mime_types      => WEBrick::HTTPUtils::DefaultMimeTypes,
+          :working_directory => File.expand_path(configuration.rails_root.to_s)
+        }
+        server.mount('/', DispatchServlet, mount_parameters)
+
+        trap("INT") { stop_server }
+
+        require File.expand_path("#{configuration.rails_root}/config/environment")
+        require "dispatcher"
+        server.start
+      end
+
+      def stop_server
+        server.shutdown if server
+      end
+
+      def create_webrick_server #:nodoc:
+        WEBrick::HTTPServer.new({
+          :Port => configuration.internal_app_server_port,
+          :BindAddress => configuration.internal_app_server_host,
+          :ServerType  => WEBrick::SimpleServer,
+          :MimeTypes => WEBrick::HTTPUtils::DefaultMimeTypes,
+          :Logger => configuration.new_logger,
+          :AccessLog => []
+        })
+      end
+    end
+  end
+end

data/lib/polonium/test_case.rb

@@ -12,27 +12,9 @@ module Polonium
     self.use_transactional_fixtures = false
     self.use_instantiated_fixtures  = true
 
-    include TestUnitDsl
+    include SeleniumDsl
     def setup
-      #   set "setup_once" to true
-      #   to prevent fixtures from being re-loaded and data deleted from the DB.
-      #   this is handy if you want to generate a DB full of sample data
-      #   from the tests.  Make sure none of your selenium tests manually
-      #   reset data!
-      #TODO: make this configurable
-      setup_once = false
-
       raise "Cannot use transactional fixtures if ActiveRecord concurrency is turned on (which is required for Selenium tests to work)." if self.class.use_transactional_fixtures
-      unless setup_once
-        ActiveRecord::Base.connection.update('SET FOREIGN_KEY_CHECKS = 0')
-        super
-        ActiveRecord::Base.connection.update('SET FOREIGN_KEY_CHECKS = 1')
-      else
-        unless InstanceMethods.const_defined?("ALREADY_SETUP_ONCE")
-          super
-          InstanceMethods.const_set("ALREADY_SETUP_ONCE", true)
-        end
-      end
       @selenium_driver = configuration.driver
     end
 

data/lib/polonium/wait_for.rb

@@ -12,7 +12,10 @@ module Polonium
       configuration = Context.new(message)
       begin_time = time_class.now
       while (time_class.now - begin_time) < timeout
-        return if yield(configuration)
+        if value = yield(configuration)
+          return value
+        end
+        return value if value
         sleep 0.25
       end
       flunk(configuration.message + " (after #{timeout} sec)")

data/lib/polonium.rb

@@ -1,6 +1,8 @@
+require 'rubygems'
 require 'socket'
 require 'logger'
 require "stringio"
+require "resolv-replace"
 
 require "active_record"
 
@@ -13,11 +15,11 @@ require "selenium"
 require "polonium/extensions/module"
 require "polonium/wait_for"
 require "polonium/driver"
-require "polonium/server_runner"
-require "polonium/mongrel_selenium_server_runner"
-require "polonium/webrick_selenium_server_runner"
+require "polonium/server_runners/server_runner"
+require "polonium/server_runners/external_server_runner"
+require "polonium/server_runners/mongrel_server_runner"
+require "polonium/server_runners/webrick_server_runner"
 require "polonium/dsl/selenium_dsl"
-require "polonium/dsl/test_unit_dsl"
 require "polonium/configuration"
 require "polonium/values_match"
 require "polonium/page"

data/spec/polonium/configuration_spec.rb

@@ -1,24 +1,6 @@
 require File.expand_path(File.dirname(__FILE__) + "/../spec_helper")
 
 module Polonium
-  describe Configuration, ".instance" do
-    attr_reader :configuration
-    before(:each) do
-      Configuration.instance = nil
-      @configuration = Configuration.new
-    end
-
-    it "should create a new Configuration if it hasn't been called yet" do
-      mock(Configuration).new.returns(configuration)
-      Configuration.instance.should == configuration
-    end
-
-    it "should reuse the existing Configuration if it has been called.  So new/establish_environment should only be called once." do
-      Configuration.instance.should_not be_nil
-      dont_allow(Configuration).new
-    end
-  end
-
   describe Configuration do
     attr_reader :configuration
     before(:each) do
@@ -86,70 +68,27 @@ module Polonium
       passed_driver.should == driver
       start_called.should == true
     end
+  end
 
-    it "creates a Webrick Server Runner" do
-      configuration.selenium_server_port = 4000
-      configuration.selenium_server_host = "localhost"
-      dir = File.dirname(__FILE__)
-      configuration.rails_root = dir
-      configuration.rails_env = "test"
-
-      runner = configuration.create_webrick_runner
-      runner.should be_an_instance_of(WebrickSeleniumServerRunner)
-      runner.configuration.should == configuration
-      runner.thread_class.should == Thread
-      runner.socket.should == Socket
-      runner.dispatch_servlet.should == DispatchServlet
-      runner.environment_path.should == File.expand_path("#{dir}/config/environment")
-    end
-
-    it "creates webrick http server" do
-      configuration.internal_app_server_port = 4000
-      configuration.internal_app_server_host = "localhost"
-
-      mock_logger = "logger"
-      mock(configuration).new_logger {mock_logger}
-      mock(WEBrick::HTTPServer).new({
-        :Port => 4000,
-        :BindAddress => "localhost",
-        :ServerType  => WEBrick::SimpleServer,
-        :MimeTypes => WEBrick::HTTPUtils::DefaultMimeTypes,
-        :Logger => mock_logger,
-        :AccessLog => []
-      })
-      server = configuration.create_webrick_server
-    end
-
-    it "creates Mongrel Server Runner" do
-      server = configuration.create_mongrel_runner
-      server.should be_instance_of(MongrelSeleniumServerRunner)
-      server.configuration.should == configuration
-      server.thread_class.should == Thread
-    end
-
-    it "creates Mongrel configurator" do
-      configuration.internal_app_server_host = "localhost"
-      configuration.internal_app_server_port = 4000
-      configuration.rails_env = "test"
-      configuration.rails_root = rails_root = File.dirname(__FILE__)
-
-      configurator = configuration.create_mongrel_configurator
-      configurator.defaults[:host].should == "localhost"
-      configurator.defaults[:port].should == 4000
-      configurator.defaults[:cwd].should == configuration.rails_root
-      configurator.defaults[:log_file].should == "#{configuration.rails_root}/log/mongrel.log"
-      configurator.defaults[:pid_file].should == "#{configuration.rails_root}/log/mongrel.pid"
-      configurator.defaults[:environment].should == "test"
-      configurator.defaults[:docroot].should == "#{rails_root}/public"
-      configurator.defaults[:mime_map].should be_nil
-      configurator.defaults[:daemon].should == false
-      configurator.defaults[:debug].should == false
-      configurator.defaults[:includes].should == ["mongrel"]
-      configurator.defaults[:config_script].should be_nil
+  describe ".instance" do
+    attr_reader :configuration
+    before(:each) do
+      Configuration.instance = nil
+      @configuration = Configuration.new
+    end
+
+    it "should create a new Configuration if it hasn't been called yet" do
+      mock(Configuration).new.returns(configuration)
+      Configuration.instance.should == configuration
+    end
+
+    it "should reuse the existing Configuration if it has been called.  So new/establish_environment should only be called once." do
+      Configuration.instance.should_not be_nil
+      dont_allow(Configuration).new
     end
   end
 
-  describe Configuration, "#establish_environment" do
+  describe "#establish_environment" do
     attr_reader :configuration
     before(:each) do
       @old_configuration = Configuration.instance
@@ -296,7 +235,7 @@ module Polonium
     end
   end
 
-  describe Configuration, "#stop_driver_if_necessary" do
+  describe "#stop_driver_if_necessary" do
     attr_reader :configuration
     before(:each) do
       @configuration = Configuration.new
@@ -330,30 +269,33 @@ module Polonium
 
   end
 
-  describe Configuration, "#create_server_runner where application server engine is mongrel" do
-    it "creates a mongrel server runner" do
-      configuration = Configuration.new
-      configuration.app_server_engine = :mongrel
-      runner = configuration.create_server_runner
-      runner.should be_instance_of(MongrelSeleniumServerRunner)
+  describe "#create_app_server_runner" do
+    describe "when server engine in mongrel" do
+      it "creates a mongrel server runner" do
+        configuration = Configuration.new
+        configuration.app_server_engine = :mongrel
+        runner = configuration.create_app_server_runner
+        runner.class.should == ServerRunners::MongrelServerRunner
+        runner.configuration.should == configuration
+      end
     end
-  end
 
-  describe Configuration, "#create_server_runner where application server engine is webrick" do
-    before do
-      Object.const_set :RAILS_ROOT, "foobar"
-      require 'webrick_server'
-    end
+    describe "when server engine is webrick" do
+      before do
+        Object.const_set :RAILS_ROOT, "foobar"
+      end
 
-    after do
-      Object.instance_eval {remove_const :RAILS_ROOT}
-    end
+      after do
+        Object.instance_eval {remove_const :RAILS_ROOT}
+      end
 
-    it "creates a webrick server runner" do
-      configuration = Configuration.new
-      configuration.app_server_engine = :webrick
-      runner = configuration.create_server_runner
-      runner.should be_instance_of(WebrickSeleniumServerRunner)
+      it "creates a webrick server runner" do
+        configuration = Configuration.new
+        configuration.app_server_engine = :webrick
+        runner = configuration.create_app_server_runner
+        runner.class.should == ServerRunners::WebrickServerRunner
+        runner.configuration.should == configuration
+      end
     end
   end
 end