1begin
2 require 'rubygems'
3 require 'heckle'
4rescue LoadError ; raise "You must gem install heckle to use --heckle" ; end
5
6module Spec
7 module Runner
8 # Creates a new Heckler configured to heckle all methods in the classes
9 # whose name matches +filter+
10 class HeckleRunner
11 def initialize(filter, heckle_class=Heckler)
12 @filter = filter
13 @heckle_class = heckle_class
14 end
15
16 # Runs all the example groups held by +rspec_options+ once for each of the
17 # methods in the matched classes.
18 def heckle_with
19 if @filter =~ /(.*)[#\.](.*)/
20 heckle_method($1, $2)
21 else
22 heckle_class_or_module(@filter)
23 end
24 end
25
26 def heckle_method(class_name, method_name)
27 verify_constant(class_name)
28 heckle = @heckle_class.new(class_name, method_name, rspec_options)
29 heckle.validate
30 end
31
32 def heckle_class_or_module(class_or_module_name)
33 verify_constant(class_or_module_name)
34 pattern = /^#{class_or_module_name}/
35 classes = []
36 ObjectSpace.each_object(Class) do |klass|
37 classes << klass if klass.name =~ pattern
38 end
39
40 classes.each do |klass|
41 klass.instance_methods(false).each do |method_name|
42 heckle = @heckle_class.new(klass.name, method_name, rspec_options)
43 heckle.validate
44 end
45 end
46 end
47
48 def verify_constant(name)
49 begin
50 # This is defined in Heckle
51 name.to_class
52 rescue
53 raise "Heckling failed - \"#{name}\" is not a known class or module"
54 end
55 end
56 end
57
58 #Supports Heckle 1.2 and prior (earlier versions used Heckle::Base)
59 class Heckler < (Heckle.const_defined?(:Base) ? Heckle::Base : Heckle)
60 def initialize(klass_name, method_name, rspec_options)
61 super(klass_name, method_name)
62 @rspec_options = rspec_options
63 end
64
65 def tests_pass?
66 success = @rspec_options.run_examples
67 success
68 end
69
70 end
71 end
72end