DATAMONGO-1008 - Polishing.

Slightly changed the implementation of the 2dsphere check, Minor refactorings in the test case.

Original pull request: #210.

Author: odrotbohm

spring-data-mongodb/src/main/java/org/springframework/data/mongodb/core/DefaultIndexOperations.java

@@ -18,14 +18,15 @@
 import static org.springframework.data.domain.Sort.Direction.*;
 
 import java.util.ArrayList;
+import java.util.Arrays;
+import java.util.Collection;
 import java.util.List;
 
 import org.springframework.dao.DataAccessException;
 import org.springframework.data.mongodb.core.index.IndexDefinition;
 import org.springframework.data.mongodb.core.index.IndexField;
 import org.springframework.data.mongodb.core.index.IndexInfo;
 import org.springframework.util.Assert;
-import org.springframework.util.ObjectUtils;
 
 import com.mongodb.DBCollection;
 import com.mongodb.DBObject;
@@ -42,6 +43,7 @@ public class DefaultIndexOperations implements IndexOperations {
 
 	private static final Double ONE = Double.valueOf(1);
 	private static final Double MINUS_ONE = Double.valueOf(-1);
+	private static final Collection<String> TWO_D_IDENTIFIERS = Arrays.asList("2d", "2dsphere");
 
 	private final MongoOperations mongoOperations;
 	private final String collectionName;
@@ -141,7 +143,7 @@ private List<IndexInfo> getIndexData(List<DBObject> dbObjectList) {
 
 						Object value = keyDbObject.get(key);
 
-						if (isGeoIndex(value)) {
+						if (TWO_D_IDENTIFIERS.contains(value)) {
 							indexFields.add(IndexField.geo(key));
 						} else {
 
@@ -168,8 +170,4 @@ private List<IndexInfo> getIndexData(List<DBObject> dbObjectList) {
 			}
 		});
 	}
-
-	private boolean isGeoIndex(Object indexType) {
-		return ObjectUtils.nullSafeEquals("2d", indexType) || ObjectUtils.nullSafeEquals("2dsphere", indexType);
-	}
 }

spring-data-mongodb/src/test/java/org/springframework/data/mongodb/core/DefaultIndexOperationsIntegrationTests.java

@@ -15,7 +15,7 @@
  */
 package org.springframework.data.mongodb.core;
 
-import static org.hamcrest.core.Is.*;
+import static org.hamcrest.CoreMatchers.*;
 import static org.junit.Assert.*;
 
 import org.junit.Before;
@@ -25,34 +25,37 @@
 import org.springframework.data.mongodb.core.index.IndexInfo;
 import org.springframework.test.context.ContextConfiguration;
 import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;
-import org.springframework.util.ClassUtils;
 import org.springframework.util.ObjectUtils;
 
 import com.mongodb.BasicDBObject;
 import com.mongodb.DBCollection;
 import com.mongodb.DBObject;
 
 /**
+ * Integration tests for {@link DefaultIndexOperations}.
+ * 
  * @author Christoph Strobl
+ * @author Oliver Gierke
  */
 @RunWith(SpringJUnit4ClassRunner.class)
 @ContextConfiguration("classpath:infrastructure.xml")
-public class DefaultIndexOperationTests {
+public class DefaultIndexOperationsIntegrationTests {
 
-	private static final String COLLECTION_NAME = ClassUtils.getShortNameAsProperty(DefaultIndexOperationTests.class);
-	private static final DBObject GEO_SPHERE_2D = new BasicDBObject("loaction", "2dsphere");
+	static final DBObject GEO_SPHERE_2D = new BasicDBObject("loaction", "2dsphere");
 
 	@Autowired MongoTemplate template;
-	private DefaultIndexOperations indexOps;
-	private DBCollection collection;
+	DefaultIndexOperations indexOps;
+	DBCollection collection;
 
 	@Before
 	public void setUp() {
 
-		this.collection = this.template.getDb().getCollection(COLLECTION_NAME);
-		dropAllIndexes();
+		String collectionName = this.template.getCollectionName(DefaultIndexOperationsIntegrationTestsSample.class);
 
-		indexOps = new DefaultIndexOperations(template, COLLECTION_NAME);
+		this.collection = this.template.getDb().getCollection(collectionName);
+		this.collection.dropIndexes();
+
+		this.indexOps = new DefaultIndexOperations(template, collectionName);
 	}
 
 	/**
@@ -61,7 +64,7 @@ public void setUp() {
 	@Test
 	public void getIndexInfoShouldBeAbleToRead2dsphereIndex() {
 
-		createIndex(GEO_SPHERE_2D);
+		collection.createIndex(GEO_SPHERE_2D);
 
 		IndexInfo info = findAndReturnIndexInfo(GEO_SPHERE_2D);
 		assertThat(info.getIndexFields().get(0).isGeo(), is(true));
@@ -72,11 +75,11 @@ private IndexInfo findAndReturnIndexInfo(DBObject keys) {
 	}
 
 	@SuppressWarnings("deprecation")
-	private IndexInfo findAndReturnIndexInfo(Iterable<IndexInfo> candidates, DBObject keys) {
+	private static IndexInfo findAndReturnIndexInfo(Iterable<IndexInfo> candidates, DBObject keys) {
 		return findAndReturnIndexInfo(candidates, DBCollection.genIndexName(keys));
 	}
 
-	private IndexInfo findAndReturnIndexInfo(Iterable<IndexInfo> candidates, String name) {
+	private static IndexInfo findAndReturnIndexInfo(Iterable<IndexInfo> candidates, String name) {
 
 		for (IndexInfo info : candidates) {
 			if (ObjectUtils.nullSafeEquals(name, info.getName())) {
@@ -86,12 +89,5 @@ private IndexInfo findAndReturnIndexInfo(Iterable<IndexInfo> candidates, String
 		throw new AssertionError(String.format("Index with %s was not found", name));
 	}
 
-	private void createIndex(DBObject keys) {
-		template.getDb().getCollection(COLLECTION_NAME).createIndex(keys);
-	}
-
-	private void dropAllIndexes() {
-		this.collection.dropIndexes();
-	}
-
+	static class DefaultIndexOperationsIntegrationTestsSample {}
 }