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

@@ -2,31 +2,47 @@
 
 describe Libvirt::Ruby::Connect do
   let(:connect) { Libvirt::Ruby::Connect }
 
   context "when calling method #dispatcher" do
-    context "with a valid libvirt function" do
+    context "with libvirt installed" do
       before :each do
-        connect.stub(:attach_function).with("virConnectClose", "virConnectClose", [], :int).and_return(true)
-        connect.stub(:send).with("virConnectClose", [])
+        connect.stub(:ffi_lib).with("libvirt")
       end
 
-      after :each do
-        connect.dispatcher('Close', [:int])
-      end
+      context "and 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
 
-      it "should attach it as a binding for C's function" do
-        connect.should_receive(:attach_function).with("virConnectClose", "virConnectClose", [], :int).and_return(true)
+        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)
+        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 "and an invalid libvirt function" do
+        before :each do
+          connect.stub(:attach_function).with("virConnectAbc", "virConnectAbc", [], :int).and_raise(FFI::NotFoundError.new('Abc', 'libvirt'))
+        end
+
+        it "should raise an exception" do
+          lambda { connect.dispatcher('Abc', [], :int) }.should raise_error(Libvirt::Ruby::Exceptions::InvalidFunction)
+        end
       end
     end
 
-    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)
+    context "without libvirt installed" do
+      it "should raise an exception" do
+        lambda { connect.dispatcher('Abc', [], :int) }.should raise_error(Libvirt::Ruby::Exceptions::MissingLib)
       end
     end
   end
 end
\ No newline at end of file