lib/neo4j/active_rel/persistence.rb in neo4j-6.0.0.alpha.12 vs lib/neo4j/active_rel/persistence.rb in neo4j-6.0.0.rc.1

@@ -1,21 +1,28 @@
 module Neo4j::ActiveRel
   module Persistence
     extend ActiveSupport::Concern
+    include Neo4j::Shared::Cypher::RelIdentifiers
     include Neo4j::Shared::Persistence
 
-    attr_writer :from_node_identifier, :to_node_identifier
-
     class RelInvalidError < RuntimeError; end
     class ModelClassInvalidError < RuntimeError; end
     class RelCreateFailedError < RuntimeError; end
 
     def from_node_identifier
       @from_node_identifier || :from_node
     end
 
     def to_node_identifier
       @to_node_identifier || :to_node
+    end
+
+    def from_node_identifier=(id)
+      @from_node_identifier = id.to_sym
+    end
+
+    def to_node_identifier=(id)
+      @to_node_identifier = id.to_sym
     end
 
     def cypher_identifier
       @cypher_identifier || :rel
     end