lib/testlab/node.rb in testlab-0.1.0 vs lib/testlab/node.rb in testlab-0.2.0

@@ -8,29 +8,33 @@
   # @author Zachary Patten <[e-mail]>
   class Node < ZTK::DSL::Base
     STATUS_KEYS   = %w(id instance_id state user ip port provider con net rtr).map(&:to_sym)
 
     # Sub-Modules
+    autoload :Actions,       'testlab/node/actions'
     autoload :Bind,          'testlab/node/bind'
     autoload :ClassMethods,  'testlab/node/class_methods'
     autoload :Lifecycle,     'testlab/node/lifecycle'
     autoload :LXC,           'testlab/node/lxc'
     autoload :MethodMissing, 'testlab/node/method_missing'
     autoload :Resolv,        'testlab/node/resolv'
     autoload :SSH,           'testlab/node/ssh'
     autoload :Status,        'testlab/node/status'
 
+    include TestLab::Node::Actions
     include TestLab::Node::Bind
     include TestLab::Node::Lifecycle
     include TestLab::Node::LXC
     include TestLab::Node::MethodMissing
     include TestLab::Node::Resolv
     include TestLab::Node::SSH
     include TestLab::Node::Status
 
     extend  TestLab::Node::ClassMethods
 
+    include TestLab::Utility::Misc
+
     # Associations and Attributes
     belongs_to :labfile,    :class_name => 'TestLab::Lab'
 
     has_many   :routers,    :class_name => 'TestLab::Router'
     has_many   :containers, :class_name => 'TestLab::Container'
@@ -43,10 +47,10 @@
 
     def initialize(*args)
       super(*args)
 
       @ui       = TestLab.ui
-      @provider = self.provider.new(self.config)
+      @provider = self.provider.new(self.config, @ui)
     end
 
   end
 
 end