Stub windows only constants / methods in unit tests to work outside of Windows

1 files changed, 8 insertions, 0 deletions

diff --git a/spec/unit/wmi_spec.rb b/spec/unit/wmi_spec.rb
index 3ab648d..6ce84fb 100644
--- a/spec/unit/wmi_spec.rb
+++ b/spec/unit/wmi_spec.rb
@@ -149,6 +149,10 @@ describe WmiLite::Wmi do
   end
 
   shared_examples_for 'an invalid query' do
+    before(:each) do
+      stub_const('WIN32OLE', Class.new)
+      WIN32OLE.stub(:new).with("WbemScripting.SWbemLocator").and_return(wbem_locator)
+    end
     let(:unparseable_error) { 'unparseableerror' }
     it 'should raise an exception' do
       wbem_connection.should_receive(:ExecQuery).and_raise(WIN32OLERuntimeError.new(unparseable_error))
@@ -169,6 +173,10 @@ describe WmiLite::Wmi do
   end
 
   shared_examples_for 'an invalid namespace' do
+    before(:each) do
+      stub_const('WIN32OLE', Class.new)
+      WIN32OLE.stub(:new).with("WbemScripting.SWbemLocator").and_return(wbem_locator)
+    end
     let(:unparseable_error) { 'unparseableerror' }
     it 'should raise an exception' do
       wbem_locator.should_receive(:ConnectServer).and_raise(WIN32OLERuntimeError.new('unparseableerror'))