Skip to content

Commit

Permalink
feat(config): Add instance.dbName option for providing custom db name
Browse files Browse the repository at this point in the history
  • Loading branch information
nodkz committed Aug 24, 2017
1 parent 99e3883 commit 94c14b3
Show file tree
Hide file tree
Showing 2 changed files with 12 additions and 5 deletions.
1 change: 1 addition & 0 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -47,6 +47,7 @@ All options are optional.
const mongod = new MongodbMemoryServer({
instance: {
port?: ?number, // by default choose any free port
dbName?: string, // by default generate random dbName
dbPath?: string, // by default create in temp directory
storageEngine?: string, // by default `ephemeralForTest`
debug?: boolean, // by default false
Expand Down
16 changes: 11 additions & 5 deletions src/MongoMemoryServer.js
Original file line number Diff line number Diff line change
Expand Up @@ -13,6 +13,7 @@ export type MongoMemoryServerOptsT = {
instance: {
port?: ?number,
dbPath?: string,
dbName?: string,
storageEngine?: string,
debug?: boolean | Function,
},
Expand All @@ -32,6 +33,7 @@ export type MongoMemoryServerOptsT = {
export type MongoInstanceDataT = {
port: number,
dbPath: string,
dbName: string,
uri: string,
storageEngine: string,
childProcess: ChildProcess,
Expand All @@ -41,9 +43,12 @@ export type MongoInstanceDataT = {
},
};

async function generateConnectionString(port: number, dbName: ?string): Promise<string> {
const db = dbName || (await uuid());
return `mongodb://localhost:${port}/${db}`;
async function generateDbName(dbName?: string): Promise<string> {
return dbName || uuid();
}

async function generateConnectionString(port: number, dbName: string): Promise<string> {
return `mongodb://localhost:${port}/${dbName}`;
}

export default class MongoMemoryServer {
Expand Down Expand Up @@ -106,7 +111,8 @@ export default class MongoMemoryServer {
data.port = await getport(instOpts.port);
this.debug = Debug(`Mongo[${data.port}]`);
this.debug.enabled = !!this.opts.debug;
data.uri = await generateConnectionString(data.port);
data.dbName = await generateDbName(instOpts.dbName);
data.uri = await generateConnectionString(data.port, data.dbName);
data.storageEngine = instOpts.storageEngine || 'ephemeralForTest';
if (instOpts.dbPath) {
data.dbPath = instOpts.dbPath;
Expand Down Expand Up @@ -172,7 +178,7 @@ export default class MongoMemoryServer {
return generateConnectionString(port, otherDbName);
}
// generate new random db name
return generateConnectionString(port);
return generateConnectionString(port, await generateDbName());
}

return uri;
Expand Down

0 comments on commit 94c14b3

Please sign in to comment.