spec/libvirt-ruby/connect_spec.rb in libvirt-ruby-0.0.1 vs spec/libvirt-ruby/connect_spec.rb in libvirt-ruby-0.0.2

@@ -2,23 +2,31 @@
 
 describe Libvirt::Ruby::Connect do
   let(:connect) { Libvirt::Ruby::Connect }
 
   context "when calling method #dispatcher" do
-    before :each do
-      connect.stub(:attach_function).with("virConnectClose", "virConnectClose", [], :int).and_return(true)
-      connect.stub(:send).with("virConnectClose", [])
-    end
+    context "with a valid libvirt function" do
+      before :each do
+        connect.stub(:attach_function).with("virConnectClose", "virConnectClose", [], :int).and_return(true)
+        connect.stub(:send).with("virConnectClose", [])
+      end
 
-    after :each do
-      connect.dispatcher('Close', [:int])
-    end
+      after :each do
+        connect.dispatcher('Close', [:int])
+      end
 
-    it "should attach it as a binding for C's function" do
-      connect.should_receive(:attach_function).with("virConnectClose", "virConnectClose", [], :int).and_return(true)
+      it "should attach it as a binding for C's function" do
+        connect.should_receive(:attach_function).with("virConnectClose", "virConnectClose", [], :int).and_return(true)
+      end
+
+      it "should call the new attached method" do
+        connect.should_receive(:send).with("virConnectClose", [])
+      end
     end
 
-    it "should call the new attached method" do
-      connect.should_receive(:send).with("virConnectClose", [])
+    context "with an invalid libvirt function" do
+      it "should return an invalid function message" do
+        lambda { connect.dispatcher('Abc', [:int]) }.should raise_error(Libvirt::Ruby::Exceptions::InvalidFunction)
+      end
     end
   end
 end
\ No newline at end of file