1 files changed, 163 insertions, 0 deletions
diff --git a/jstests/geo_array2.js b/jstests/geo_array2.js
new file mode 100644
index 0000000..28cb152
--- /dev/null
+++ b/jstests/geo_array2.js
@@ -0,0 +1,163 @@
+// Check the semantics of near calls with multiple locations
+
+t = db.geoarray2
+t.drop();
+
+var numObjs = 10;
+var numLocs = 100;
+
+// Test the semantics of near / nearSphere / etc. queries with multiple keys per object
+
+for( var i = -1; i < 2; i++ ){
+	for(var j = -1; j < 2; j++ ){
+		
+		locObj = []
+		
+		if( i != 0 || j != 0 )
+			locObj.push( { x : i * 50 + Random.rand(), 
+			               y : j * 50 + Random.rand() } )
+	    locObj.push( { x : Random.rand(), 
+	    			   y : Random.rand() } )
+	    locObj.push( { x : Random.rand(),
+	    			   y : Random.rand() } )
+	    			   
+	    t.insert({ name : "" + i + "" + j , loc : locObj , type : "A" })
+	    t.insert({ name : "" + i + "" + j , loc : locObj , type : "B" })
+	}
+}
+
+t.ensureIndex({ loc : "2d" , type : 1 })
+
+assert.isnull( db.getLastError() )
+
+print( "Starting testing phase... ")
+
+for( var t = 0; t < 2; t++ ){
+	
+var type = t == 0 ? "A" : "B"
+
+for( var i = -1; i < 2; i++ ){
+	for(var j = -1; j < 2; j++ ){
+		
+		var center = [ i * 50 , j * 50 ]
+		var count = i == 0 && j == 0 ? 2 * 9 : 1
+		var objCount = i == 0 && j == 0 ? 2 : 1
+		
+		// Do near check
+				
+		var nearResults = db.runCommand( { geoNear : "geoarray2" , 
+											  near : center ,
+											   num : count,
+											 query : { type : type } } ).results
+		//printjson( nearResults )								 
+			
+	    var objsFound = {}
+		var lastResult = 0;
+		for( var k = 0; k < nearResults.length; k++ ){
+			
+			// All distances should be small, for the # of results
+			assert.gt( 1.5 , nearResults[k].dis )
+			// Distances should be increasing
+			assert.lte( lastResult, nearResults[k].dis )
+			// Objs should be of the right type
+			assert.eq( type, nearResults[k].obj.type )
+			
+			lastResult = nearResults[k].dis
+			
+			var objKey = "" + nearResults[k].obj._id
+			
+			if( objKey in objsFound ) objsFound[ objKey ]++
+			else objsFound[ objKey ] = 1
+			
+		}
+				
+		// Make sure we found the right objects each time
+		// Note: Multiple objects could be found for diff distances.
+		for( var q in objsFound ){
+			assert.eq( objCount , objsFound[q] )
+		}
+		
+				
+		// Do nearSphere check
+		
+		// Earth Radius
+		var eRad = 6371
+				
+		nearResults = db.geoarray2.find( { loc : { $nearSphere : center , $maxDistance : 500 /* km */ / eRad }, type : type } ).toArray()
+		
+		assert.eq( nearResults.length , count )
+			
+	    objsFound = {}
+		lastResult = 0;
+		for( var k = 0; k < nearResults.length; k++ ){			
+			var objKey = "" + nearResults[k]._id
+			if( objKey in objsFound ) objsFound[ objKey ]++
+			else objsFound[ objKey ] = 1
+			
+		}
+		
+		// Make sure we found the right objects each time
+		for( var q in objsFound ){
+			assert.eq( objCount , objsFound[q] )
+		}
+		
+		
+		
+		// Within results do not return duplicate documents
+		
+		var count = i == 0 && j == 0 ? 9 : 1
+		var objCount = i == 0 && j == 0 ? 1 : 1
+		
+		// Do within check
+		objsFound = {}
+		
+		var box = [ [center[0] - 1, center[1] - 1] , [center[0] + 1, center[1] + 1] ]
+		
+		//printjson( box )
+		
+		var withinResults = db.geoarray2.find({ loc : { $within : { $box :  box } } , type : type }).toArray()
+		
+		assert.eq( withinResults.length , count )
+		
+		for( var k = 0; k < withinResults.length; k++ ){
+			var objKey = "" + withinResults[k]._id
+			if( objKey in objsFound ) objsFound[ objKey ]++
+			else objsFound[ objKey ] = 1
+		}
+		
+		//printjson( objsFound )
+		
+		// Make sure we found the right objects each time
+		for( var q in objsFound ){
+			assert.eq( objCount , objsFound[q] )
+		}
+		
+		
+		// Do within check (circle)
+		objsFound = {}
+		
+		withinResults = db.geoarray2.find({ loc : { $within : { $center :  [ center, 1.5 ] } } , type : type }).toArray()
+		
+		assert.eq( withinResults.length , count )
+		
+		for( var k = 0; k < withinResults.length; k++ ){
+			var objKey = "" + withinResults[k]._id
+			if( objKey in objsFound ) objsFound[ objKey ]++
+			else objsFound[ objKey ] = 1
+		}
+		
+		// Make sure we found the right objects each time
+		for( var q in objsFound ){
+			assert.eq( objCount , objsFound[q] )
+		}
+		
+		
+		
+	}
+}
+
+}
+
+
+
+