Fixed olivernn#117

Updated to add the actual reference instead of relying on the map's key.

lib/index.js

@@ -319,8 +319,12 @@ lunr.Index.prototype.search = function (query) {
         if (pos > -1) queryVector.insert(pos, tf * idf * similarityBoost)
 
         // add all the documents that have this key into a set
-        Object.keys(self.tokenStore.get(key)).forEach(function (ref) { set.add(ref) })
-
+        var documents = self.tokenStore.get(key);
+        for(var ref in documents){
+          if(!documents.hasOwnProperty(ref))continue;
+          set.add(documents[ref].ref);
+        }
+
         return memo.union(set)
       }, new lunr.SortedSet)
 

test/search_test.js

@@ -27,6 +27,10 @@ module('search', {
       id: 'e',
       title: 'title',
       body: 'hand',
+    },{
+      id: 10,
+      title: 'ten',
+      body: 'ten 10'
     }]).forEach(function (doc) { idx.add(doc) })
 
     this.idx = idx
@@ -75,3 +79,10 @@ test('search boosts exact matches', function () {
 
   ok(results[0].score > results[1].score)
 })
+
+test('returning the correct type for reference', function () {
+  var results = this.idx.search('ten');
+  equal(results.length, 1)
+  equal(typeof results[0].ref, 'number')
+  equal(results[0].ref, 10)
+})