Update code from https://github.com/dmi-try/marionette-collective

[packages/precise/mcollective.git] / spec / unit / data_spec.rb

diff --git a/spec/unit/data_spec.rb b/spec/unit/data_spec.rb
index a117ce0bf4db828ea96f5ca49df758d85fc2d3a2..cc7dbf8c2a5fdc0e2880aa3c162a8f30955143f8 100644 (file)
--- a/spec/unit/data_spec.rb
+++ b/spec/unit/data_spec.rb
@@ -17,6 +17,8 @@ module MCollective
 
         ddl = mock
         ddl.stubs(:meta).returns({:timeout => 1})
+        ddl.stubs(:dataquery_interface).returns({:rspec => nil})
+        ddl.stubs(:dataquery_interface).returns({:output => {}})
         DDL.stubs(:new).returns(ddl)
         Data::Base.expects(:activate?).returns(false)
         PluginManager.expects("[]").with("rspec_data").returns(Data::Base.new)
@@ -30,6 +32,7 @@ module MCollective
 
         ddl = mock
         ddl.stubs(:meta).returns({:timeout => 1})
+        ddl.stubs(:dataquery_interface).returns({:output => {}})
         DDL.stubs(:new).returns(ddl)
         Data::Base.expects(:activate?).raises("rspec failure")
         Log.expects(:debug).once.with("Disabling data plugin rspec_data due to exception RuntimeError: rspec failure")