spec/polonium/driver_spec.rb in polonium-0.2.2 vs spec/polonium/driver_spec.rb in polonium-0.3.0

@@ -74,11 +74,11 @@
         end
         dont_allow(driver).do_command("type", ["id=foobar", "The Text"])
 
         proc {
           driver.type "id=foobar", "The Text"
-        }.should raise_error(Test::Unit::AssertionFailedError)
+        }.should raise_error(PoloniumError)
       end
     end
 
     describe "#click" do
       it "click when element is present and types" do
@@ -98,11 +98,11 @@
         end
         dont_allow(driver).do_command("click", [])
 
         proc {
           driver.click "id=foobar"
-        }.should raise_error(Test::Unit::AssertionFailedError)
+        }.should raise_error(PoloniumError)
       end
     end
 
     describe "#select" do
       it "types when element is present and types" do
@@ -121,11 +121,11 @@
         end
         dont_allow(driver).do_command("select", ["id=foobar", "value=3"])
 
         proc {
           driver.select "id=foobar", "value=3"
-        }.should raise_error(Test::Unit::AssertionFailedError)
+        }.should raise_error(PoloniumError)
       end
     end
 
     describe "#click" do
       it "click when element is present and types" do
@@ -145,10 +145,10 @@
         end
         dont_allow(driver).do_command("click", ["id=foobar"])
 
         proc {
           driver.click "id=foobar"
-        }.should raise_error(Test::Unit::AssertionFailedError)
+        }.should raise_error(PoloniumError)
       end
     end
   end
 end