lib/neo4j/active_rel/persistence.rb in neo4j-3.0.4 vs lib/neo4j/active_rel/persistence.rb in neo4j-4.0.0.rc.1

@@ -22,10 +22,11 @@
       confirm_node_classes
       create_magic_properties
       set_timestamps
       properties = convert_properties_to :db, props
       rel = _create_rel(from_node, to_node, properties)
+      return self unless rel.respond_to?(:_persisted_obj)
       init_on_load(rel._persisted_obj, from_node, to_node, @rel_type)
       true
     end
 
     module ClassMethods
@@ -59,11 +60,11 @@
     end
 
     def _create_rel(from_node, to_node, *args)
       props = self.class.default_property_values(self)
       props.merge!(args[0]) if args[0].is_a?(Hash)
-      set_classname(props)
+      set_classname(props, true)
       _rel_creation_query(from_node, to_node, props)
     end
 
     def class_as_constant(type)
       given_class = self.class.send(type)
@@ -97,6 +98,6 @@
         raise RelCreateFailedError, "Unable to create relationship. from_node: #{from_node}, to_node: #{to_node}"
       end
     end
 
   end
-end
\ No newline at end of file
+end