lib/neo4j/migration.rb in neo4j-3.0.1 vs lib/neo4j/migration.rb in neo4j-3.0.2

@@ -11,15 +11,23 @@
 
     def print_output(string)
       print string unless !!ENV['silenced']
     end
 
+    def default_path
+      Rails.root if defined? Rails
+    end
+
+    def joined_path(path)
+      File.join(path, 'db', 'neo4j-migrate')
+    end
+
     class AddIdProperty < Neo4j::Migration
       attr_reader :models_filename
 
-      def initialize
-        @models_filename = File.join(Rails.root.join('db', 'neo4j-migrate'), 'add_id_property.yml')
+      def initialize(path = default_path)
+        @models_filename = File.join(joined_path(path), 'add_id_property.yml')
       end
 
       def migrate
         models = ActiveSupport::HashWithIndifferentAccess.new(YAML.load_file(models_filename))[:models]
         output "This task will add an ID Property every node in the given file."
@@ -107,12 +115,12 @@
       end
     end
 
     class AddClassnames < Neo4j::Migration
 
-      def initialize
+      def initialize(path = default_path)
         @classnames_filename = 'add_classnames.yml'
-        @classnames_filepath = File.join(Rails.root.join('db', 'neo4j-migrate'), classnames_filename)
+        @classnames_filepath = File.join(joined_path(path), classnames_filename)
       end
 
       def migrate
         output "Adding classnames. This make take some time."
         execute(true)