Polish contribution

spring-boot-autoconfigure/src/main/java/org/springframework/boot/autoconfigure/mongo/embedded/EmbeddedMongoAutoConfiguration.java

@@ -35,6 +35,7 @@ import de.flapdoodle.embed.mongo.config.IMongodConfig;
 import de.flapdoodle.embed.mongo.config.MongodConfigBuilder;
 import de.flapdoodle.embed.mongo.config.Net;
 import de.flapdoodle.embed.mongo.config.RuntimeConfigBuilder;
+import de.flapdoodle.embed.mongo.config.Storage;
 import de.flapdoodle.embed.mongo.distribution.Feature;
 import de.flapdoodle.embed.mongo.distribution.IFeatureAwareVersion;
 import de.flapdoodle.embed.process.config.IRuntimeConfig;
@@ -69,6 +70,7 @@ import org.springframework.util.Assert;
  *
  * @author Henryk Konsek
  * @author Andy Wilkinson
+ * @author Yogesh Lonkar
  * @since 1.3.0
  */
 @Configuration
@@ -126,7 +128,12 @@ public class EmbeddedMongoAutoConfiguration {
 		MongodConfigBuilder builder = new MongodConfigBuilder()
 				.version(featureAwareVersion);
 		if (this.embeddedProperties.getStorage() != null) {
-			builder.replication(this.embeddedProperties.getStorage());
+			builder.replication(
+					new Storage(this.embeddedProperties.getStorage().getDatabaseDir(),
+							this.embeddedProperties.getStorage().getReplSetName(),
+							this.embeddedProperties.getStorage().getOplogSize() != null
+									? this.embeddedProperties.getStorage().getOplogSize()
+									: 0));
 		}
 		if (getPort() > 0) {
 			builder.net(new Net(getHost().getHostAddress(), getPort(),
@@ -139,7 +146,6 @@ public class EmbeddedMongoAutoConfiguration {
 		return builder.build();
 	}
 
-
 	private int getPort() {
 		if (this.properties.getPort() == null) {
 			return MongoProperties.DEFAULT_PORT;

spring-boot-autoconfigure/src/main/java/org/springframework/boot/autoconfigure/mongo/embedded/EmbeddedMongoProperties.java

@@ -1,5 +1,5 @@
 /*
- * Copyright 2012-2015 the original author or authors.
+ * Copyright 2012-2016 the original author or authors.
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
@@ -28,6 +28,7 @@ import org.springframework.boot.context.properties.ConfigurationProperties;
  * Configuration properties for Embedded Mongo.
  *
  * @author Andy Wilkinson
+ * @author Yogesh Lonkar
  * @since 1.3.0
  */
 @ConfigurationProperties(prefix = "spring.mongodb.embedded")
@@ -63,29 +64,40 @@ public class EmbeddedMongoProperties {
 	}
 
 	public Storage getStorage() {
-		return storage;
+		return this.storage;
 	}
 
 	public void setStorage(Storage storage) {
 		this.storage = storage;
 	}
 
-	public static class Storage extends de.flapdoodle.embed.mongo.config.Storage {
+	public static class Storage {
 
-		private int oplogSize;
+		/**
+		 * Maximum size of the oplog in megabytes.
+		 */
+		private Integer oplogSize;
+
+		/**
+		 * Name of the replica set.
+		 */
 		private String replSetName;
+
+		/**
+		 * Directory used for data storage.
+		 */
 		private String databaseDir;
 
-		public int getOplogSize() {
+		public Integer getOplogSize() {
-			return oplogSize;
+			return this.oplogSize;
 		}
 
-		public void setOplogSize(int oplogSize) {
+		public void setOplogSize(Integer oplogSize) {
 			this.oplogSize = oplogSize;
 		}
 
 		public String getReplSetName() {
-			return replSetName;
+			return this.replSetName;
 		}
 
 		public void setReplSetName(String replSetName) {
@@ -93,7 +105,7 @@ public class EmbeddedMongoProperties {
 		}
 
 		public String getDatabaseDir() {
-			return databaseDir;
+			return this.databaseDir;
 		}
 
 		public void setDatabaseDir(String databaseDir) {

spring-boot-autoconfigure/src/test/java/org/springframework/boot/autoconfigure/mongo/embedded/EmbeddedMongoAutoConfigurationTests.java

@@ -16,11 +16,13 @@
 
 package org.springframework.boot.autoconfigure.mongo.embedded;
 
+import java.io.File;
 import java.net.UnknownHostException;
 
 import com.mongodb.CommandResult;
 import com.mongodb.MongoClient;
 import de.flapdoodle.embed.mongo.config.IMongodConfig;
+import de.flapdoodle.embed.mongo.config.Storage;
 import de.flapdoodle.embed.mongo.distribution.Feature;
 import org.junit.After;
 import org.junit.Test;
@@ -35,6 +37,7 @@ import org.springframework.context.annotation.AnnotationConfigApplicationContext
 import org.springframework.context.annotation.Bean;
 import org.springframework.context.annotation.Configuration;
 import org.springframework.data.mongodb.core.MongoTemplate;
+import org.springframework.util.FileSystemUtils;
 import org.springframework.util.SocketUtils;
 
 import static org.assertj.core.api.Assertions.assertThat;
@@ -113,28 +116,61 @@ public class EmbeddedMongoAutoConfigurationTests {
 		}
 	}
 
-	/**
-	 * test dbpath configuration is loaded for mongodb process.
-	 */
 	@Test
-	public void dbPathIsAvailableInMongoConfiguration() {
+	public void defaultStorageConfiguration() {
-		ConfigurableApplicationContext parent = new AnnotationConfigApplicationContext();
+		this.context = new AnnotationConfigApplicationContext();
-		parent.refresh();
+		EnvironmentTestUtils.addEnvironment(this.context, "spring.data.mongodb.port=0");
-		try {
+		this.context.register(EmbeddedMongoAutoConfiguration.class,
-			this.context = new AnnotationConfigApplicationContext();
+				MongoClientConfiguration.class,
-			this.context.setParent(parent);
+				PropertyPlaceholderAutoConfiguration.class);
-			EnvironmentTestUtils.addEnvironment(this.context, "spring.data.mongodb.port=0",
+		this.context.refresh();
-					"spring.mongodb.embedded.storage.databaseDir=/Users/yogeshlo/db",
+		Storage replication = this.context.getBean(IMongodConfig.class).replication();
-					"spring.mongodb.embedded.storage.oplogSize=0");
+		assertThat(replication.getOplogSize()).isEqualTo(0);
-			this.context.register(EmbeddedMongoAutoConfiguration.class, MongoClientConfiguration.class,
+		assertThat(replication.getDatabaseDir()).isNull();
-					PropertyPlaceholderAutoConfiguration.class);
+		assertThat(replication.getReplSetName()).isNull();
-			this.context.refresh();
+	}
-			IMongodConfig mongoConfig = this.context.getBean(IMongodConfig.class);
+
-			assertThat(mongoConfig.replication().getDatabaseDir()).isEqualTo("/Users/yogeshlo/db");
+	@Test
-		}
+	public void mongoWritesToCustomDatabaseDir() {
-		finally {
+		File customDatabaseDir = new File("target/custom-database-dir");
-			parent.close();
+		FileSystemUtils.deleteRecursively(customDatabaseDir);
-		}
+		this.context = new AnnotationConfigApplicationContext();
+		EnvironmentTestUtils.addEnvironment(this.context, "spring.data.mongodb.port=0",
+				"spring.mongodb.embedded.storage.databaseDir="
+						+ customDatabaseDir.getPath());
+		this.context.register(EmbeddedMongoAutoConfiguration.class,
+				MongoClientConfiguration.class,
+				PropertyPlaceholderAutoConfiguration.class);
+		this.context.refresh();
+		assertThat(customDatabaseDir).isDirectory();
+		assertThat(customDatabaseDir.listFiles()).isNotEmpty();
+	}
+
+	@Test
+	public void customOpLogSizeIsAppliedToConfiguration() {
+		this.context = new AnnotationConfigApplicationContext();
+		EnvironmentTestUtils.addEnvironment(this.context, "spring.data.mongodb.port=0",
+				"spring.mongodb.embedded.storage.oplogSize=10");
+		this.context.register(EmbeddedMongoAutoConfiguration.class,
+				MongoClientConfiguration.class,
+				PropertyPlaceholderAutoConfiguration.class);
+		this.context.refresh();
+		assertThat(this.context.getBean(IMongodConfig.class).replication().getOplogSize())
+				.isEqualTo(10);
+	}
+
+	@Test
+	public void customReplicaSetNameIsAppliedToConfiguration() {
+		this.context = new AnnotationConfigApplicationContext();
+		EnvironmentTestUtils.addEnvironment(this.context, "spring.data.mongodb.port=0",
+				"spring.mongodb.embedded.storage.replSetName=testing");
+		this.context.register(EmbeddedMongoAutoConfiguration.class,
+				MongoClientConfiguration.class,
+				PropertyPlaceholderAutoConfiguration.class);
+		this.context.refresh();
+		assertThat(
+				this.context.getBean(IMongodConfig.class).replication().getReplSetName())
+						.isEqualTo("testing");
 	}
 
 	private void assertVersionConfiguration(String configuredVersion,

spring-boot-docs/src/main/asciidoc/appendix-application-properties.adoc

@@ -730,6 +730,9 @@ content into your application; rather pick only the properties that you need.
 
 	# EMBEDDED MONGODB ({sc-spring-boot-autoconfigure}/mongo/embedded/EmbeddedMongoProperties.{sc-ext}[EmbeddedMongoProperties])
 	spring.mongodb.embedded.features=SYNC_DELAY # Comma-separated list of features to enable.
+	spring.mongodb.embedded.storage.databaseDir= # Directory used for data storage.
+	spring.mongodb.embedded.storage.oplogSize= # Maximum size of the oplog in megabytes.
+	spring.mongodb.embedded.storage.replSetName= # Name of the replica set.
 	spring.mongodb.embedded.version=2.6.10 # Version of Mongo to use.
 
 	# REDIS ({sc-spring-boot-autoconfigure}/redis/RedisProperties.{sc-ext}[RedisProperties])