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
|
(function() {
if (typeof getToolTest === 'undefined') {
load('jstests/configs/plain_28.config.js');
}
// Tests using mongorestore to restore only a subset of a dump (either a
// single db or a single collection) from a larger dump.
jsTest.log('Testing restoration of a subset of a dump');
var toolTest = getToolTest('partial_restore');
var commonToolArgs = getCommonToolArguments();
// where we'll put the dump
var dumpTarget = 'partial_restore_dump';
resetDbpath(dumpTarget);
// we'll insert data into three collections spread across two dbs
var dbOne = toolTest.db.getSiblingDB('dbOne');
var dbTwo = toolTest.db.getSiblingDB('dbTwo');
var collOne = dbOne.collOne;
var collTwo = dbOne.collTwo;
var collThree = dbTwo.collThree;
// insert a bunch of data
var dataOne = [];
var dataTwo = [];
var dataThree = [];
for (var i = 0; i < 50; i++) {
collOne.insert({_id: i+'_collOne'});
collTwo.insert({_id: i+'_collTwo'});
collThree.insert({_id: i+'_collThree'});
}
collOne.insertMany(dataOne);
collTwo.insertMany(dataTwo);
collThree.insertMany(dataThree);
// sanity check the insertion worked
assert.eq(50, collOne.count());
assert.eq(50, collTwo.count());
assert.eq(50, collThree.count());
// dump the data
var ret = toolTest.runTool.apply(toolTest, ['dump']
.concat(getDumpTarget(dumpTarget))
.concat(commonToolArgs));
assert.eq(0, ret);
// drop the databases
dbOne.dropDatabase();
dbTwo.dropDatabase();
// restore a single db
ret = toolTest.runTool.apply(toolTest, ['restore', '--db', 'dbOne']
.concat(getRestoreTarget(dumpTarget+'/dbOne'))
.concat(commonToolArgs));
assert.eq(0, ret);
// make sure the restore worked, and nothing else but that db was restored
assert.eq(50, collOne.count());
assert.eq(50, collTwo.count());
assert.eq(0, collThree.count());
// drop the data
dbOne.dropDatabase();
// restore a single collection
ret = toolTest.runTool.apply(toolTest, ['restore',
'--db', 'dbOne',
'--collection', 'collTwo']
.concat(getRestoreTarget(dumpTarget+'/dbOne/collTwo.bson'))
.concat(commonToolArgs));
assert.eq(0, ret);
// make sure the restore worked, and nothing else but that collection was restored
assert.eq(0, collOne.count());
assert.eq(50, collTwo.count());
assert.eq(0, collThree.count());
// success
toolTest.stop();
}());
|