fix bug in case the partial resolution of an object changes some prop…

…erties
jpmckinney · Nov 25, 2013 · c585761 · c585761
1 parent 18ceba1
commit c585761
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/lib/pupa/processor.rb b/lib/pupa/processor.rb
@@ -151,7 +151,10 @@ def import
           object = objects[id]
           resolve_foreign_keys(object, object_id_to_database_id)
           # The dependency graph strategy only works if there are no foreign objects.
-          object_id_to_database_id[id] = import_object(object)
+
+          database_id = import_object(object)
+          object_id_to_database_id[id] = database_id
+          object_id_to_database_id[database_id] = database_id
         end
       else
         size = objects.size
@@ -170,8 +173,11 @@ def import
             begin
               resolve_foreign_keys(object, object_id_to_database_id)
               resolve_foreign_objects(object, object_id_to_database_id)
-              object_id_to_database_id[id] = import_object(object)
               progress_made = true
+
+              database_id = import_object(object)
+              object_id_to_database_id[id] = database_id
+              object_id_to_database_id[database_id] = database_id
             rescue Pupa::Errors::MissingDatabaseIdError
               false
             end