lib/foodtaster/rspec/dsl_methods.rb in foodtaster-0.0.7 vs lib/foodtaster/rspec/dsl_methods.rb in foodtaster-0.0.8

@@ -1,22 +1,19 @@
 module Foodtaster
   module RSpec
     module DslMethods
       def require_vm(vm_name)
-        let(vm_name) do
-          get_vm(vm_name)
-        end
-
+        define_method(vm_name) { get_vm(vm_name) }
         before(:all) { 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]
+        rollback = options.key?(:rollback) ? options[:rollback] : !Foodtaster.config.skip_rollback
 
         before(:all) do
           vm = get_vm(vm_name)
-          vm.rollback unless skip_rollback
+          vm.rollback if rollback
 
           run_chef_on(vm_name, &block)
         end
       end
     end