lib/bebox/node.rb in bebox-0.1.4 vs lib/bebox/node.rb in bebox-0.1.5

@@ -72,10 +72,11 @@
 
     # Create the checkpoints for the prepared nodes
     def create_prepare_checkpoint(started_at)
       self.started_at = started_at
       self.finished_at = DateTime.now.to_s
+      Bebox::Environment.create_checkpoint_directories(project_root, environment)
       generate_file_from_template("#{Bebox::FilesHelper::templates_path}/node/prepared_node.yml.erb", "#{self.project_root}/.checkpoints/environments/#{self.environment}/prepared_nodes/#{self.hostname}.yml", {node: self})
     end
 
     # Create the puppet hiera template file
     def create_hiera_template
@@ -103,9 +104,10 @@
     # Create checkpoint for node
     def create_node_checkpoint
       # Set the creation time for the node
       self.created_at = DateTime.now.to_s
       # Create the checkpoint file from template
+      Bebox::Environment.create_checkpoint_directories(project_root, environment)
       generate_file_from_template("#{Bebox::FilesHelper::templates_path}/node/node.yml.erb", "#{project_root}/.checkpoints/environments/#{environment}/nodes/#{hostname}.yml", {node: self})
     end
 
     # Remove checkpoints for node
     def remove_checkpoints
\ No newline at end of file