datastore: return a mix of auto-inserted and already existing keys. fixes #17.

Author: rakyll

lib/datastore/index.js

@@ -195,9 +195,10 @@ Transaction.prototype.saveAll = function(keys, objs, callback) {
     if (err || !resp) {
       return callback(err);
     }
-    var keys = [];
-    resp.mutationResult.insertAutoIdKeys && resp.mutationResult.insertAutoIdKeys.forEach(function(item) {
-      keys.push(entity.keyFromKeyProto(item));
+    resp.mutationResult.insertAutoIdKeys = resp.mutationResult.insertAutoIdKeys || [];
+    var i = 0;
+    resp.mutationResult.insertAutoIdKeys.forEach(function(item) {
+      keys[insertIndexes[i++]] = entity.keyFromKeyProto(item);
     });
     callback(null, keys);
   });