Updated mcollective.init according to OSCI-658

[packages/precise/mcollective.git] / spec / unit / plugins / mcollective / discovery / mc_spec.rb

diff --git a/spec/unit/plugins/mcollective/discovery/mc_spec.rb b/spec/unit/plugins/mcollective/discovery/mc_spec.rb
new file mode 100644 (file)
index 0000000..bcba8d3
--- /dev/null
+++ b/spec/unit/plugins/mcollective/discovery/mc_spec.rb
@@ -0,0 +1,40 @@
+#!/usr/bin/env rspec
+
+require 'spec_helper'
+
+require File.dirname(__FILE__) + '/../../../../../plugins/mcollective/discovery/mc.rb'
+
+module MCollective
+  class Discovery
+    describe Mc do
+      describe "#discover" do
+        before do
+          @reply = mock
+          @reply.stubs(:payload).returns({:senderid => "rspec"})
+
+          @client = mock
+          @client.stubs(:sendreq)
+          @client.stubs(:unsubscribe)
+          @client.stubs(:receive).returns(@reply)
+
+          Log.stubs(:debug)
+        end
+
+        it "should send the ping request via the supplied client" do
+          @client.expects(:sendreq).with("ping", "discovery", Util.empty_filter).returns("123456")
+          Mc.discover(Util.empty_filter, 1, 1, @client)
+        end
+
+        it "should stop early if a limit is supplied" do
+          @client.stubs(:receive).returns(@reply).times(10)
+          Mc.discover(Util.empty_filter, 1, 10, @client).should == ("rspec," * 10).split(",")
+        end
+
+        it "should unsubscribe from the discovery reply source" do
+          @client.expects(:unsubscribe).with("discovery", :reply)
+          Mc.discover(Util.empty_filter, 1, 10, @client).should == ("rspec," * 10).split(",")
+        end
+      end
+    end
+  end
+end