Merge pull request #25 from LRDesign/handling_modules_in_step_lists

Adding ability to include modules in step lists

Author: TablesawTablesawsen

lib/rspec-steps/builder.rb

@@ -8,6 +8,9 @@ def build_example_group
       describer = @describer
 
       RSpec.describe(*describer.group_args, describer.metadata) do
+        describer.modules.each do |mod|
+          mod.apply(self)
+        end
         describer.let_list.each do |letter|
           letter.define_on(describer.step_list, self)
         end

lib/rspec-steps/describer.rb

@@ -3,9 +3,9 @@
 require 'rspec-steps/hook'
 require 'rspec-steps/step-list'
 require 'rspec-steps/lets'
+require 'rspec-steps/modules'
 
 module RSpec::Steps
-
   class Describer
     def initialize(args, metadata, &block)
       @group_args = args
@@ -17,9 +17,10 @@ def initialize(args, metadata, &block)
       @step_list = StepList.new
       @hooks = []
       @let_list = []
+      @modules = []
       instance_eval(&block)
     end
-    attr_reader  :group_args, :let_list, :step_list, :hooks, :metadata
+    attr_reader  :group_args, :modules, :let_list, :step_list, :hooks, :metadata
 
     def step(*args, &action)
       metadata = {}
@@ -45,8 +46,18 @@ def shared_steps(*args, &block)
       SharedSteps[name] = Describer.new(args, {:caller => caller}, &block)
     end
 
+    def include(mod)
+      @modules << ModuleInclusion.new(mod)
+    end
+
+    def extend(mod)
+      @modules << ModuleExtension.new(mod)
+    end
+
     def perform_steps(name)
       describer = SharedSteps.fetch(name)
+      @modules += describer.modules
+      @let_list += describer.let_list
       @hooks += describer.hooks
       @step_list += describer.step_list
     end

lib/rspec-steps/modules.rb

@@ -0,0 +1,23 @@
+module RSpec::Steps
+  class ModuleExtension
+    def initialize(mod)
+      @mod = mod
+    end
+
+    def apply(target)
+      mod = @mod
+      target.instance_eval { extend mod }
+    end
+  end
+
+  class ModuleInclusion
+    def initialize(mod)
+      @mod = mod
+    end
+
+    def apply(target)
+      mod = @mod
+      target.instance_eval { include mod }
+    end
+  end
+end

rspec-steps.gemspec

@@ -38,6 +38,7 @@ Gem::Specification.new do |spec|
   lib/rspec-steps/hook.rb
   lib/rspec-steps/step-result.rb
   lib/rspec-steps/lets.rb
+  lib/rspec-steps/modules.rb
   lib/rspec-steps/monkeypatching.rb
   doc/README
   doc/Specifications

spec/example_group_spec.rb

@@ -19,6 +19,29 @@
       end
     end
 
+    it "should handle inclusion and extension with modules" do
+      module Testing
+        def something
+          42
+        end
+      end
+
+      group = nil
+      sandboxed do
+        group = RSpec.steps "Test Steps" do
+          include Testing
+          extend Testing
+
+          it("accesses a method from a module"){ expect(something).to eq 42 }
+        end
+        group.run
+      end
+
+      group.examples.each do |example|
+        expect(example.metadata[:execution_result].status).to eq(:passed)
+      end
+    end
+
     it "should define let blocks correctly" do
       group = nil
       sandboxed do