fix specs for exception handling

- setup a class heirarchy of exceptions so i can easily rethrow
  any internal ohai error
- catch all the non-internal errors from running the plugin and
  just warn on those
- convert the argumenterror raises to ohai invalidplugin errors

1 files changed, 2 insertions, 2 deletions

diff --git a/spec/unit/runner_spec.rb b/spec/unit/runner_spec.rb
index a515bdc5..b429fa8c 100644
--- a/spec/unit/runner_spec.rb
+++ b/spec/unit/runner_spec.rb
@@ -29,7 +29,7 @@ describe Ohai::Runner, "run_plugin" do
 
   describe "when running an invalid plugin" do
     it "should raise error" do
-      lambda { @runner.run_plugin(double("Ohai::NotPlugin")) }.should raise_error(ArgumentError)
+      lambda { @runner.run_plugin(double("Ohai::NotPlugin")) }.should raise_error(Ohai::Exceptions::InvalidPlugin)
     end
   end
 
@@ -120,7 +120,7 @@ describe Ohai::Runner, "run_plugin" do
       let(:version) { :versionBla }
 
       it "should raise error" do
-        lambda { @runner.run_plugin(plugin) }.should raise_error(ArgumentError)
+        lambda { @runner.run_plugin(plugin) }.should raise_error(Ohai::Exceptions::InvalidPlugin)
       end
     end