forked from mikhail-angelov/mongo-unit
-
Notifications
You must be signed in to change notification settings - Fork 0
/
index.js
163 lines (147 loc) · 3.85 KB
/
index.js
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
'use strict';
const Debug = require('debug')
const portfinder = require('portfinder')
const client = require('mongodb').MongoClient
const fs = require('fs')
const ps = require('ps-node')
const debug = Debug('mongo-unit')
const dataFolder = '/.mongo-unit'
const defaultTempDir = __dirname + dataFolder
const defaultMongoOpts = {
dbName:'test',
dbpath: defaultTempDir,
port: 27017,
}
var mongodHelper
var dbUrl
function runMogo(opts, port){
const MongodHelper = require('mongodb-prebuilt').MongodHelper
opts.port = port
mongodHelper = new MongodHelper(['--port', port, '--dbpath', opts.dbpath,'--storageEngine', 'ephemeralForTest']);
return mongodHelper.run()
.then(()=>{
dbUrl = 'mongodb://localhost:' + port+'/'+opts.dbName
debug(`mongo is started on ${dbUrl}`)
return dbUrl
})
}
function start(opts) {
const mongo_opts = Object.assign(defaultMongoOpts, (opts || {}))
if (dbUrl) {
return Promise.resolve(dbUrl)
} else {
makeSureTempDirExist(mongo_opts.dbpath)
return makeSureOtherMongoProcessesKilled(mongo_opts.dbpath)
.then(()=>getFreePort(mongo_opts.port))
.then(port => runMogo(mongo_opts, port))
}
}
function delay(time){
return new Promise(resolve=>setTimeout(resolve, time))
}
function stop(){
mongodHelper && mongodHelper.mongoBin.childProcess.kill()
dbUrl = null;
return delay(100) //this is small delay to make sure kill signal is sent
}
function getUrl() {
if(dbUrl){
return dbUrl
}else{
throw new Error('Please start mongo-unit firstm then use this API')
}
}
function load(data) {
return client.connect(getUrl())
.then(db => {
const queries = Object.keys(data).map(col => {
const collection = db.collection(col)
return collection.insert(data[col])
})
return Promise.all(queries).then(() => db.close())
})
}
function clean(data) {
return client.connect(getUrl())
.then(db => {
const queries = Object.keys(data).map(col => {
const collection = db.collection(col)
return collection.drop()
})
return Promise.all(queries).then(() => db.close())
})
}
function drop() {
return client.connect(getUrl())
.then(db => db.dropDatabase().then(() => db.close()))
}
function getFreePort(possiblePort) {
portfinder.basePort = possiblePort
return new Promise((resolve, reject) => portfinder.getPort((err, port) => {
if (err) {
debug(`cannot get free port: ${err}`)
reject(err)
} else {
resolve(port)
}
}))
}
function makeSureTempDirExist(dir) {
try {
fs.mkdirSync(dir);
} catch (e) {
if (e.code !== "EEXIST") {
console.log('cannot create db folder', dir, e)
throw e;
}
}
}
function makeSureOtherMongoProcessesKilled(dataFolder){
return new Promise((resolve, reject)=>{
ps.lookup({
psargs:['-A'],
command: 'mongod',
arguments: dataFolder
}, (err, resultList ) => {
if (err) {
console.log('ps-node error', err)
return reject( err )
}
resultList.forEach(process =>{
if( process ){
console.log( 'KILL PID: %s, COMMAND: %s, ARGUMENTS: %s', process.pid, process.command, process.arguments )
ps.kill(process.pid)
}
});
return resolve()
})
})
}
function initDb(url, data) {
return client.connect(url)
.then(db => {
const requests = Object.keys(data).map(col => {
const collection = db.collection(col)
return collection.insert(data[col])
})
return Promise.all(requests)
})
}
function dropDb(url) {
return client.connect(url)
.then(db => db.collections())
.then(collections => {
const requests = collections.map(col => col.drop())
return Promise.all(requests)
})
}
module.exports = {
start,
stop,
getUrl,
load,
clean,
drop,
initDb,
dropDb,
}