Skip to content

...will provide a platform neutral way for running mongodb in unittests.

License

Notifications You must be signed in to change notification settings

Don-CA/de.flapdoodle.embed.mongo

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Organization Flapdoodle OSS

Build Status Maven Central

We are now a github organization. You are invited to participate. Starting with version 2 we are going to support only java 8 or higher. If you are looking for the older version you can find it in the 1.7 branch.

Embedded MongoDB

Embedded MongoDB will provide a platform neutral way for running mongodb in unittests.

Why?

  • dropping databases causing some pains (often you have to wait long time after each test)
  • its easy, much easier as installing right version by hand
  • you can change version per test

How?

  • download mongodb (and cache it)
  • extract it (and cache it)
  • java uses its process api to start and monitor the mongo process
  • you run your tests
  • java kills the mongo process

License

We use http://www.apache.org/licenses/LICENSE-2.0

We need your help?

Poll: Which MongoDB version should stay supported?

Dependencies

Build on top of

Other ways to use Embedded MongoDB

Comments about Embedded MongoDB in the Wild

Other MongoDB Stuff

Backward binary compatibility and API changes

There is a report on backward binary compatibility and API changes for the library: https://abi-laboratory.pro/java/tracker/timeline/de.flapdoodle.embed.mongo/ -> thanks @lvc

Howto

Maven

Snapshots (Repository http://oss.sonatype.org/content/repositories/snapshots)

<dependency>
	<groupId>de.flapdoodle.embed</groupId>
	<artifactId>de.flapdoodle.embed.mongo</artifactId>
	<version>3.0.1-SNAPSHOT</version>
</dependency>

Gradle

Make sure you have mavenCentral() in your repositories or that your enterprise/local server proxies the maven central repository.

dependencies {
	testCompile group: "de.flapdoodle.embed", name: "de.flapdoodle.embed.mongo", version: "3.0.1-SNAPSHOT"
}

Build from source

When you fork or clone our branch you should always be able to build the library by running

mvn package

Changelog

Changelog

Supported Versions

Versions: some older, a stable and a development version Support for Linux, Windows and MacOSX.

Usage

MongodStarter starter = MongodStarter.getDefaultInstance();

int port = Network.getFreeServerPort();
MongodConfig mongodConfig = MongodConfig.builder()
    .version(Version.Main.PRODUCTION)
    .net(new Net(port, Network.localhostIsIPv6()))
    .build();

MongodExecutable mongodExecutable = null;
try {
  mongodExecutable = starter.prepare(mongodConfig);
  MongodProcess mongod = mongodExecutable.start();

  try (MongoClient mongo = new MongoClient("localhost", port)) {
    DB db = mongo.getDB("test");
    DBCollection col = db.createCollection("testCol", new BasicDBObject());
    col.save(new BasicDBObject("testDoc", new Date()));
  }

} finally {
  if (mongodExecutable != null)
    mongodExecutable.stop();
}

Usage - Optimization

You should make the MongodStarter instance or the RuntimeConfig instance static (per Class or per JVM). The main purpose of that is the caching of extracted executables and library files. This is done by the ArtifactStore instance configured with the RuntimeConfig instance. Each instance uses its own cache so multiple RuntimeConfig instances will use multiple ArtifactStores an multiple caches with much less cache hits:)

Usage - custom mongod filename

If you do not restrict bindId to localhost you get windows firewall dialog popups. To avoid them you can choose a stable executable name with UserTempNaming. This way the firewall dialog only popups once. See Executable Collision

int port = Network.getFreeServerPort();

Command command = Command.MongoD;

RuntimeConfig runtimeConfig = Defaults.runtimeConfigFor(command)
.artifactStore(Defaults.extractedArtifactStoreFor(command)
    .withDownloadConfig(Defaults.downloadConfigFor(command).build())
    .executableNaming(new UserTempNaming()))
.build();

MongodConfig mongodConfig = MongodConfig.builder()
    .version(Version.Main.PRODUCTION)
    .net(new Net(port, Network.localhostIsIPv6()))
    .build();

MongodStarter runtime = MongodStarter.getInstance(runtimeConfig);

MongodExecutable mongodExecutable = null;
try {
  mongodExecutable = runtime.prepare(mongodConfig);
  MongodProcess mongod = mongodExecutable.start();

  try (MongoClient mongo = new MongoClient("localhost", port)) {
    DB db = mongo.getDB("test");
    DBCollection col = db.createCollection("testCol", new BasicDBObject());
    col.save(new BasicDBObject("testDoc", new Date()));
  }

} finally {
  if (mongodExecutable != null)
    mongodExecutable.stop();
}

Unit Tests

public abstract class AbstractMongoDBTest extends TestCase {


  /**
   * please store Starter or RuntimeConfig in a static final field
   * if you want to use artifact store caching (or else disable caching) 
   */
  private static final MongodStarter starter = MongodStarter.getDefaultInstance();

  private MongodExecutable _mongodExe;
  private MongodProcess _mongod;

  private MongoClient _mongo;
  private int port;
  
  @Override
  protected void setUp() throws Exception {
    port = Network.getFreeServerPort();
    _mongodExe = starter.prepare(createMongodConfig());
    _mongod = _mongodExe.start();

    super.setUp();

    _mongo = new MongoClient("localhost", port);
  }
  
  public int port() {
    return port;
  }

  protected IMongodConfig createMongodConfig() throws UnknownHostException, IOException {
    return createMongodConfigBuilder().build();
  }

  protected MongodConfigBuilder createMongodConfigBuilder() throws UnknownHostException, IOException {
    return new MongodConfigBuilder()
      .version(Version.Main.PRODUCTION)
      .net(new Net(port, Network.localhostIsIPv6()));
  }

  @Override
  protected void tearDown() throws Exception {
    super.tearDown();

    _mongod.stop();
    _mongodExe.stop();
  }

  public Mongo getMongo() {
    return _mongo;
  }

}

... with some more help

MongodForTestsFactory factory = null;
try {
  factory = MongodForTestsFactory.with(Version.Main.PRODUCTION);

  try (MongoClient mongo = factory.newMongo()) {
    DB db = mongo.getDB("test-" + UUID.randomUUID());
    DBCollection col = db.createCollection("testCol", new BasicDBObject());
    col.save(new BasicDBObject("testDoc", new Date()));
  }

} finally {
  if (factory != null)
    factory.shutdown();
}

Customize Download URL

Command command = Command.MongoD;

RuntimeConfig runtimeConfig = Defaults.runtimeConfigFor(command)
    .artifactStore(Defaults.extractedArtifactStoreFor(command)
        .withDownloadConfig(Defaults.downloadConfigFor(command)
            .downloadPath((__) -> "http://my.custom.download.domain/")
            .build()))
    .build();

Customize Proxy for Download

Command command = Command.MongoD;

RuntimeConfig runtimeConfig = Defaults.runtimeConfigFor(command)
    .artifactStore(Defaults.extractedArtifactStoreFor(command)
        .withDownloadConfig(Defaults.downloadConfigFor(command)
            .proxyFactory(new HttpProxyFactory("fooo", 1234))
            .build()))
    .build();

Customize Artifact Storage

Directory artifactStorePath = new FixedPath(System.getProperty("user.home") + "/.embeddedMongodbCustomPath");
TempNaming executableNaming = new UUIDTempNaming();

Command command = Command.MongoD;    

RuntimeConfig runtimeConfig = Defaults.runtimeConfigFor(command)
    .artifactStore(Defaults.extractedArtifactStoreFor(command)
        .withDownloadConfig(Defaults.downloadConfigFor(command)
            .artifactStorePath(artifactStorePath)
            .build())
        .executableNaming(executableNaming))
    .build();

MongodStarter runtime = MongodStarter.getInstance(runtimeConfig);
MongodExecutable mongodExe = runtime.prepare(mongodConfig);

Usage - custom mongod process output

... to console with line prefix

ProcessOutput processOutput = new ProcessOutput(Processors.namedConsole("[mongod>]"),
    Processors.namedConsole("[MONGOD>]"), Processors.namedConsole("[console>]"));

RuntimeConfig runtimeConfig = Defaults.runtimeConfigFor(Command.MongoD)
    .processOutput(processOutput)
    .build();

MongodStarter runtime = MongodStarter.getInstance(runtimeConfig);

... to file

...
StreamProcessor mongodOutput = Processors.named("[mongod>]",
    new FileStreamProcessor(File.createTempFile("mongod", "log")));
StreamProcessor mongodError = new FileStreamProcessor(File.createTempFile("mongod-error", "log"));
StreamProcessor commandsOutput = Processors.namedConsole("[console>]");

RuntimeConfig runtimeConfig = Defaults.runtimeConfigFor(Command.MongoD)
    .processOutput(new ProcessOutput(mongodOutput, mongodError, commandsOutput))
    .build();

MongodStarter runtime = MongodStarter.getInstance(runtimeConfig);
...

...
// ...
public class FileStreamProcessor implements StreamProcessor {

  private final FileOutputStream outputStream;

  public FileStreamProcessor(File file) throws FileNotFoundException {
    outputStream = new FileOutputStream(file);
  }

  @Override
  public void process(String block) {
    try {
      outputStream.write(block.getBytes());
    } catch (IOException e) {
      e.printStackTrace();
    }
  }

  @Override
  public void onProcessed() {
    try {
      outputStream.close();
    } catch (IOException e) {
      e.printStackTrace();
    }
  }

}
// ...
// <-
...

... to java logging

Logger logger = LoggerFactory.getLogger(getClass().getName());

ProcessOutput processOutput = new ProcessOutput(Processors.logTo(logger, Slf4jLevel.INFO), Processors.logTo(logger,
    Slf4jLevel.ERROR), Processors.named("[console>]", Processors.logTo(logger, Slf4jLevel.DEBUG)));


RuntimeConfig runtimeConfig = Defaults.runtimeConfigFor(Command.MongoD, logger)
    .processOutput(processOutput)
    .artifactStore(Defaults.extractedArtifactStoreFor(Command.MongoD)
        .download(Defaults.downloadConfigFor(Command.MongoD)
            .progressListener(new Slf4jProgressListener(logger))))
    .build();

MongodStarter runtime = MongodStarter.getInstance(runtimeConfig);

... to default java logging (the easy way)

Logger logger = LoggerFactory.getLogger(getClass().getName());

RuntimeConfig runtimeConfig = Defaults.runtimeConfigFor(Command.MongoD, logger)
    .build();

MongodStarter runtime = MongodStarter.getInstance(runtimeConfig);

... to null device

Logger logger = LoggerFactory.getLogger(getClass().getName());

RuntimeConfig runtimeConfig = Defaults.runtimeConfigFor(Command.MongoD, logger)
    .processOutput(ProcessOutput.getDefaultInstanceSilent())
    .build();

MongodStarter runtime = MongodStarter.getInstance(runtimeConfig);

Custom Version

int port = 12345;
MongodConfig mongodConfig = MongodConfig.builder()
    .version(Versions.withFeatures(de.flapdoodle.embed.process.distribution.Version.of("2.7.1"), Feature.SYNC_DELAY))
    .net(new Net(port, Network.localhostIsIPv6()))
    .build();

MongodStarter runtime = MongodStarter.getDefaultInstance();
MongodProcess mongod = null;

MongodExecutable mongodExecutable = null;
try {
  mongodExecutable = runtime.prepare(mongodConfig);
  mongod = mongodExecutable.start();

...


} finally {
  if (mongod != null) {
    mongod.stop();
  }
  if (mongodExecutable != null)
    mongodExecutable.stop();
}

Main Versions

IFeatureAwareVersion version = Version.V2_2_5;
// uses latest supported 2.2.x Version
version = Version.Main.V2_2;
// uses latest supported production version
version = Version.Main.PRODUCTION;
// uses latest supported development version
version = Version.Main.DEVELOPMENT;

Use Free Server Port

Warning: maybe not as stable, as expected.

... by hand

int port = Network.getFreeServerPort();

... automagic

MongodConfig mongodConfig = MongodConfig.builder().version(Version.Main.PRODUCTION).build();

MongodStarter runtime = MongodStarter.getDefaultInstance();

MongodExecutable mongodExecutable = null;
MongodProcess mongod = null;
try {
  mongodExecutable = runtime.prepare(mongodConfig);
  mongod = mongodExecutable.start();

  try (MongoClient mongo = new MongoClient(
      new ServerAddress(mongodConfig.net().getServerAddress(), mongodConfig.net().getPort()))) {
...

  }

} finally {
  if (mongod != null) {
    mongod.stop();
  }
  if (mongodExecutable != null)
    mongodExecutable.stop();
}

... custom timeouts

MongodConfig mongodConfig = MongodConfig.builder()
    .version(Version.Main.PRODUCTION)
    .timeout(new Timeout(30000))
    .build();

Command Line Post Processing

CommandLinePostProcessor postProcessor = new CommandLinePostProcessor() {
      @Override
      public List<String> process(Distribution distribution, List<String> args) {
        return null;
      }
    };
...

RuntimeConfig runtimeConfig = Defaults.runtimeConfigFor(Command.MongoD)
    .commandLinePostProcessor(postProcessor)
    .build();

Custom Command Line Options

We changed the syncDelay to 0 which turns off sync to disc. To turn on default value used defaultSyncDelay().

MongodConfig mongodConfig = MongodConfig.builder()
    .version(Version.Main.PRODUCTION)
    .cmdOptions(MongoCmdOptions.builder()
        .syncDelay(10)
        .useNoPrealloc(false)
        .useSmallFiles(false)
        .useNoJournal(false)
        .enableTextSearch(true)
        .build())
    .build();

Snapshot database files from temp dir

We changed the syncDelay to 0 which turns off sync to disc. To get the files to create an snapshot you must turn on default value (use defaultSyncDelay()).

MongodConfig mongodConfig = MongodConfig.builder()
    .version(Version.Main.PRODUCTION)
    .processListener(new CopyDbFilesFromDirBeforeProcessStop(destination))
    .cmdOptions(MongoCmdOptions.builder()
        .useDefaultSyncDelay(true)
        .build())
    .build();

Custom database directory

If you set a custom database directory, it will not be deleted after shutdown

Storage replication = new Storage("/custom/databaseDir",null,0);

MongodConfig mongodConfig = MongodConfig.builder()
    .version(Version.Main.PRODUCTION)
    .replication(replication)
    .build();

Start mongos with mongod instance

this is an very easy example to use mongos and mongod

  int port = 12121;
  int defaultConfigPort = 12345;
  String defaultHost = "localhost";

  MongodProcess mongod = startMongod(defaultConfigPort);

  try {
    MongosProcess mongos = startMongos(port, defaultConfigPort, defaultHost);
    try {
      MongoClient mongoClient = new MongoClient(defaultHost, defaultConfigPort);
      System.out.println("DB Names: " + mongoClient.getDatabaseNames());
    } finally {
      mongos.stop();
    }
  } finally {
    mongod.stop();
  }

  private MongosProcess startMongos(int port, int defaultConfigPort, String defaultHost) throws UnknownHostException,
      IOException {
    IMongosConfig mongosConfig = new MongosConfigBuilder()
      .version(Version.Main.PRODUCTION)
      .net(new Net(port, Network.localhostIsIPv6()))
      .configDB(defaultHost + ":" + defaultConfigPort)
      .build();

    MongosExecutable mongosExecutable = MongosStarter.getDefaultInstance().prepare(mongosConfig);
    MongosProcess mongos = mongosExecutable.start();
    return mongos;
  }

  private MongodProcess startMongod(int defaultConfigPort) throws UnknownHostException, IOException {
    IMongodConfig mongoConfigConfig = new MongodConfigBuilder()
      .version(Version.Main.PRODUCTION)
      .net(new Net(defaultConfigPort, Network.localhostIsIPv6()))
      .configServer(true)
      .build();

    MongodExecutable mongodExecutable = MongodStarter.getDefaultInstance().prepare(mongoConfigConfig);
    MongodProcess mongod = mongodExecutable.start();
    return mongod;
  }

Import JSON file with mongoimport command

int defaultConfigPort = Network.getFreeServerPort();
String database = "importTestDB";
String collection = "importedCollection";

MongodConfig mongoConfigConfig = MongodConfig.builder()
    .version(Version.Main.PRODUCTION)
    .net(new Net(defaultConfigPort, Network.localhostIsIPv6()))
    .build();

MongodExecutable mongodExecutable = MongodStarter.getDefaultInstance().prepare(mongoConfigConfig);
MongodProcess mongod = mongodExecutable.start();

try {
  MongoImportConfig mongoImportConfig = MongoImportConfig.builder()
      .version(Version.Main.PRODUCTION)
      .net(new Net(defaultConfigPort, Network.localhostIsIPv6()))
      .databaseName(database)
      .collectionName(collection)
      .isUpsertDocuments(true)
      .isDropCollection(true)
      .isJsonArray(true)
      .importFile(jsonFile)
      .build();

  MongoImportExecutable mongoImportExecutable = MongoImportStarter.getDefaultInstance().prepare(mongoImportConfig);
  MongoImportProcess mongoImport = mongoImportExecutable.start();
  try {
...

  }
  finally {
    mongoImport.stop();
  }
}
finally {
  mongod.stop();
}

Executable Collision

There is a good chance of filename collisions if you use a custom naming schema for the executable (see Usage - custom mongod filename). If you got an exception, then you should make your RuntimeConfig or MongoStarter class or jvm static (static final in your test class or singleton class for all tests).


YourKit is kindly supporting open source projects with its full-featured Java Profiler. YourKit, LLC is the creator of innovative and intelligent tools for profiling Java and .NET applications. Take a look at YourKit's leading software products: YourKit Java Profiler and YourKit .NET Profiler.

About

...will provide a platform neutral way for running mongodb in unittests.

http://flapdoodle-oss.github.io/de.flapdoodle.embed.mongo

Resources

License

Stars

Watchers

Forks

Packages

No packages published

Languages

  • Java 100.0%