lib/lightio/library/base.rb in lightio-0.4.2 vs lib/lightio/library/base.rb in lightio-0.4.3

@@ -96,20 +96,27 @@
         if obj.instance_of? self
           obj
         else
           mock_obj = allocate
           mock_obj.instance_variable_set(:@obj, obj)
+          mock_obj.__send__(:call_lightio_initialize)
           mock_obj
         end
       end
     end
 
     def initialize(*args)
       obj = self.class.send(:call_method_from_ancestors, :mock_klass).send(:origin_new, *args)
       @obj = obj
+      call_lightio_initialize
+      @obj
     end
 
     private
+    def call_lightio_initialize
+      __send__(:lightio_initialize) if respond_to?(:lightio_initialize, true)
+    end
+
     def light_io_raw_obj
       @obj
     end
 
     class << self
\ No newline at end of file