From ea9fccbf2afe97b395bab580e0a58f4fd805be99 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Bar=C4=B1=C5=9F=20Soner=20U=C5=9Fakl=C4=B1?=
 <barisusakli@gmail.com>
Date: Fri, 6 Mar 2015 15:12:55 -0500
Subject: [PATCH] fix -inf +inf

added test for -inf +inf
---
 src/database/mongo/sorted.js | 11 +++++++----
 tests/database/sorted.js     |  7 +++++++
 2 files changed, 14 insertions(+), 4 deletions(-)

diff --git a/src/database/mongo/sorted.js b/src/database/mongo/sorted.js
index 565189dd8f..5b06e22cfb 100644
--- a/src/database/mongo/sorted.js
+++ b/src/database/mongo/sorted.js
@@ -204,14 +204,17 @@ module.exports = function(db, module) {
 		if (!key) {
 			return callback();
 		}
-		var scoreQuery = {};
+
+		var query = {_key: key};
 		if (min !== '-inf') {
-			scoreQuery.$gte = min;
+			query.score = {$gte: min};
 		}
 		if (max !== '+inf') {
-			scoreQuery.$lte = max;
+			query.score = query.score || {};
+			query.score.$lte = max;
 		}
-		db.collection('objects').count({_key: key, score: scoreQuery}, function(err, count) {
+
+		db.collection('objects').count(query, function(err, count) {
 			callback(err, count ? count : 0);
 		});
 	};
diff --git a/tests/database/sorted.js b/tests/database/sorted.js
index 68e6f587da..9e92a4c388 100644
--- a/tests/database/sorted.js
+++ b/tests/database/sorted.js
@@ -172,6 +172,13 @@ describe('Sorted Set methods', function() {
 				assert.equal(count, 2);
 				done();
 			});
+
+			db.sortedSetCount('sortedSetTest1', '-inf', '+inf', function(err, count) {
+				assert.equal(err, null);
+				assert.equal(arguments.length, 2);
+				assert.equal(count, 3);
+				done();
+			});	
 		});
 	});