From 30aa719082009fd47e26ceb7f8f8eff2023b4f34 Mon Sep 17 00:00:00 2001
From: Katsuyoshi Ito <kito@itosoft.com>
Date: Sun, 30 Jun 2013 00:29:48 +0900
Subject: [PATCH] to setup controller using block.

I implemented @colinta 's idea which was discussed below link.
https://groups.google.com/d/msg/rubymotion/M9T6LybaEUs/AWip_Vwy5_4J

When you want to setup UIViewController, you can setup it in a block of tests method.

# Example
  describe MyViewController do

    tests MyViewController do |controller|
      # you can set data to the controller here
      controller.my_data = "foo"
    end

    .
    .
  end
---
 lib/motion/project/template/ios/spec-helpers/ui.rb | 10 +++++++---
 1 file changed, 7 insertions(+), 3 deletions(-)

diff --git a/lib/motion/project/template/ios/spec-helpers/ui.rb b/lib/motion/project/template/ios/spec-helpers/ui.rb
index 90a99a65..df4880da 100644
--- a/lib/motion/project/template/ios/spec-helpers/ui.rb
+++ b/lib/motion/project/template/ios/spec-helpers/ui.rb
@@ -167,8 +167,8 @@ module Bacon
     # TODO
     # * :navigation => true
     # * :tab => true
-    def tests(controller_class, options = {})
-      @controller_class, @options = controller_class, options
+    def tests(controller_class, options = {}, &block)
+      @controller_class, @options, @after_created_block = controller_class, options, block
       extend Bacon::Functional::API
       extend Bacon::Functional::ContextExt
     end
@@ -220,7 +220,11 @@ module Bacon
           else
             c = @controller_class.alloc.init
           end
-          send(@options[:after_created], c) if @options[:after_created]
+          if @after_created_block
+            @after_created_block.call(c)
+          else
+            send(@options[:after_created], c) if @options[:after_created]
+          end
           c
         end
       end