lib/sunzi/cloud.rb in sunzi-1.5.1 vs lib/sunzi/cloud.rb in sunzi-1.5.2

@@ -1,18 +1,24 @@
 module Sunzi
   class Cloud
+    include Sunzi::Utility
+
     def initialize(cli, provider)
       @subject = case provider
       when 'linode'
         Sunzi::Cloud::Linode.new(cli, provider)
       when 'digital_ocean'
         Sunzi::Cloud::DigitalOcean.new(cli, provider)
       else
-        abort_with "#{provider} is not valid!"
+        abort_with "Provider #{provider} is not valid!"
       end
     end
 
     def method_missing(sym, *args, &block)
       @subject.send sym, *args, &block
+    end
+
+    def respond_to?(method)
+      @subject.respond_to?(sym) || super
     end
   end
 end