Updated mcollective.init according to OSCI-658

[packages/precise/mcollective.git] / spec / unit / generators / data_generator_spec.rb

diff --git a/spec/unit/generators/data_generator_spec.rb b/spec/unit/generators/data_generator_spec.rb
new file mode 100644 (file)
index 0000000..863b6e2
--- /dev/null
+++ b/spec/unit/generators/data_generator_spec.rb
@@ -0,0 +1,37 @@
+#! /usr/bin/env rspec
+
+require 'spec_helper'
+
+module MCollective
+  module Generators
+      describe DataGenerator do
+
+        before :each do
+          DataGenerator.stubs(:create_metadata_string).returns("meta\n")
+        end
+
+        describe "#create_ddl" do
+          it "create the correct ddl string" do
+            DataGenerator.any_instance.stubs(:create_plugin_content)
+            DataGenerator.any_instance.stubs(:create_plugin_string)
+            DataGenerator.any_instance.stubs(:write_plugins)
+
+            ddl = DataGenerator.new("foo", ["output"]).ddl
+            expected = File.read(File.join(File.dirname(__FILE__), "snippets", "data_ddl")).chop
+            ddl.should == expected
+          end
+        end
+
+        describe "#create_plugin_content" do
+          it "should create the correct plugin content" do
+            DataGenerator.any_instance.stubs(:create_ddl)
+            DataGenerator.any_instance.stubs(:create_plugin_string)
+            DataGenerator.any_instance.stubs(:write_plugins)
+
+            ddl = DataGenerator.new("foo", ["output"]).content
+            ddl.should == "      query do |what|\n        result[:output] = nil\n      end\n"
+          end
+        end
+      end
+  end
+end