activegraph 10.0.0.pre.alpha.11 → 10.0.0.pre.beta.1

data/lib/{neo4j → active_graph}/shared/filtered_hash.rb

@@ -1,6 +1,6 @@
-module Neo4j::Shared
+module ActiveGraph::Shared
   class FilteredHash
-    class InvalidHashFilterType < Neo4j::Error; end
+    class InvalidHashFilterType < ActiveGraph::Error; end
     VALID_SYMBOL_INSTRUCTIONS = [:all, :none]
     VALID_HASH_INSTRUCTIONS = [:on]
     VALID_INSTRUCTIONS_TYPES = [Hash, Symbol]

data/lib/{neo4j → active_graph}/shared/identity.rb

@@ -1,4 +1,4 @@
-module Neo4j
+module ActiveGraph
   module Shared
     module Identity
       def ==(other)
@@ -21,7 +21,7 @@ module Neo4j
         if self.class.id_property_name
           send(self.class.id_property_name)
         else
-          # ActiveRel
+          # Relationship
           neo_id
         end
       end

data/lib/{neo4j → active_graph}/shared/initialize.rb

@@ -1,8 +1,8 @@
-module Neo4j::Shared
+module ActiveGraph::Shared
   module Initialize
     extend ActiveSupport::Concern
 
-    # Implements the Neo4j::Node#wrapper and Neo4j::Relationship#wrapper method
+    # Implements the ActiveGraph::Node#wrapper and ActiveGraph::Relationship#wrapper method
     # so that we don't have to care if the node is wrapped or not.
     # @return self
     def wrapper
@@ -12,7 +12,7 @@ module Neo4j::Shared
     private
 
     def convert_and_assign_attributes(properties)
-      @attributes ||= Neo4j::AttributeSet.new(self.class.attributes_nil_hash, self.class.attributes.keys)
+      @attributes ||= ActiveGraph::AttributeSet.new(self.class.attributes_nil_hash, self.class.attributes.keys)
       stringify_attributes!(@attributes, properties)
       self.default_properties = properties if respond_to?(:default_properties=)
       self.class.declared_properties.convert_properties_to(self, :ruby, @attributes)

data/lib/{neo4j → active_graph}/shared/marshal.rb

@@ -1,4 +1,4 @@
-module Neo4j
+module ActiveGraph
   module Shared
     module Marshal
       extend ActiveSupport::Concern

data/lib/{neo4j → active_graph}/shared/mass_assignment.rb

@@ -1,4 +1,4 @@
-module Neo4j::Shared
+module ActiveGraph::Shared
   # MassAssignment allows you to bulk set and update attributes
   #
   # Including MassAssignment into your model gives it a set of mass assignment
@@ -6,7 +6,7 @@ module Neo4j::Shared
   #
   # @example Usage
   #   class Person
-  #     include Neo4j::Shared::MassAssignment
+  #     include ActiveGraph::Shared::MassAssignment
   #   end
   #
   # Originally part of ActiveAttr, https://github.com/cgriego/active_attr

data/lib/{neo4j → active_graph}/shared/permitted_attributes.rb

@@ -1,4 +1,4 @@
-module Neo4j::Shared
+module ActiveGraph::Shared
   module PermittedAttributes
     extend ActiveSupport::Concern
     include ActiveModel::ForbiddenAttributesProtection

data/lib/{neo4j → active_graph}/shared/persistence.rb

@@ -1,4 +1,4 @@
-module Neo4j::Shared
+module ActiveGraph::Shared
   # rubocop:disable Metrics/ModuleLength
   module Persistence
     # rubocop:enable Metrics/ModuleLength
@@ -89,7 +89,7 @@ module Neo4j::Shared
       @_create_or_updating = true
       apply_default_values
       result = _persisted_obj ? update_model : create_model
-      current_transaction = Neo4j::ActiveBase.current_transaction
+      current_transaction = ActiveGraph::Base.current_transaction
 
       current_transaction.mark_failed if result == false && current_transaction
 
@@ -196,7 +196,7 @@ module Neo4j::Shared
     alias update_column update_db_property
 
     def update_db_properties(hash)
-      fail ::Neo4j::Error, 'can not update on a new record object' unless persisted?
+      fail ::ActiveGraph::Error, 'can not update on a new record object' unless persisted?
       self.class.run_transaction do
         db_values = props_for_db(hash)
         neo4j_query(query_as(:n).set(n: db_values))
@@ -229,7 +229,7 @@ module Neo4j::Shared
 
     module ClassMethods
       def run_transaction(run_in_tx = true)
-        Neo4j::ActiveBase.run_transaction(run_in_tx) { |tx| yield tx }
+        ActiveGraph::Base.run_transaction(run_in_tx) { |tx| yield tx }
       end
     end
 

data/lib/{neo4j → active_graph}/shared/property.rb

@@ -1,13 +1,13 @@
-module Neo4j::Shared
+module ActiveGraph::Shared
   module Property
     extend ActiveSupport::Concern
 
-    include Neo4j::Shared::MassAssignment
-    include Neo4j::Shared::TypecastedAttributes
+    include ActiveGraph::Shared::MassAssignment
+    include ActiveGraph::Shared::TypecastedAttributes
     include ActiveModel::Dirty
 
-    class UndefinedPropertyError < Neo4j::Error; end
-    class MultiparameterAssignmentError < Neo4j::Error; end
+    class UndefinedPropertyError < ActiveGraph::Error; end
+    class MultiparameterAssignmentError < ActiveGraph::Error; end
 
     attr_reader :_persisted_obj
 
@@ -24,15 +24,15 @@ module Neo4j::Shared
 
     def inspect
       attribute_descriptions = inspect_attributes.map do |key, value|
-        "#{Neo4j::ANSI::CYAN}#{key}: #{Neo4j::ANSI::CLEAR}#{value.inspect}"
+        "#{ActiveGraph::ANSI::CYAN}#{key}: #{ActiveGraph::ANSI::CLEAR}#{value.inspect}"
       end.join(', ')
 
       separator = ' ' unless attribute_descriptions.empty?
-      "#<#{Neo4j::ANSI::YELLOW}#{self.class.name}#{Neo4j::ANSI::CLEAR}#{separator}#{attribute_descriptions}>"
+      "#<#{ActiveGraph::ANSI::YELLOW}#{self.class.name}#{ActiveGraph::ANSI::CLEAR}#{separator}#{attribute_descriptions}>"
     end
 
     def initialize(attributes = nil)
-      @attributes ||= Neo4j::AttributeSet.new({}, self.class.attributes.keys)
+      @attributes ||= ActiveGraph::AttributeSet.new({}, self.class.attributes.keys)
       attributes = process_attributes(attributes)
       modded_attributes = inject_defaults!(attributes)
       validate_attributes!(modded_attributes)
@@ -79,7 +79,7 @@ module Neo4j::Shared
     def extract_writer_methods!(attributes)
       return attributes if attributes.blank?
       {}.tap do |writer_method_props|
-        attributes.each_key do |key|
+        attributes.keys.each do |key|
           writer_method_props[key] = attributes.delete(key) if self.respond_to?("#{key}=")
         end
       end
@@ -166,7 +166,7 @@ module Neo4j::Shared
       end
 
       # @param [Symbol] name The property name
-      # @param [Neo4j::Shared::AttributeDefinition] attr_def A cloned AttributeDefinition to reuse
+      # @param [ActiveGraph::Shared::AttributeDefinition] attr_def A cloned AttributeDefinition to reuse
       # @param [Hash] options An options hash to use in the new property definition
       def inherit_property(name, attr_def, options = {})
         build_property(name, options) do |prop_name|

data/lib/{neo4j → active_graph}/shared/query_factory.rb

@@ -1,5 +1,5 @@
-module Neo4j::Shared
-  # Acts as a bridge between the node and rel models and Neo4j::Core::Query.
+module ActiveGraph::Shared
+  # Acts as a bridge between the node and rel models and ActiveGraph::Core::Query.
   # If the object is persisted, it returns a query matching; otherwise, it returns a query creating it.
   # This class does not execute queries, so it keeps no record of what identifiers have been set or what has happened in previous factories.
   class QueryFactory
@@ -29,14 +29,14 @@ module Neo4j::Shared
       graph_object.persisted? ? match_query : create_query
     end
 
-    # @param [Neo4j::Core::Query] query An instance of Neo4j::Core::Query upon which methods will be chained.
+    # @param [ActiveGraph::Core::Query] query An instance of ActiveGraph::Core::Query upon which methods will be chained.
     def base_query=(query)
       return if query.blank?
       @base_query = query
     end
 
     def base_query
-      @base_query || Neo4j::ActiveBase.new_query
+      @base_query || ActiveGraph::Base.new_query
     end
 
     protected
@@ -92,7 +92,7 @@ module Neo4j::Shared
     private
 
     def filtered_props
-      Neo4j::Shared::FilteredHash.new(graph_object.props_for_create, graph_object.creates_unique_option).filtered_base
+      ActiveGraph::Shared::FilteredHash.new(graph_object.props_for_create, graph_object.creates_unique_option).filtered_base
     end
 
     def query_string(create_props)

data/lib/{neo4j → active_graph}/shared/rel_type_converters.rb

@@ -1,14 +1,14 @@
-module Neo4j::Shared
-  # This module controls changes to relationship type based on Neo4j::Config.transform_rel_type.
-  # It's used whenever a rel type is automatically determined based on ActiveRel model name or
+module ActiveGraph::Shared
+  # This module controls changes to relationship type based on ActiveGraph::Config.transform_rel_type.
+  # It's used whenever a rel type is automatically determined based on Relationship model name or
   # association type.
   module RelTypeConverters
     def decorated_rel_type(type)
-      @decorated_rel_type ||= Neo4j::Shared::RelTypeConverters.decorated_rel_type(type)
+      @decorated_rel_type ||= ActiveGraph::Shared::RelTypeConverters.decorated_rel_type(type)
     end
 
     class << self
-      # Determines how relationship types should look when inferred based on association or ActiveRel model name.
+      # Determines how relationship types should look when inferred based on association or Relationship model name.
       # With the exception of `:none`, all options will call `underscore`, so `ThisClass` becomes `this_class`, with capitalization
       # determined by the specific option passed.
       # Valid options:
@@ -17,7 +17,7 @@ module Neo4j::Shared
       # * :legacy - downcases and prepends `#`, so ThisClass becomes `#this_class`
       # * :none - uses the string version of whatever is passed with no modifications
       def rel_transformer
-        @rel_transformer ||= Neo4j::Config[:transform_rel_type].nil? ? :upcase : Neo4j::Config[:transform_rel_type]
+        @rel_transformer ||= ActiveGraph::Config[:transform_rel_type].nil? ? :upcase : ActiveGraph::Config[:transform_rel_type]
       end
 
       # @param [String,Symbol] type The raw string or symbol to be used as the basis of the relationship type

data/lib/{neo4j → active_graph}/shared/serialized_properties.rb

@@ -1,4 +1,4 @@
-module Neo4j::Shared
+module ActiveGraph::Shared
   # This module adds the `serialize` class method. It lets you store hashes and arrays in Neo4j properties.
   # Be aware that you won't be able to search within serialized properties and stuff use indexes. If you do a regex search for portion of a string
   # property, the search happens in Cypher and you may take a performance hit.

data/lib/{neo4j → active_graph}/shared/type_converters.rb

@@ -4,7 +4,7 @@ require 'bigdecimal/util'
 require 'active_support/core_ext/big_decimal/conversions'
 require 'active_support/core_ext/string/conversions'
 
-module Neo4j::Shared
+module ActiveGraph::Shared
   class Boolean; end
 
   module TypeConverters
@@ -100,8 +100,6 @@ module Neo4j::Shared
 
         def to_ruby(value)
           value.respond_to?(:to_date) ? value.to_date : Time.at(value).utc.to_date
-        rescue Exception => e
-          raise e
         end
 
         alias to_db to_ruby
@@ -179,7 +177,7 @@ module Neo4j::Shared
         end
 
         def db_type
-          Neo4j::Shared::Boolean
+          ActiveGraph::Shared::Boolean
         end
 
         alias convert_type db_type
@@ -253,7 +251,7 @@ module Neo4j::Shared
 
         return unless @options[:case_sensitive].nil?
 
-        @options[:case_sensitive] = Neo4j::Config.enums_case_sensitive
+        @options[:case_sensitive] = ActiveGraph::Config.enums_case_sensitive
       end
 
       def converted?(value)
@@ -283,10 +281,10 @@ module Neo4j::Shared
           value.map(&method(:to_db))
         elsif @options[:case_sensitive]
           @enum_keys[value.to_s.to_sym] ||
-            fail(Neo4j::Shared::Enum::InvalidEnumValueError, 'Value passed to an enum property must match one of the enum keys')
+            fail(ActiveGraph::Shared::Enum::InvalidEnumValueError, 'Value passed to an enum property must match one of the enum keys')
         else
           @enum_keys[value.to_s.downcase.to_sym] ||
-            fail(Neo4j::Shared::Enum::InvalidEnumValueError, 'Case-insensitive (downcased) value passed to an enum property must match one of the enum keys')
+            fail(ActiveGraph::Shared::Enum::InvalidEnumValueError, 'Case-insensitive (downcased) value passed to an enum property must match one of the enum keys')
         end
       end
     end
@@ -304,7 +302,7 @@ module Neo4j::Shared
     end
 
     # Modifies a hash's values to be of types acceptable to Neo4j or matching what the user defined using `type` in property definitions.
-    # @param [Neo4j::Shared::Property] obj A node or rel that mixes in the Property module
+    # @param [ActiveGraph::Shared::Property] obj A node or rel that mixes in the Property module
     # @param [Symbol] medium Indicates the type of conversion to perform.
     # @param [Hash] properties A hash of symbol-keyed properties for conversion.
     def convert_properties_to(obj, medium, properties)
@@ -313,7 +311,7 @@ module Neo4j::Shared
         next if skip_conversion?(obj, key, value)
 
         converted_value = convert_property(key, value, direction)
-        if properties.is_a?(Neo4j::AttributeSet)
+        if properties.is_a?(ActiveGraph::AttributeSet)
           properties.write_cast_value(key, converted_value)
         else
           properties[key] = converted_value
@@ -335,11 +333,11 @@ module Neo4j::Shared
     end
 
     def typecaster_for(value)
-      Neo4j::Shared::TypeConverters.typecaster_for(value)
+      ActiveGraph::Shared::TypeConverters.typecaster_for(value)
     end
 
     def typecast_attribute(typecaster, value)
-      Neo4j::Shared::TypeConverters.typecast_attribute(typecaster, value)
+      ActiveGraph::Shared::TypeConverters.typecast_attribute(typecaster, value)
     end
 
     private
@@ -368,8 +366,8 @@ module Neo4j::Shared
 
     class << self
       def included(_)
-        Neo4j::Shared::TypeConverters.constants.each do |constant_name|
-          constant = Neo4j::Shared::TypeConverters.const_get(constant_name)
+        ActiveGraph::Shared::TypeConverters.constants.each do |constant_name|
+          constant = ActiveGraph::Shared::TypeConverters.const_get(constant_name)
           register_converter(constant) if constant.respond_to?(:convert_type)
         end
       end

data/lib/{neo4j → active_graph}/shared/typecasted_attributes.rb

@@ -1,4 +1,4 @@
-module Neo4j::Shared
+module ActiveGraph::Shared
   # TypecastedAttributes allows types to be declared for your attributes
   #
   # Types are declared by passing the :type option to the attribute class
@@ -11,7 +11,7 @@ module Neo4j::Shared
   #
   # @example Usage
   #   class Person
-  #     include Neo4j::Shared::TypecastedAttributes
+  #     include ActiveGraph::Shared::TypecastedAttributes
   #     attribute :age, :type => Integer
   #   end
   #
@@ -23,7 +23,7 @@ module Neo4j::Shared
   # Originally part of ActiveAttr, https://github.com/cgriego/active_attr
   module TypecastedAttributes
     extend ActiveSupport::Concern
-    include Neo4j::Shared::Attributes
+    include ActiveGraph::Shared::Attributes
 
     included do
       attribute_method_suffix '_before_type_cast'
@@ -39,7 +39,7 @@ module Neo4j::Shared
     #
     # @return [Object, nil] The attribute value before typecasting
     def attribute_before_type_cast(name)
-      @attributes ||= Neo4j::AttributeSet.new({}, self.class.attributes.keys)
+      @attributes ||= ActiveGraph::AttributeSet.new({}, self.class.attributes.keys)
       @attributes.fetch_value(name.to_s)
     end
 
@@ -67,8 +67,8 @@ module Neo4j::Shared
     def _attribute_typecaster(attribute_name)
       type = _attribute_type(attribute_name)
       default_typecaster = self.class.attributes[attribute_name].typecaster
-      caster = default_typecaster || Neo4j::Shared::TypeConverters.typecaster_for(type)
-      caster || fail(Neo4j::UnknownTypeConverterError, "Unable to cast to type #{type}")
+      caster = default_typecaster || ActiveGraph::Shared::TypeConverters.typecaster_for(type)
+      caster || fail(ActiveGraph::UnknownTypeConverterError, "Unable to cast to type #{type}")
     end
 
     module ClassMethods
@@ -92,7 +92,7 @@ module Neo4j::Shared
       end
 
       def typecast_attribute(typecaster, value)
-        Neo4j::Shared::TypeConverters.typecast_attribute(typecaster, value)
+        ActiveGraph::Shared::TypeConverters.typecast_attribute(typecaster, value)
       end
     end
   end

data/lib/{neo4j → active_graph}/shared/typecaster.rb

@@ -1,4 +1,4 @@
-module Neo4j
+module ActiveGraph
   module Shared
     # This module provides a convenient way of registering a custom Typecasting class. Custom Typecasters all follow a simple pattern.
     #
@@ -27,7 +27,7 @@ module Neo4j
     #       alias_method :call, :to_ruby
     #     end
     #
-    #     include Neo4j::Shared::Typecaster
+    #     include ActiveGraph::Shared::Typecaster
     #   end
     #
     # This would allow you to use `property :my_prop, type: Range` in a model.
@@ -46,7 +46,7 @@ module Neo4j
 
     module Typecaster
       def self.included(other)
-        Neo4j::Shared::TypeConverters.register_converter(other)
+        ActiveGraph::Shared::TypeConverters.register_converter(other)
       end
     end
   end

data/lib/{neo4j → active_graph}/shared/validations.rb

@@ -1,4 +1,4 @@
-module Neo4j
+module ActiveGraph
   module Shared
     module Validations
       extend ActiveSupport::Concern

data/lib/{neo4j → active_graph}/tasks/migration.rake

@@ -1,23 +1,23 @@
 require 'rake'
 require 'active_support/concern'
-require 'neo4j/migration'
+require 'active_graph/migration'
 
 if !defined?(Rails) && !Rake::Task.task_defined?('environment')
   desc 'Run a script against the database to perform system-wide changes'
   task :environment do
-    require 'neo4j/session_manager'
+    require 'active_graph/session_manager'
     require 'ostruct'
     neo4j_url = ENV['NEO4J_URL'] || 'http://localhost:7474'
     $LOAD_PATH.unshift File.dirname('./')
-    Neo4j::ActiveBase.on_establish_session do
-      Neo4j::ActiveBase.new_driver(neo4j_url)
+    ActiveGraph::Base.on_establish_session do
+      ActiveGraph::Base.new_driver(neo4j_url)
     end
   end
 end
 
 namespace :neo4j do
   task :allow_migrations do
-    Neo4j::Migrations.currently_running_migrations = true
+    ActiveGraph::Migrations.currently_running_migrations = true
   end
   desc 'Run a script against the database to perform system-wide changes'
   task :legacy_migrate, [:task_name, :subtask] => :environment do |_, args|
@@ -25,7 +25,7 @@ namespace :neo4j do
     migration_task = args[:task_name]
     task_name_constant = migration_task.split('_').map(&:capitalize).join('')
     begin
-      migration_class = "Neo4j::Migration::#{task_name_constant}".constantize
+      migration_class = "ActiveGraph::Migration::#{task_name_constant}".constantize
     rescue NameError
       load File.join(path, 'db', 'neo4j-migrate', "#{migration_task}.rb")
       migration_class = task_name_constant.to_s.constantize
@@ -50,7 +50,7 @@ namespace :neo4j do
     SCHEMA_YAML_PATH = 'db/neo4j/schema.yml'
     SCHEMA_YAML_COMMENT = <<COMMENT
 # This file is auto-generated from the current state of the database. Instead
-# of editing this file, please use the migrations feature of ActiveNode to
+# of editing this file, please use the migrations feature of Node to
 # incrementally modify your database, and then regenerate this schema definition.
 #
 # Note that this schema.yml definition is the authoritative source for your
@@ -64,7 +64,7 @@ namespace :neo4j do
 COMMENT
 
     def check_neo4j_version_3
-      if Neo4j::ActiveBase.transaction.version > '3.0.0'
+      if ActiveGraph::Base.transaction.version > '3.0.0'
         yield
       else
         puts 'WARNING: This task does not work for versions of Neo4j before 3.0.0'
@@ -74,11 +74,11 @@ COMMENT
     desc 'Creates a db/neo4j/schema.yml file which represents the indexes / constraints in the Neo4j DB'
     task dump: :environment do
       check_neo4j_version_3 do
-        require 'neo4j/migrations/schema'
+        require 'active_graph/migrations/schema'
 
-        schema_data = Neo4j::Migrations::Schema.fetch_schema_data(Neo4j::ActiveBase.transaction)
+        schema_data = ActiveGraph::Migrations::Schema.fetch_schema_data(ActiveGraph::Base.transaction)
 
-        runner = Neo4j::Migrations::Runner.new
+        runner = ActiveGraph::Migrations::Runner.new
         schema_data[:versions] = runner.complete_migration_versions.sort
 
         FileUtils.mkdir_p(File.dirname(SCHEMA_YAML_PATH))
@@ -91,20 +91,20 @@ COMMENT
     desc "Loads a db/neo4j/schema.yml file into the database\nOptionally removes schema elements which aren't in the schema.yml file (defaults to false)"
     task :load, [:remove_missing] => :environment do |_t, args|
       check_neo4j_version_3 do
-        require 'neo4j/migrations/schema'
+        require 'active_graph/migrations/schema'
 
         args.with_defaults(remove_missing: false)
 
         schema_data = YAML.safe_load(File.read(SCHEMA_YAML_PATH), [Symbol])
 
-        Neo4j::Transaction.subscribe_to_query(&method(:puts))
+        ActiveGraph::Transaction.subscribe_to_query(&method(:puts))
 
-        Neo4j::ActiveBase.run_transaction do
-          Neo4j::Migrations::Schema.synchronize_schema_data(Neo4j::ActiveBase.transaction, schema_data, args[:remove_missing])
+        ActiveGraph::Base.run_transaction do
+          ActiveGraph::Migrations::Schema.synchronize_schema_data(ActiveGraph::Base.transaction, schema_data, args[:remove_missing])
         end
 
-        Neo4j::ActiveBase.run_transaction do
-          runner = Neo4j::Migrations::Runner.new
+        ActiveGraph::Base.run_transaction do
+          runner = ActiveGraph::Migrations::Runner.new
           runner.mark_versions_as_complete(schema_data[:versions]) # Run in test mode?
         end
       end
@@ -114,7 +114,7 @@ COMMENT
   namespace :migrate do
     desc 'Run all pending migrations'
     task all: [:allow_migrations, :environment] do
-      runner = Neo4j::Migrations::Runner.new
+      runner = ActiveGraph::Migrations::Runner.new
       runner.all
 
       Rake::Task['neo4j:schema:dump'].invoke
@@ -123,7 +123,7 @@ COMMENT
     desc 'Run a migration given its VERSION'
     task up: [:allow_migrations, :environment] do
       version = ENV['VERSION'] || fail(ArgumentError, 'VERSION is required')
-      runner = Neo4j::Migrations::Runner.new
+      runner = ActiveGraph::Migrations::Runner.new
       runner.up version
 
       Rake::Task['neo4j:schema:dump'].invoke
@@ -132,7 +132,7 @@ COMMENT
     desc 'Revert a migration given its VERSION'
     task down: [:allow_migrations, :environment] do
       version = ENV['VERSION'] || fail(ArgumentError, 'VERSION is required')
-      runner = Neo4j::Migrations::Runner.new
+      runner = ActiveGraph::Migrations::Runner.new
       runner.down version
 
       Rake::Task['neo4j:schema:dump'].invoke
@@ -140,21 +140,21 @@ COMMENT
 
     desc 'Print a report of migrations status'
     task status: [:allow_migrations, :environment] do
-      runner = Neo4j::Migrations::Runner.new
+      runner = ActiveGraph::Migrations::Runner.new
       runner.status
     end
 
     desc 'Resolve an incomplete version state'
     task resolve: [:allow_migrations, :environment] do
       version = ENV['VERSION'] || fail(ArgumentError, 'VERSION is required')
-      runner = Neo4j::Migrations::Runner.new
+      runner = ActiveGraph::Migrations::Runner.new
       runner.resolve version
     end
 
     desc 'Resolve an incomplete version state'
     task reset: [:allow_migrations, :environment] do
       version = ENV['VERSION'] || fail(ArgumentError, 'VERSION is required')
-      runner = Neo4j::Migrations::Runner.new
+      runner = ActiveGraph::Migrations::Runner.new
       runner.reset version
     end
   end
@@ -162,7 +162,7 @@ COMMENT
   desc 'Rollbacks migrations given a STEP number'
   task rollback: [:allow_migrations, :environment] do
     steps = (ENV['STEP'] || 1).to_i
-    runner = Neo4j::Migrations::Runner.new
+    runner = ActiveGraph::Migrations::Runner.new
     runner.rollback(steps)
 
     Rake::Task['neo4j:schema:dump'].invoke
@@ -186,7 +186,7 @@ COMMENT
     path = File.join('db/neo4j/migrate', "#{DateTime.now.utc.strftime('%Y%m%d%H%M%S')}_#{migration_class_name.underscore}.rb")
 
     content = <<-CONTENT
-class #{migration_class_name} < Neo4j::Migrations::Base
+class #{migration_class_name} < ActiveGraph::Migrations::Base
   def up
     add_#{index_or_constraint} #{label.to_sym.inspect}, #{property_name.to_sym.inspect}, force: true
   end