closes #272: createCollection with command

Author: William Delanoue

src/main/java/com/github/fakemongo/FongoConnection.java

@@ -1,39 +1,5 @@
 package com.github.fakemongo;
 
-import static com.mongodb.FongoDBCollection.bsonDocument;
-import static com.mongodb.FongoDBCollection.bsonDocuments;
-import static com.mongodb.FongoDBCollection.dbObject;
-import static com.mongodb.FongoDBCollection.dbObjects;
-import static com.mongodb.FongoDBCollection.decode;
-import static com.mongodb.FongoDBCollection.decoderContext;
-import static com.mongodb.bulk.WriteRequest.Type.INSERT;
-import static com.mongodb.bulk.WriteRequest.Type.REPLACE;
-import static java.util.Arrays.asList;
-
-import java.util.ArrayList;
-import java.util.Collections;
-import java.util.Iterator;
-import java.util.List;
-import java.util.Map;
-import java.util.concurrent.TimeUnit;
-
-import org.bson.BsonArray;
-import org.bson.BsonBoolean;
-import org.bson.BsonDocument;
-import org.bson.BsonDocumentReader;
-import org.bson.BsonDouble;
-import org.bson.BsonInt32;
-import org.bson.BsonInt64;
-import org.bson.BsonNull;
-import org.bson.BsonString;
-import org.bson.BsonValue;
-import org.bson.Document;
-import org.bson.FieldNameValidator;
-import org.bson.codecs.Codec;
-import org.bson.codecs.Decoder;
-import org.slf4j.Logger;
-import org.slf4j.LoggerFactory;
-
 import com.mongodb.AggregationOutput;
 import com.mongodb.BasicDBObject;
 import com.mongodb.BulkUpdateRequestBuilder;
@@ -45,6 +11,12 @@
 import com.mongodb.DBObject;
 import com.mongodb.FongoDB;
 import com.mongodb.FongoDBCollection;
+import static com.mongodb.FongoDBCollection.bsonDocument;
+import static com.mongodb.FongoDBCollection.bsonDocuments;
+import static com.mongodb.FongoDBCollection.dbObject;
+import static com.mongodb.FongoDBCollection.dbObjects;
+import static com.mongodb.FongoDBCollection.decode;
+import static com.mongodb.FongoDBCollection.decoderContext;
 import com.mongodb.MongoClient;
 import com.mongodb.MongoCommandException;
 import com.mongodb.MongoNamespace;
@@ -60,6 +32,8 @@
 import com.mongodb.bulk.UpdateRequest;
 import com.mongodb.bulk.WriteConcernError;
 import com.mongodb.bulk.WriteRequest;
+import static com.mongodb.bulk.WriteRequest.Type.INSERT;
+import static com.mongodb.bulk.WriteRequest.Type.REPLACE;
 import com.mongodb.connection.BulkWriteBatchCombiner;
 import com.mongodb.connection.ClusterId;
 import com.mongodb.connection.Connection;
@@ -72,6 +46,29 @@
 import com.mongodb.internal.validator.UpdateFieldNameValidator;
 import com.mongodb.operation.FongoBsonArrayWrapper;
 import com.mongodb.util.JSON;
+import java.util.ArrayList;
+import static java.util.Arrays.asList;
+import java.util.Collections;
+import java.util.Iterator;
+import java.util.List;
+import java.util.Map;
+import java.util.concurrent.TimeUnit;
+import org.bson.BsonArray;
+import org.bson.BsonBoolean;
+import org.bson.BsonDocument;
+import org.bson.BsonDocumentReader;
+import org.bson.BsonDouble;
+import org.bson.BsonInt32;
+import org.bson.BsonInt64;
+import org.bson.BsonNull;
+import org.bson.BsonString;
+import org.bson.BsonValue;
+import org.bson.Document;
+import org.bson.FieldNameValidator;
+import org.bson.codecs.Codec;
+import org.bson.codecs.Decoder;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
 
 /**
  *
@@ -388,7 +385,11 @@ private int delete(DBCollection collection, WriteConcern writeConcern, List<Dele
   public <T> T command(String database, BsonDocument command, boolean slaveOk, FieldNameValidator fieldNameValidator, Decoder<T> commandResultDecoder) {
     final DB db = fongo.getDB(database);
     LOG.debug("command() database:{}, command:{}", database, command);
-    if (command.containsKey("count")) {
+    if (command.containsKey("create")) {
+      final FongoDBCollection dbCollection = (FongoDBCollection) db.getCollection(command.get("create").asString().getValue());
+
+      return (T) new BsonDocument("ok", BsonBoolean.TRUE);
+    } else if (command.containsKey("count")) {
       final FongoDBCollection dbCollection = (FongoDBCollection) db.getCollection(command.get("count").asString().getValue());
       final DBObject query = dbObject(command, "query");
       final long limit = command.containsKey("limit") ? command.getInt64("limit").longValue() : -1;

src/test/java/com/github/fakemongo/AbstractFongoV3Test.java

@@ -1129,6 +1129,16 @@ public void should_ping_fongo() {
     Assertions.assertThat(ping.getDouble("ok")).isEqualTo(1.0);
   }
 
+  @Test
+  public void should_command_create_collection() {
+    // Given
+    // When
+    fongoRule.getDatabase().createCollection("hello");
+
+    // Then
+    Assertions.assertThat(fongoRule.getDatabase().getCollection("hello")).isNotNull();
+  }
+
   @Test
   public void shouldFindOneDocumentWithoutTheOptionalFieldWithNoIndex() {
     // Given