lib/foodtaster/rspec/dsl_methods.rb in foodtaster-0.0.1 vs lib/foodtaster/rspec/dsl_methods.rb in foodtaster-0.0.3

@@ -1,19 +1,23 @@
 module Foodtaster
   module RSpec
     module DslMethods
-      def run_chef_on(vm_name, &block)
+      def require_vm(vm_name)
         Foodtaster::RSpecRun.current.require_vm(vm_name)
 
-        skip_rollback = true
+        let(vm_name) { get_vm(vm_name) }
+      end
 
+      def run_chef_on(vm_name, options = {}, &block)
+        require_vm(vm_name)
+        skip_rollback = Foodtaster.config.skip_rollback || options[:skip_rollback]
+
         before(:all) do
           vm = get_vm(vm_name)
           vm.rollback unless skip_rollback
+
           run_chef_on(vm_name, &block)
         end
-
-        let(vm_name) { get_vm(vm_name) }
       end
     end
   end
 end