From 4e56d9c57529883ba6ba0572716e4cbb3f47000c Mon Sep 17 00:00:00 2001
From: Christoph Strobl <cstrobl@pivotal.io>
Date: Mon, 12 Dec 2016 09:27:45 +0100
Subject: [PATCH] DATAMONGO-1549 - Polishing $count (aggregation stage).

Original Pull Request: #422
---
 .../mongodb/core/aggregation/CountOperation.java     | 12 ++++++------
 .../core/aggregation/CountOperationUnitTests.java    |  7 ++-----
 2 files changed, 8 insertions(+), 11 deletions(-)

diff --git a/spring-data-mongodb/src/main/java/org/springframework/data/mongodb/core/aggregation/CountOperation.java b/spring-data-mongodb/src/main/java/org/springframework/data/mongodb/core/aggregation/CountOperation.java
index ed4c274c2..c99c1521e 100644
--- a/spring-data-mongodb/src/main/java/org/springframework/data/mongodb/core/aggregation/CountOperation.java
+++ b/spring-data-mongodb/src/main/java/org/springframework/data/mongodb/core/aggregation/CountOperation.java
@@ -22,12 +22,12 @@ import com.mongodb.BasicDBObject;
 import com.mongodb.DBObject;
 
 /**
- * Encapsulates the aggregation framework {@code $count}-operation.
- * <p>
- * We recommend to use the static factory method {@link Aggregation#count()} instead of creating instances of this
- * class directly.
- * 
- * @see https://docs.mongodb.com/manual/reference/operator/aggregation/count/#pipe._S_count
+ * Encapsulates the aggregation framework {@code $count}-operation. <br />
+ * We recommend to use the static factory method {@link Aggregation#count()} instead of creating instances of this class
+ * directly.
+ *
+ * @see <a href=
+ *      "https://docs.mongodb.com/manual/reference/operator/aggregation/count/#pipe._S_count">https://docs.mongodb.com/manual/reference/operator/aggregation/count/</a>
  * @author Mark Paluch
  * @since 1.10
  */
diff --git a/spring-data-mongodb/src/test/java/org/springframework/data/mongodb/core/aggregation/CountOperationUnitTests.java b/spring-data-mongodb/src/test/java/org/springframework/data/mongodb/core/aggregation/CountOperationUnitTests.java
index 2b60442f1..a91441558 100644
--- a/spring-data-mongodb/src/test/java/org/springframework/data/mongodb/core/aggregation/CountOperationUnitTests.java
+++ b/spring-data-mongodb/src/test/java/org/springframework/data/mongodb/core/aggregation/CountOperationUnitTests.java
@@ -20,12 +20,11 @@ import static org.junit.Assert.*;
 
 import org.junit.Test;
 
-import com.mongodb.DBObject;
 import com.mongodb.util.JSON;
 
 /**
  * Unit tests for {@link CountOperation}.
- * 
+ *
  * @author Mark Paluch
  */
 public class CountOperationUnitTests {
@@ -45,9 +44,7 @@ public class CountOperationUnitTests {
 	public void shouldRenderCorrectly() {
 
 		CountOperation countOperation = new CountOperation("field");
-		DBObject dbObject = countOperation.toDBObject(Aggregation.DEFAULT_CONTEXT);
-
-		assertThat(dbObject, is(JSON.parse("{$count : \"field\" }")));
+		assertThat(countOperation.toDBObject(Aggregation.DEFAULT_CONTEXT), is(JSON.parse("{$count : \"field\" }")));
 	}
 
 	/**