chantek.rb in orangutan-0.0.5

- old
+ new
@@ -1,51 +1,87 @@
-require 'orangutan/stub_base'
-require 'orangutan/expectation'
-require 'orangutan/call'
-
-module Orangutan 
-  class Chantek
-    attr_reader :calls, :stubs
-  
-    def initialize
-      @calls = []
-      @expectations = {}
-      @stubs= {}
-    end
-  
-    def stub name, params={}
-      return @stubs[name] if @stubs[name]
-      c = Class.new(StubBase) do
-        if params[:clr_interface]
-          include params[:clr_interface]
-          params[:clr_interface].to_clr_type.get_methods.each do |m_info|
-            snake = m_info.name.scan(/[A-Z][a-z0-9]*/).map {|a|a.downcase}.join('_').to_sym
-            define_method snake do |*args|
-              yield_container, return_container = __react__(snake, args)
-              yield *yield_container.value if yield_container && block_given?
-              return __return__(method, return_container)
-            end
-          end
-        end
-      end
-      @stubs[name] = c.new(name, self, params[:recursive])
-    end
-  
-    def when name
-      expectations_for_name = @expectations[name]
-      @expectations[name] = expectations_for_name = [] unless expectations_for_name
-      expectation = Orangutan::Expectation.new
-      expectations_for_name << expectation
-      expectation
-    end
-  
-    def first_match name, method, args
-      expectations_for_name = @expectations[name]
-      if expectations_for_name
-        expectations_for_name.each do |expectation|
-            return expectation if expectation.matches?(method, *args)
-        end
-      end
-      nil
-    end
-  end
+require 'orangutan/stub_base'
+require 'orangutan/expectation'
+require 'orangutan/call'
+require 'orangutan/reflector'
+
+module Orangutan 
+  module Chantek
+    attr_reader :calls, :stubs, :expectations
+
+    def call name, method, *args
+      Call.new(name, method, args)
+    end
+
+    def reset_stubs
+      @calls = []
+      @expectations = {}
+      @stubs= {}
+      @events = {}
+    end
+
+    def stub name, params={}
+      return @stubs[name] if @stubs[name]
+      clazz = Class.new(StubBase)
+      @events[name] = {}
+      implement_interface clazz, params[:clr_interface], @events[name]
+      @stubs[name] = clazz.new(name, self, params[:recursive])
+    end
+
+    def implement_interface clazz, interface, events
+      return unless interface
+      reflector = Reflector.new(interface)
+      clazz.instance_eval { include interface }
+      reflector.methods.each {|method| implement_method clazz, method }
+      reflector.events.each do |event|
+        events[event] = []
+        implement_event clazz, event, events[event]
+      end
+    end
+
+    def implement_method clazz, name
+      clazz.instance_eval do
+        define_method name do |*args|
+          yield_container, return_container = __react__(name, args)
+          yield_container.value.each {|v| yield *v } if yield_container && block_given?
+          return __return__(name, return_container)
+        end
+      end
+    end
+
+    def implement_event clazz, name, delegates
+      method = "add_#{name}".to_sym
+      clazz.instance_eval do
+        define_method method  do |delegate|
+          __react__(method, [])
+          delegates << delegate
+        end
+      end
+    end
+
+    def so_when name
+      expectations_for_name = @expectations[name]
+      @expectations[name] = expectations_for_name = [] unless expectations_for_name
+      expectation = Orangutan::Expectation.new
+      expectations_for_name << expectation
+      expectation
+    end
+  
+    def first_match name, method, args
+      expectations_for_name = @expectations[name]
+      if expectations_for_name
+        expectations_for_name.each do |expectation|
+            return expectation if expectation.matches?(method, *args)
+        end
+      end
+      nil
+    end
+
+    def fire_event name, event, *args
+      events_for_name = @events[name]
+      raise "failed to find any events for #{name}" unless events_for_name
+      # it makes no sense, but 'events_for_name[event]' returns nil so in desperation, we iterate over the events
+      events_for_name.each do |event_name,delegates|
+        delegates.each { |delegate| delegate.invoke *args } if event_name == event
+      end
+    end
+  end
 end
\ No newline at end of file