lib/testlab/provisioners/apt.rb in testlab-1.0.0 vs lib/testlab/provisioners/apt.rb in testlab-1.0.1

@@ -27,11 +27,16 @@
       #   provision.
       # @return [Boolean] True if successful.
       def on_container_provision(container)
         @ui.logger.debug { "APT Provisioner: Container #{container.id}" }
 
-        bootstrap_template = File.join(TestLab::Provisioner.template_dir, "apt", "bootstrap.erb")
-        container.ssh.bootstrap(ZTK::Template.render(bootstrap_template, @config))
+        container.bootstrap(ZTK::Template.render(provision_template, @config))
+      end
+
+    private
+
+      def provision_template
+        File.join(TestLab::Provisioner.template_dir, 'apt', 'provision.erb')
       end
 
     end
 
   end