always return arrays when fed arrays

Author: juliangruber

index.js

@@ -31,7 +31,8 @@ var backslash = code('\\');
  */
 
 function extract(buf, keys){
-  if (!Array.isArray(keys)) keys = [keys];
+  var multi = Array.isArray(keys);
+  if (!multi) keys = [keys];
 
   var values = [];
   var matched = {};
@@ -86,9 +87,9 @@ function extract(buf, keys){
     if (values.length == keys.length) break;
   }
 
-  return keys.length == 1
-    ? values[0]
-    : values;
+  return multi
+    ? values
+    : values[0];
 }
 
 /**

test.js

@@ -12,6 +12,10 @@ describe('extract(buf, key)', function(){
     var buf = toBuf({ a: '0', b: '1', c: '2' });
     equal(extract(buf, ['a', 'c']), ['0', '2']);
   })
+  it('should always return arrays when fed arrays', function(){
+    var buf = toBuf({ foo: 'bar' });
+    equal(extract(buf, ['foo']), ['bar']);
+  })
   it('should end on ,', function(){
     var buf = toBuf({ foo: 'bar', bar: 'baz' });
     equal(extract(buf, 'foo'), 'bar');