unittests/links_multi_test.rb in watir-1.5.4 vs unittests/links_multi_test.rb in watir-1.5.5

@@ -3,34 +3,48 @@
 
 $LOAD_PATH.unshift File.join(File.dirname(__FILE__), '..') if $0 == __FILE__
 require 'unittests/setup'
 
 class TC_Links_Multi < Test::Unit::TestCase
-  include Watir
   
   def setup
-    $ie.goto $htmlRoot + 'links_multi.html'
+    browser.goto $htmlRoot + 'links_multi.html'
   end
   
   def test_existing
-    assert_match(/not.html/, $ie.link(:class_name, 'Papa').href)
-    assert_match(/mama.html/, $ie.link(:text, 'click').href)
+    assert_match(/not.html/, browser.link(:class_name, 'Papa').href)
+    assert_match(/mama.html/, browser.link(:text, 'click').href)
   end
   
   def test_class_alias
-    assert_match(/not.html/, $ie.link(:class, 'Papa').href)
+    assert_match(/not.html/, browser.link(:class, 'Papa').href)
   end
 
   def test_hash_syntax
-    assert_match(/not.html/, $ie.link(:class_name => 'Papa').href)
-    assert_match(/mama.html/, $ie.link(:text => 'click').href)
+    assert_match(/not.html/, browser.link(:class_name => 'Papa').href)
+    assert_match(/mama.html/, browser.link(:text => 'click').href)
   end
   
   def test_class_and_text
-    assert_match(/papa.html/, $ie.link(:class => 'Papa', :text => 'click').href)
+    assert_match(/papa.html/, browser.link(:class => 'Papa', :text => 'click').href)
   end
   
   def test_class_and_index
-    assert_match(/papa.html/, $ie.link(:class => 'Papa', :index => 2).href)
+    assert_match(/papa.html/, browser.link(:class => 'Papa', :index => 2).href)
   end  
+
+  include Watir::Exception
+  def test_not_found_single
+    exception = assert_raise(UnknownObjectException) do
+      browser.link(:id, 'Missing').href
+    end
+    assert_equal('Unable to locate element, using :id, "Missing"', exception.message)
+  end
+  
+  def test_not_found_with_multi
+    exception = assert_raise(UnknownObjectException) do
+      browser.link(:class => 'Missing', :index => 2).href
+    end
+    assert_equal('Unable to locate element, using {:class=>"Missing", :index=>2}', exception.message)
+  end
   
 end
\ No newline at end of file