mcollective-test 0.3.0 → 0.4.0

data/lib/mcollective/test.rb

@@ -2,6 +2,7 @@ module MCollective
   module Test
     autoload :Util, "mcollective/test/util"
     autoload :Matchers, "mcollective/test/matchers"
+    autoload :DataTest, "mcollective/test/data_test.rb"
     autoload :LocalAgentTest, "mcollective/test/local_agent_test.rb"
     autoload :RemoteAgentTest, "mcollective/test/remote_agent_test.rb"
     autoload :ApplicationTest, "mcollective/test/application_test.rb"

data/lib/mcollective/test/data_test.rb

@@ -0,0 +1,44 @@
+module MCollective
+  module Test
+    class DataTest
+      attr_reader :config, :logger, :plugin
+
+      include Test::Util
+
+      def initialize(data, options={})
+        config = options.fetch(:config, {})
+
+        ARGV.clear
+
+        @config = create_config_mock(config)
+        @logger = create_logger_mock
+
+        @plugin = load_data(data, options[:data_file])
+        @plugin.stubs(:ddl_validate).returns(true)
+      end
+
+      def load_data(data, data_file=nil)
+        classname = "MCollective::Data::%s " % data.capitalize
+
+        PluginManager.delete(data.downcase)
+
+        if data_file
+          raise("Cannot find data file %s for data plugin %s" % [data_file, data]) unless File.exist?(data_file)
+          load data_file
+        else
+          PluginManager.loadclass(classname)
+        end
+
+        ddl = DDL::Base.new(data, :data, false)
+
+        DDL.stubs(:new).with(data, :data).returns(ddl)
+
+        unless PluginManager.pluginlist.include?(data.downcase)
+          PluginManager << {:type => data, :class => "MCollective::Data::%s" % data.capitalize, :single_instance => false}
+        end
+
+        PluginManager[data.downcase]
+      end
+    end
+  end
+end

data/lib/mcollective/test/matchers/rpc_result_items.rb

@@ -15,7 +15,12 @@ module MCollective
           end
 
           [actual].flatten.each do |result|
-            result = result.results if result.is_a?(MCollective::RPC::Result)
+            if result.is_a?(MCollective::RPC::Result)
+              result = result.results
+            elsif result.is_a?(MCollective::Data::Result)
+              result = {:data => result.instance_variable_get("@data")}
+            end
+
             @nodeid = result[:data][:test_sender]
             @actual << result
             @expected.each do |e|

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification 
 name: mcollective-test
 version: !ruby/object:Gem::Version 
-  hash: 19
+  hash: 15
   prerelease: false
   segments: 
   - 0
-  - 3
+  - 4
   - 0
-  version: 0.3.0
+  version: 0.4.0
 platform: ruby
 authors: 
 - R.I.Pienaar
@@ -15,7 +15,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2012-10-17 00:00:00 +01:00
+date: 2013-01-07 00:00:00 +00:00
 default_executable: 
 dependencies: []
 
@@ -32,6 +32,7 @@ files:
 - lib/mcollective/test/local_agent_test.rb
 - lib/mcollective/test/matchers.rb
 - lib/mcollective/test/remote_agent_test.rb
+- lib/mcollective/test/data_test.rb
 - lib/mcollective/test/matchers/rpc_metadata.rb
 - lib/mcollective/test/matchers/rpc_result_items.rb
 - lib/mcollective/test/matchers/rpc_status.rb