test/example_wrapper_test.rb in surrounded-0.2.0 vs test/example_wrapper_test.rb in surrounded-0.2.1

@@ -1,16 +1,14 @@
 require 'test_helper'
 
 # If you want to use wrappers, here's how you could
-require 'delegate'
 class WrapperContext
   extend Surrounded::Context
 
   apply_roles_on(:trigger)
-  setup(:admin, :task)
+  initialize(:admin, :task)
 
-  class Admin < SimpleDelegator
-    include Surrounded
+  wrap :admin do
     def some_admin_method
       'hello from the admin wrapper!'
     end
   end
 
\ No newline at end of file