lib/ammeter/rspec/generator/example.rb in ammeter-0.2.9 vs lib/ammeter/rspec/generator/example.rb in ammeter-1.0.0

@@ -1,8 +1,24 @@
 require 'rspec/core'
 require 'ammeter/rspec/generator/example/generator_example_group'
 
 RSpec::configure do |c|
-  c.include Ammeter::RSpec::Rails::GeneratorExampleGroup, :type => :generator, :example_group => {
-    :file_path => /spec[\\\/]generators/
-  }
+  def c.escaped_path(*parts)
+    Regexp.compile(parts.join('[\\\/]') + '[\\\/]')
+  end
+
+  generator_path_regex = c.escaped_path(%w[spec generators])
+  if RSpec::Core::Version::STRING >= '3'
+    c.include Ammeter::RSpec::Rails::GeneratorExampleGroup,
+      :type          => :generator,
+      :file_path     => lambda { |file_path, metadata|
+        metadata[:type].nil? && generator_path_regex =~ file_path
+      }
+
+  else #rspec2
+
+    c.include Ammeter::RSpec::Rails::GeneratorExampleGroup, :type => :generator, :example_group => {
+      :file_path => generator_path_regex
+    }
+  end
+
 end