bebox 0.0.1 → 0.1.0

data/lib/bebox/wizards/project_wizard.rb

@@ -1,10 +1,8 @@
-require 'bebox/project'
-require 'net/http'
-require 'uri'
 
 module Bebox
   class ProjectWizard
     include Bebox::Logger
+    include Bebox::WizardsHelper
     # Bebox boxes directory
     BEBOX_BOXES_PATH = '~/.bebox/boxes'
 
@@ -16,24 +14,9 @@ module Bebox
       bebox_boxes_setup
       # Asks to choose an existing box
       current_box = choose_box(get_existing_boxes)
-      # If choose to download/select new box
-      if current_box.nil?
-        # Keep asking for valid uri or overwriting until confirmation
-        confirm = false
-        begin
-          # Asks vagrant box location to user if not choose an existing box
-          valid_box_uri = ask_uri
-          # Confirm if the box already exist
-          confirm = box_exists?(valid_box_uri) ? confirm_overwrite? : true
-        end while !confirm
-        # Setup the box with the valid uri
-        set_box(valid_box_uri)
-      else
-        valid_box_uri = current_box
-      end
-      vagrant_box_base = "#{BEBOX_BOXES_PATH}/#{valid_box_uri}"
+      vagrant_box_base = "#{BEBOX_BOXES_PATH}/#{get_valid_box_uri(current_box)}"
       # Asks user to choose vagrant box provider
-      vagrant_box_provider = ask_box_provider
+      vagrant_box_provider = choose_option(%w{virtualbox vmware}, 'Choose the vagrant box provider')
       # Set default environments
       default_environments = %w{vagrant staging production}
       # Project creation
@@ -42,19 +25,26 @@ module Bebox
       ok "Project '#{project_name}' created!.\nMake: cd #{project_name}\nNow you can add new environments or new nodes to your project.\nSee bebox help."
     end
 
+    # If choose to download/select new box get a valid uri
+    def get_valid_box_uri(current_box)
+      return (valid_box_uri = current_box) unless current_box.nil?
+      # Keep asking for valid uri or overwriting until confirmation
+      confirm = false
+      begin
+        # Asks vagrant box location to user if not choose an existing box
+        valid_box_uri = ask_uri
+        # Confirm if the box already exist
+        confirm = box_exists?(valid_box_uri) ? confirm_action?('There is already a box with that name, do you want to overwrite it?') : true
+      end while !confirm
+      # Setup the box with the valid uri
+      set_box(valid_box_uri)
+    end
+
     # Check if there's an existing project in that dir
     def project_exists?(parent_path, project_name)
       Dir.exists?("#{parent_path}/#{project_name}")
     end
 
-    # Menu to choose vagrant box provider
-    def ask_box_provider
-      choose do |menu|
-        menu.header = title('Choose the vagrant box provider')
-        menu.choices('virtualbox', 'vmware')
-      end
-    end
-
     # Setup the bebox boxes directory
     def bebox_boxes_setup
       # Create user project directories
@@ -65,48 +55,40 @@ module Bebox
 
     # Asks vagrant box location to user until is valid
     def ask_uri
-      vbox_uri =  ask(highline_quest('Write the URI (http, local_path) for the vagrant box to be used in the project:')) do |q|
-        q.default = 'http://puppet-vagrant-boxes.puppetlabs.com/ubuntu-server-12042-x64-vbox4210-nocm.box'
-      end
+      vbox_uri = write_input('Write the URI (http, local_path) for the vagrant box to be used in the project:', 'http://puppet-vagrant-boxes.puppetlabs.com/ubuntu-server-12042-x64-vbox4210-nocm.box')
       # If valid return uri if not keep asking for uri
       uri_valid?(vbox_uri) ? (return vbox_uri) : ask_uri
     end
 
     # Setup the box in the bebox boxes directory
     def set_box(box_uri)
+      require 'uri'
       uri = URI.parse(box_uri)
       if uri.scheme == ('http' || 'https')
+        info 'Downloading box ...'
         download_box(uri)
       else
-        file_name = uri.path.split('/').last
-        `ln -fs #{uri.path} #{BEBOX_BOXES_PATH}/#{file_name}`
+        `ln -fs #{uri.path} #{BEBOX_BOXES_PATH}/#{uri.path.split('/').last}`
       end
     end
 
     # Validate uri download or local box existence
     def uri_valid?(vbox_uri)
+      require 'uri'
       uri = URI.parse(vbox_uri)
-      if uri.scheme == ('http' || 'https')
-        request = Net::HTTP.new uri.host
-        response = request.request_head uri.path
-        if response.code.to_i == 302
-          uri = URI.parse(response['location'])
-          request = Net::HTTP.new uri.host
-          response = request.request_head uri.path
-        end
-        ( response.code.to_i == 200) ? (return true) : error('Download link not valid!.')
-      else
-        File.file?(uri.path) ? (return true) : error('File path not exist!.')
-      end
+      %w{http https}.include?(uri.scheme) ? http_uri_valid?(uri) : file_uri_valid?(uri)
     end
 
-    # Ask for confirmation of overwrite a box
-    def confirm_overwrite?
-      quest 'There is already a box with that name, do you want to overwrite it?'
-      response =  ask(highline_quest('(y/n)')) do |q|
-        q.default = 'n'
-      end
-      return response == 'y' ? true : false
+    def http_uri_valid?(uri)
+      require 'net/http'
+      request = Net::HTTP.new uri.host
+      response = request.request_head uri.path
+      error('Redirections not supported.') if response.code.to_i == 302
+      ( response.code.to_i == 200) ? (return true) : error('Download link not valid!.')
+    end
+
+    def file_uri_valid?(uri)
+      File.file?(uri.path) ? (return true) : error('File path not exist!.')
     end
 
     # Check if a box with the same name already exist
@@ -121,41 +103,29 @@ module Bebox
       # Converts the bebox boxes directory to an absolute pathname
       expanded_directory = File.expand_path("#{BEBOX_BOXES_PATH}")
       # Get an array of bebox boxes paths
-      Dir["#{expanded_directory}/*"].reject {|f| File.directory? f}
+      boxes = Dir["#{expanded_directory}/*"].reject {|f| File.directory? f}
+      boxes.map{|box| box.split('/').last}
     end
 
     # Asks to choose an existing box in the bebox boxes directory
     def choose_box(boxes)
       # Menu to choose vagrant box provider
       other_box_message = 'Download/Select a new box'
-      current_box = choose do |menu|
-        menu.header = title('Choose an existing box or download/select a new box')
-        boxes.each do |box|
-          menu.choice(box.split('/').last)
-        end
-        menu.choice(other_box_message)
-      end
+      boxes << other_box_message
+      current_box = choose_option(boxes, 'Choose an existing box or download/select a new box')
       current_box = (current_box == other_box_message) ? nil : current_box
     end
 
     # Download a box by the specified uri
     def download_box(uri)
-      require 'progressbar'
-      info 'Downloading box ...'
       @counter = 0
-      # Manage redirections
-      request = Net::HTTP.new uri.host
-      response = request.request_head uri.path
-      uri = URI.parse(response['location']) if response.code.to_i == 302
-
-      # Set url variables
       url = uri.path
-      url_base = uri.host
       file_name = uri.path.split('/').last
-      expanded_directory = File.expand_path("#{BEBOX_BOXES_PATH}")
-
+      expanded_directory = File.expand_path(BEBOX_BOXES_PATH)
       # Download file to bebox boxes tmp
-      Net::HTTP.start(url_base) do |http|
+      require 'net/http'
+      require 'uri'
+      Net::HTTP.start(uri.host) do |http|
         response = http.request_head(URI.escape(url))
         ProgressBar
         pbar = ProgressBar.new('file name:', response['content-length'].to_i)

data/lib/bebox/wizards/provision_wizard.rb

@@ -1,53 +1,48 @@
-require 'bebox/provision'
-require 'bebox/node'
 
 module Bebox
   class ProvisionWizard
     include Bebox::Logger
+    include Bebox::WizardsHelper
     # Apply a step for the nodes in a environment
     def apply_step(project_root, environment, step)
       # Check if environment has configured the ssh keys
-      if Bebox::Environment.check_environment_access(project_root, environment)
-        # Check already in step nodes
-        nodes_to_step = check_nodes_to_step(project_root, environment, step)
-        # Output the nodes that are ready for provisioning step-N
-        if nodes_to_step.count > 0
-          title "Provisioning #{step} in nodes:"
-          nodes_to_step.each{|node| msg(node.hostname)}
-          linebreak
-          # Apply the nodes provisioning for step-N
-          nodes_to_step.each do |node|
-            title "Applying #{step} in node #{node.hostname}:"
-            role = Bebox::Provision.role_from_node(project_root, step, node.hostname)
-            profiles = Bebox::Provision.profiles_from_role(project_root, role) unless role.nil?
-            # Before apply generate the Puppetfile with modules from all associated profiles
-            Bebox::Provision.generate_puppetfile(project_root, step, profiles) unless profiles.nil?
-            # Before apply generate the roles and profiles modules structure for puppet step
-            Bebox::Provision.generate_roles_and_profiles(project_root, step, role, profiles)
-            provision = Bebox::Provision.new(project_root, environment, node, step)
-            provision.apply.success? ? (ok "Node '#{node.hostname}' provisioned to #{step}.") : (error "An error ocurred in the provision of #{step} for node '#{node.hostname}'")
-          end
-        else
-          warn "There are no nodes for provision in #{step}. No changes were made."
-        end
-      else
-        warn "Please add a ssh key pair (id_rsa, id_rsa.pub) in config/keys/environments/#{environment} to do this step."
+      (return warn "Please add a ssh key pair (id_rsa, id_rsa.pub) in config/keys/environments/#{environment} to do this step.") unless Bebox::Environment.check_environment_access(project_root, environment)
+      nodes_to_step = Bebox::Node.nodes_in_environment(project_root, environment, previous_checkpoint(step))
+      # Check if there are nodes for provisioning step-N
+      (return warn "There are no nodes for provision in #{step}. No changes were made.") unless nodes_to_step.count > 0
+      nodes_for_provisioning(nodes_to_step, step)
+      # Apply the nodes provisioning for step-N
+      in_step_nodes = Bebox::Node.list(project_root, environment, "steps/#{step}")
+      nodes_to_step.each do |node|
+        next unless check_node_to_step(node, in_step_nodes, step)
+        provision_step_in_node(project_root, environment, step, in_step_nodes, node)
       end
     end
 
-    # Check the nodes already in step and ask confirmation to re-do-it
-    def check_nodes_to_step(project_root, environment, step)
-      nodes_to_step = []
-      nodes = Bebox::Node.nodes_in_environment(project_root, environment, previous_checkpoint(step))
-      in_step_nodes = Bebox::Node.list(project_root, environment, "steps/#{step}")
-      nodes.each do |node|
-        if in_step_nodes.include?(node.hostname)
-          nodes_to_step << node if confirm_node_step?(node, step)
-        else
-          nodes_to_step << node
-        end
-      end
-      nodes_to_step
+    def provision_step_in_node(project_root, environment, step, in_step_nodes, node)
+      title "Provisioning #{step} in node #{node.hostname}:"
+      role = Bebox::Provision.role_from_node(project_root, step, node.hostname)
+      profiles = Bebox::Provision.profiles_from_role(project_root, role) unless role.nil?
+      # Before apply generate the Puppetfile with modules from all associated profiles
+      Bebox::Provision.generate_puppetfile(project_root, step, profiles) unless profiles.nil?
+      # Before apply generate the roles and profiles modules structure for puppet step
+      Bebox::Provision.generate_roles_and_profiles(project_root, step, role, profiles)
+      provision = Bebox::Provision.new(project_root, environment, node, step)
+      provision.apply.success? ? (ok "Node '#{node.hostname}' provisioned to #{step}.") : (error "An error ocurred in the provision of #{step} for node '#{node.hostname}'")
+    end
+
+    def check_node_to_step(node, in_step_nodes, step)
+      return true unless in_step_nodes.include?(node.hostname)
+      message = "The node '#{node.hostname}' was already provisioned in #{step}"
+      message += " (start: #{node.checkpoint_parameter_from_file('steps/' + step, 'started_at')} - end: #{node.checkpoint_parameter_from_file('steps/' + step, 'finished_at')})."
+      message += "\nDo you want to re-provision it?"
+      confirm_action?(message)
+    end
+
+    def nodes_for_provisioning(nodes, step)
+      title "Nodes for provisioning #{step}:"
+      nodes.each{|node| msg(node.hostname)}
+      linebreak
     end
 
     # Obtain the previous checkpoint (step/phase) for a node
@@ -65,16 +60,5 @@ module Bebox
           'steps/step-2'
       end
     end
-
-    # Ask for confirmation of node step
-    def confirm_node_step?(node, step)
-      node_type = "steps/#{step}"
-      checkpoint_status = "(start: #{node.checkpoint_parameter_from_file(node_type, 'started_at')} - end: #{node.checkpoint_parameter_from_file(node_type, 'finished_at')})"
-      quest "The node '#{node.hostname}' was already provisioned in #{step} #{checkpoint_status}.\nDo you want to re-provision it?"
-      response =  ask(highline_quest('(y/n)')) do |q|
-        q.default = "n"
-      end
-      return response == 'y' ? true : false
-    end
   end
 end

data/lib/bebox/wizards/role_wizard.rb

@@ -1,8 +1,9 @@
-require 'bebox/role'
 
 module Bebox
   class RoleWizard
     include Bebox::Logger
+    include Bebox::WizardsHelper
+
     # Create a new role
     def create_new_role(project_root, role_name)
       # Check if the role name is valid
@@ -12,7 +13,7 @@ module Bebox
       \n* Underscores
       \n* Must begin with an Lowercase letter
       \n* Can not be any of: #{Bebox::RESERVED_WORDS.join(', ')}
-      \n\nNo changes were made." unless Bebox::Role.valid_name?(role_name)
+      \n\nNo changes were made." unless valid_puppet_class_name?(role_name)
       # Check if the role exist
       return error("The '#{role_name}' role already exist. No changes were made.") if role_exists?(project_root, role_name)
       # Role creation
@@ -27,12 +28,12 @@ module Bebox
       roles = Bebox::Role.list(project_root)
       # Get a role if exist.
       if roles.count > 0
-        role_name = choose_role(roles, 'Choose the role to remove:')
+        role_name = choose_option(roles, 'Choose the role to remove:')
       else
         return error "There are no roles to remove. No changes were made."
       end
       # Ask for deletion confirmation
-      return warn('No changes were made.') unless confirm_role_deletion?
+      return warn('No changes were made.') unless confirm_action?('Are you sure that you want to delete the role?')
       # Role deletion
       role = Bebox::Role.new(role_name, project_root)
       role.remove
@@ -43,9 +44,8 @@ module Bebox
     def add_profile(project_root)
       roles = Bebox::Role.list(project_root)
       profiles = Bebox::Profile.list(project_root)
-      role = choose_role(roles, 'Choose an existing role:')
-      require 'bebox/wizards/profile_wizard'
-      profile = Bebox::ProfileWizard.new.choose_profile(profiles, 'Choose the profile to add:')
+      role = choose_option(roles, 'Choose an existing role:')
+      profile = choose_option(profiles, 'Choose the profile to add:')
       if Bebox::Role.profile_in_role?(project_root, role, profile)
         return warn("Profile '#{profile}' already in the Role '#{role}'. No changes were made.")
       else
@@ -58,9 +58,8 @@ module Bebox
     def remove_profile(project_root)
       roles = Bebox::Role.list(project_root)
       profiles = Bebox::Profile.list(project_root)
-      role = choose_role(roles, 'Choose an existing role:')
-      require 'bebox/wizards/profile_wizard'
-      profile = Bebox::ProfileWizard.new.choose_profile(profiles, 'Choose the profile to remove:')
+      role = choose_option(roles, 'Choose an existing role:')
+      profile = choose_option(profiles, 'Choose the profile to remove:')
       if Bebox::Role.profile_in_role?(project_root, role, profile)
         Bebox::Role.remove_profile(project_root, role, profile)
         return ok("Profile '#{profile}' removed from Role '#{role}'.")
@@ -70,28 +69,9 @@ module Bebox
 
     end
 
-    # Asks to choose an existing role
-    def choose_role(roles, question)
-      choose do |menu|
-        menu.header = title(question)
-        roles.each do |box|
-          menu.choice(box.split('/').last)
-        end
-      end
-    end
-
     # Check if there's an existing role in the project
     def role_exists?(project_root, role_name)
       Dir.exists?("#{project_root}/puppet/roles/#{role_name}")
     end
-
-    # Ask for confirmation of role deletion
-    def confirm_role_deletion?
-      quest 'Are you sure that you want to delete the role?'
-      response =  ask(highline_quest('(y/n)')) do |q|
-        q.default = "n"
-      end
-      return response == 'y' ? true : false
-    end
   end
 end

data/lib/bebox/wizards/wizards_helper.rb

@@ -0,0 +1,41 @@
+module Bebox
+  module WizardsHelper
+    # Ask for confirmation of any action
+    def confirm_action?(message)
+      require 'highline/import'
+      quest message
+      response =  ask(highline_quest('(y/n)')) do |q|
+        q.default = "n"
+      end
+      return response == 'y' ? true : false
+    end
+
+    # Ask to write some input with validation
+    def write_input(message, default=nil, validator=nil, not_valid_message=nil)
+      require 'highline/import'
+      response =  ask(highline_quest(message)) do |q|
+        q.default = default if default
+        q.validate = /\.(.*)/ if validator
+        q.responses[:not_valid] = highline_warn(not_valid_message) if not_valid_message
+      end
+      return response
+    end
+
+    # Asks to choose an option
+    def choose_option(options, question)
+      require 'highline/import'
+      choose do |menu|
+        menu.header = title(question)
+        options.each do |option|
+          menu.choice(option)
+        end
+      end
+    end
+
+    # Check if the puppet resource has a valid name
+    def valid_puppet_class_name?(name)
+      valid_name = (name =~ /\A[a-z][a-z0-9_]*\Z/).nil? ? false : true
+      valid_name && !Bebox::RESERVED_WORDS.include?(name)
+    end
+  end
+end

data/lib/bebox.rb

@@ -1,2 +1,30 @@
-# Add requires for other files you add to your project here, so
-# you just need to require this one file in your bin file
+require 'progressbar'
+require 'colorize'
+require 'gli'
+require 'bebox/logger'
+require 'bebox/files_helper'
+require 'bebox/vagrant_helper'
+require 'bebox/wizards/wizards_helper'
+require 'bebox/commands/commands_helper'
+require 'bebox/commands/general_commands'
+require 'bebox/commands/project_commands'
+require 'bebox/commands/environment_commands'
+require 'bebox/commands/node_commands'
+require 'bebox/commands/prepare_commands'
+require 'bebox/commands/role_commands'
+require 'bebox/commands/profile_commands'
+require 'bebox/commands/provision_commands'
+require 'bebox/wizards/project_wizard'
+require 'bebox/wizards/environment_wizard'
+require 'bebox/wizards/node_wizard'
+require 'bebox/wizards/role_wizard'
+require 'bebox/wizards/profile_wizard'
+require 'bebox/wizards/provision_wizard'
+require 'bebox/cli'
+require 'bebox/version'
+require 'bebox/project'
+require 'bebox/environment'
+require 'bebox/node'
+require 'bebox/role'
+require 'bebox/profile'
+require 'bebox/provision'

data/lib/templates/project/config/deploy/steps/step-0.erb

@@ -0,0 +1,13 @@
+before 'puppet:bundle_modules', 'puppet:download_modules'
+
+namespace :puppet do
+  task :download_modules do
+    modules_path = "#{current_path}/steps/#{step_dir}/modules"
+    # Download stdlib module
+    run "cd #{modules_path} && wget -qO- https://forgeapi.puppetlabs.com/v3/files/puppetlabs-stdlib-4.2.2.tar.gz | tar xvz && mv puppetlabs-stdlib-4.2.2 stdlib"
+    # Download rbenv module
+    run "cd #{modules_path} && wget -qO- https://forgeapi.puppetlabs.com/v3/files/alup-rbenv-1.2.0.tar.gz | tar xvz && mv alup-rbenv-1.2.0 rbenv"
+    # Download sudo module
+    run "cd #{modules_path} && wget -qO- https://forgeapi.puppetlabs.com/v3/files/saz-sudo-3.0.6.tar.gz | tar xvz && mv saz-sudo-3.0.6 sudo"
+  end
+end

data/lib/templates/project/config/deploy.erb

@@ -24,6 +24,9 @@ def custom_facter_variables
   facter_variables.map{|facter_key, facter_value| "FACTER_#{facter_key}=#{facter_value}"}.join(' ')
 end
 
+# load specific task from each step
+load "config/deploy/steps/#{phase}" if phase =~ /step/
+
 namespace :deploy do
 
 	desc 'Installs tools and required dependencies in the remote machine'

data/spec/fixtures/config/deploy/steps/step-0.test

@@ -0,0 +1,13 @@
+before 'puppet:bundle_modules', 'puppet:download_modules'
+
+namespace :puppet do
+  task :download_modules do
+    modules_path = "#{current_path}/steps/#{step_dir}/modules"
+    # Download stdlib module
+    run "cd #{modules_path} && wget -qO- https://forgeapi.puppetlabs.com/v3/files/puppetlabs-stdlib-4.2.2.tar.gz | tar xvz && mv puppetlabs-stdlib-4.2.2 stdlib"
+    # Download rbenv module
+    run "cd #{modules_path} && wget -qO- https://forgeapi.puppetlabs.com/v3/files/alup-rbenv-1.2.0.tar.gz | tar xvz && mv alup-rbenv-1.2.0 rbenv"
+    # Download sudo module
+    run "cd #{modules_path} && wget -qO- https://forgeapi.puppetlabs.com/v3/files/saz-sudo-3.0.6.tar.gz | tar xvz && mv saz-sudo-3.0.6 sudo"
+  end
+end

data/spec/fixtures/config/deploy.test

@@ -24,6 +24,9 @@ def custom_facter_variables
   facter_variables.map{|facter_key, facter_value| "FACTER_#{facter_key}=#{facter_value}"}.join(' ')
 end
 
+# load specific task from each step
+load "config/deploy/steps/#{phase}" if phase =~ /step/
+
 namespace :deploy do
 
   desc 'Installs tools and required dependencies in the remote machine'

data/spec/ordered_phases_spec.rb

@@ -28,16 +28,18 @@ describe 'Test 99: ordered specs' do
 
   context '99: project destroy' do
 
+    include Bebox::VagrantHelper
+
     let(:project) { build(:project) }
     let(:node) { build(:node) }
 
     it 'should clean spec files' do
       # Test if the vagrant was halt
-      Bebox::Node.halt_vagrant_nodes(node.project_root)
-      expect(node.vagrant_box_running?).to be(false)
+      Bebox::VagrantHelper.halt_vagrant_nodes(node.project_root)
+      expect(vagrant_box_running?(node)).to be(false)
       # Test if the vagrant box was destroyed
-      node.remove_vagrant_box
-      expect(node.vagrant_box_exist?).to be(false)
+      remove_vagrant_box(node)
+      expect(vagrant_box_exist?(node)).to be(false)
       # Test if the project directory was destroyed
       project.destroy
       expect(Dir.exist?("#{project.path}")).to be(false)

data/spec/pre_prepare_spec.rb

@@ -5,6 +5,8 @@ require_relative '../spec/factories/node.rb'
 
 describe 'Test 05: Bebox::Node' do
 
+  include Bebox::VagrantHelper
+
   describe 'Pre-prepare nodes' do
 
     let(:nodes) { 1.times.map{|index| build(:node, :created, hostname: "node#{index}.server1.test")} }
@@ -15,7 +17,7 @@ describe 'Test 05: Bebox::Node' do
 
     context 'pre vagrant prepare' do
       it 'should generate the Vagrantfile' do
-        Bebox::Node.generate_vagrantfile(project_root, nodes)
+        Bebox::VagrantHelper.generate_vagrantfile(nodes)
         vagrantfile_content = File.read("#{project_root}/Vagrantfile").gsub(/\s+/, ' ').strip
         ouput_template = Tilt::ERBTemplate.new('spec/fixtures/node/Vagrantfile.test.erb')
         vagrantfile_output_content = ouput_template.render(nil, ip_address: nodes.first.ip).gsub(/\s+/, ' ').strip
@@ -44,7 +46,7 @@ describe 'Test 05: Bebox::Node' do
         end
 
         it 'should add the hosts config to hosts file' do
-          nodes.each{|node| node.add_to_local_hosts}
+          nodes.each{|node| add_to_local_hosts(node)}
           node = nodes.first
           hosts_content = File.read("#{node.local_hosts_path}/hosts").gsub(/\s+/, ' ').strip
           expect(hosts_content).to include(*nodes.map{|node| "#{node.ip} #{node.hostname}"})
@@ -54,14 +56,14 @@ describe 'Test 05: Bebox::Node' do
       describe 'vagrant setup' do
         it 'should add the node to vagrant' do
           vagrant_box_names_expected = nodes.map{|node| "#{project_name}-#{node.hostname}"}
-          nodes.each{|node| node.add_vagrant_node(project_name, vagrant_box_base)}
+          nodes.each{|node| add_vagrant_node(project_name, vagrant_box_base, node)}
           node = nodes.first
-          expect(node.installed_vagrant_box_names).to include(*vagrant_box_names_expected)
+          expect(installed_vagrant_box_names(node)).to include(*vagrant_box_names_expected)
         end
 
         it 'should up the vagrant boxes' do
-          Bebox::Node.up_vagrant_nodes(project_root)
-          nodes.each{|node| expect(node.vagrant_box_running?).to eq(true)}
+          Bebox::VagrantHelper.up_vagrant_nodes(project_root)
+          nodes.each{|node| expect(vagrant_box_running?(node)).to eq(true)}
         end
 
         it 'should connect to vagrant box through ssh' do

data/spec/profile_spec.rb

@@ -1,8 +1,12 @@
 require 'spec_helper'
 require_relative '../spec/factories/profile.rb'
+require_relative '../lib/bebox/wizards/wizards_helper'
 
 describe 'Test 08: Bebox::Profile' do
 
+  # include Wizard helper methods
+  include Bebox::WizardsHelper
+
   describe 'Manage profiles' do
 
     subject { build(:profile) }
@@ -15,15 +19,15 @@ describe 'Test 08: Bebox::Profile' do
 
       it 'should validate the profile name' do
         # Test not valid reserved words
-        Bebox::RESERVED_WORDS.each{|reserved_word| expect(Bebox::Profile.valid_name?(reserved_word)).to be (false)}
+        Bebox::RESERVED_WORDS.each{|reserved_word| expect(valid_puppet_class_name?(reserved_word)).to be (false)}
         # Test not valid start by undescore
-        expect(Bebox::Profile.valid_name?('_profile_0')).to be (false)
+        expect(valid_puppet_class_name?('_profile_0')).to be (false)
         # Test not valid contain Upper letter
-        expect(Bebox::Profile.valid_name?('Profile_0')).to be (false)
+        expect(valid_puppet_class_name?('Profile_0')).to be (false)
         # Test not valid contain dash character
-        expect(Bebox::Profile.valid_name?('profile-0')).to be (false)
+        expect(valid_puppet_class_name?('profile-0')).to be (false)
         # Test valid name not contains reserved words, start with letter, contains only downcase letters, numbers and undescores
-        expect(Bebox::Profile.valid_name?(subject.name)).to be (true)
+        expect(valid_puppet_class_name?(subject.name)).to be (true)
       end
 
       it 'should clean the profile path' do

data/spec/project_spec.rb

@@ -61,11 +61,18 @@ describe 'Test 01: Bebox::Project' do
         expect(output_file).to eq(expected_content)
       end
 
-      it 'should generate deploy file' do
-        subject.generate_deploy_file
+      it 'should generate deploy files' do
+        subject.generate_deploy_files
+        # Generate deploy.rb file
         config_deploy_content = File.read("#{subject.path}/config/deploy.rb").gsub(/\s+/, ' ').strip
         config_deploy_output_content = File.read("spec/fixtures/config/deploy.test").gsub(/\s+/, ' ').strip
         expect(config_deploy_content).to eq(config_deploy_output_content)
+        # Generate steps/step-N.rb files
+        Bebox::PROVISION_STEPS.each do |step|
+          content = File.read("spec/fixtures/config/deploy/steps/#{step}.test")
+          output = File.read("#{subject.path}/config/deploy/steps/#{step}.rb")
+          expect(output).to eq(content)
+        end
       end
 
       it 'should create Gemfile' do