mirror of
https://github.com/mongodb/mongo.git
synced 2024-11-30 00:56:44 +01:00
71 lines
2.2 KiB
JavaScript
71 lines
2.2 KiB
JavaScript
// tests for the traffic_recording commands.
|
|
(function() {
|
|
var baseName = "jstests_traffic_recording";
|
|
|
|
// Variables for this test
|
|
const recordingDir = MongoRunner.toRealDir("$dataDir/traffic_recording/");
|
|
const recordingFile = "recording.txt";
|
|
const recordingFilePath = MongoRunner.toRealDir(recordingDir + "/" + recordingFile);
|
|
|
|
// Create the recording directory if it does not already exist
|
|
mkdir(recordingDir);
|
|
|
|
// Create the options and run mongod
|
|
var opts = {auth: "", setParameter: "trafficRecordingDirectory=" + recordingDir};
|
|
m = MongoRunner.runMongod(opts);
|
|
|
|
// Get the port of the host
|
|
var serverPort = m.port;
|
|
|
|
// Set the readMode and writeMode to legacy
|
|
m.forceReadMode("legacy");
|
|
m.forceWriteMode("legacy");
|
|
|
|
// Create necessary users
|
|
adminDB = m.getDB("admin");
|
|
const testDB = m.getDB("test");
|
|
const coll = testDB.getCollection("foo");
|
|
adminDB.createUser({user: "admin", pwd: "pass", roles: jsTest.adminUserRoles});
|
|
adminDB.auth("admin", "pass");
|
|
|
|
// Start recording traffic
|
|
assert.commandWorked(adminDB.runCommand({'startRecordingTraffic': 1, 'filename': 'recording.txt'}));
|
|
|
|
// Run a few commands
|
|
testDB.runCommand({"serverStatus": 1});
|
|
coll.insert({"name": "foo biz bar"});
|
|
coll.findOne();
|
|
coll.insert({"name": "foo bar"});
|
|
coll.findOne({"name": "foo bar"});
|
|
coll.deleteOne({});
|
|
|
|
// Stop recording traffic
|
|
assert.commandWorked(testDB.runCommand({'stopRecordingTraffic': 1}));
|
|
|
|
// Shutdown Mongod
|
|
MongoRunner.stopMongod(m, null, {user: 'admin', pwd: 'password'});
|
|
|
|
// Counters
|
|
var opCodes = {};
|
|
|
|
// Pass filepath to traffic_reader helper method to get recorded info in BSON
|
|
var res = convertTrafficRecordingToBSON(recordingFilePath);
|
|
|
|
// Iterate through the results and assert the above commands are properly recorded
|
|
res.forEach((obj) => {
|
|
opCodes[obj["rawop"]["header"]["opcode"]] =
|
|
(opCodes[obj["rawop"]["header"]["opcode"]] || 0) + 1;
|
|
assert.eq(obj["seenconnectionnum"], 1);
|
|
var responseTo = obj["rawop"]["header"]["responseto"];
|
|
if (responseTo == 0) {
|
|
assert.eq(obj["destendpoint"], serverPort.toString());
|
|
} else {
|
|
assert.eq(obj["srcendpoint"], serverPort.toString());
|
|
}
|
|
});
|
|
|
|
// ensure legacy operations worked properly
|
|
assert.eq(opCodes[2002], 2);
|
|
assert.eq(opCodes[2006], 1);
|
|
})();
|