spec/unit/strategies/extraction_strategy_spec.rb in versioncake-4.0.2 vs spec/unit/strategies/extraction_strategy_spec.rb in versioncake-4.1.0

@@ -58,16 +58,16 @@
   end
 
   it "it fails to create a custom strategy for a proc with no parameters" do
     expect do
       VersionCake::ExtractionStrategy.lookup(lambda{})
-    end.to raise_error(Exception)
+    end.to raise_error(VersionCake::ExtractionStrategy::InvalidStrategyError)
   end
 
   it "it raises error when no strategy is found" do
     expect do
       VersionCake::ExtractionStrategy.lookup(:fake_extraction)
-    end.to raise_error(Exception)
+    end.to raise_error(VersionCake::ExtractionStrategy::InvalidStrategyError)
   end
 
   describe '.list' do
     let(:strategies) { :http_header }
     subject { VersionCake::ExtractionStrategy.list(strategies) }