[dm-constraints] added a method to wrap changed method visibility tasks

Signed-off-by: Balint Erdi <balint.erdi@gmail.com>

dm-constraints/lib/dm-constraints.rb

@@ -24,9 +24,9 @@ class Relationship
       # initialize is a private method in Relationship
       # and private methods can not be "advised" (hooked into)
       # in extlib.
-      public(:initialize)
-      before :initialize, :add_delete_constraint_option
-      private(:initialize)
+      with_changed_method_visibility(:initialize, :private, :public) do
+        before :initialize, :add_delete_constraint_option
+      end
     end
   end
 end

dm-constraints/lib/dm-constraints/delete_constraint.rb

@@ -16,6 +16,14 @@ def check_delete_constraint_type(cardinality, name, options = {})
             raise ArgumentError, ":constraint option must be one of #{delete_constraint_options * ', '}"
           end
         end
+
+        # TODO: that should be moved to a 'util-like' module
+        def with_changed_method_visibility(method, from_visibility, to_visibility, &block)
+          send(to_visibility, method)
+          yield
+          send(from_visibility, method)
+        end
+
       end
 
       def add_delete_constraint_option(name, repository_name, child_model, parent_model, options = {})