unittests/windows/ie-each_test.rb in watir-1.6.5 vs unittests/windows/ie-each_test.rb in watir-1.6.6.rc1

- old
+ new

@@ -1,48 +1,48 @@ -# Not intended to be run as part of a larger suite. - -$LOAD_PATH.unshift File.join(File.dirname(__FILE__), '..', '..') unless $SETUP_LOADED -require 'test/unit' -require 'watir' -require 'watir/process' -require 'watir/waiter' - -class TC_IE_Each < Test::Unit::TestCase - def setup - assert_equal 0, Watir::IE.process_count - @hits = 0 - @ie = [] - end - - def hit_me - @hits += 1 - end - - def test_zero_windows - Watir::IE.each {hit_me} - assert_equal 0, @hits - end - - def test_one_window - @ie << Watir::IE.new_process - Watir::IE.each {hit_me} - assert_equal 1, @hits - end - - def test_two_windows - @ie << Watir::IE.new_process - @ie << Watir::IE.new_process - Watir::IE.each {hit_me} - assert_equal 2, @hits - end - - def test_return_type - @ie << Watir::IE.new_process - Watir::IE.each {|ie| assert_equal(Watir::IE, ie.class)} - end - - include Watir - def teardown - @ie.each {|ie| ie.close } - wait_until {Watir::IE.process_count == 0} - end +# Not intended to be run as part of a larger suite. + +$LOAD_PATH.unshift File.join(File.dirname(__FILE__), '..', '..') unless $SETUP_LOADED +require 'test/unit' +require 'watir' +require 'watir/process' +require 'watir/waiter' + +class TC_IE_Each < Test::Unit::TestCase + def setup + assert_equal 0, Watir::IE.process_count + @hits = 0 + @ie = [] + end + + def hit_me + @hits += 1 + end + + def test_zero_windows + Watir::IE.each {hit_me} + assert_equal 0, @hits + end + + def test_one_window + @ie << Watir::IE.new_process + Watir::IE.each {hit_me} + assert_equal 1, @hits + end + + def test_two_windows + @ie << Watir::IE.new_process + @ie << Watir::IE.new_process + Watir::IE.each {hit_me} + assert_equal 2, @hits + end + + def test_return_type + @ie << Watir::IE.new_process + Watir::IE.each {|ie| assert_equal(Watir::IE, ie.class)} + end + + include Watir + def teardown + @ie.each {|ie| ie.close } + wait_until {Watir::IE.process_count == 0} + end end \ No newline at end of file