lib/infrataster/rspec.rb in infrataster-0.1.13 vs lib/infrataster/rspec.rb in infrataster-0.2.0.beta1

@@ -7,14 +7,22 @@
   config.include Infrataster::Helpers::RSpecHelper
 
   config.before(:all) do
     @infrataster_context = Infrataster::Contexts.from_example(self.class)
   end
+
   config.before(:each) do
+    if defined?(RSpec.current_example)
+      example = RSpec.current_example
+    end
     @infrataster_context = Infrataster::Contexts.from_example(example)
     @infrataster_context.before_each(example) if @infrataster_context.respond_to?(:before_each)
   end
+
   config.after(:each) do
+    if defined?(RSpec.current_example)
+      example = RSpec.current_example
+    end
     @infrataster_context.after_each(example) if @infrataster_context.respond_to?(:after_each)
   end
 
   config.after(:all) do
     Infrataster::Server.defined_servers.each do |server|