MeshCentral/MeshAgent
1
0
Fork 0
mirror of https://github.com/Ylianst/MeshAgent synced 2025-12-23 11:43:26 +00:00
Code Issues Releases Wiki Activity

Add support for local tests

Browse Source
This commit is contained in:
Bryan Roe
2022-09-14 23:58:42 -07:00
parent f4d6ee468d
commit 81c3253e05
1 changed files with 64 additions and 39 deletions
Download Patch File Download Diff File Expand all files Collapse all files

103
test/self-test.js
View File

@@ -23,6 +23,8 @@ var updateSource = null;
var promise = require('promise'); var promise = require('promise');
if (process.platform == 'win32') { global.kernel32 = require('_GenericMarshal').CreateNativeProxy('kernel32.dll'); global.kernel32.CreateMethod('GetCurrentProcess'); global.kernel32.CreateMethod('GetProcessHandleCount'); } if (process.platform == 'win32') { global.kernel32 = require('_GenericMarshal').CreateNativeProxy('kernel32.dll'); global.kernel32.CreateMethod('GetCurrentProcess'); global.kernel32.CreateMethod('GetProcessHandleCount'); }
var localOnly = false;
function getHandleCount() function getHandleCount()
{ {
@@ -162,6 +164,7 @@ if (process.argv.getParameter('help') != null)
console.log(' --Digest If specified, individually runs the HTTP Digest tests.'); console.log(' --Digest If specified, individually runs the HTTP Digest tests.');
console.log(' --FileTransfer If specified, individually runs the FileTransfer Unit Test'); console.log(' --FileTransfer If specified, individually runs the FileTransfer Unit Test');
console.log(' --KVM If specified, individually runs the KVM tests'); console.log(' --KVM If specified, individually runs the KVM tests');
console.log(' --LocalTests If specified, runs the tests that are able to run locally');
console.log(' --LocalDebug Specifies a port number for the Local Web Debug Interface'); console.log(' --LocalDebug Specifies a port number for the Local Web Debug Interface');
console.log(' --PrivacyBar If specified, causes the agent to spawn a privacy bar'); console.log(' --PrivacyBar If specified, causes the agent to spawn a privacy bar');
console.log(" --RemoteDebug Specifies a port number for the Agent's Web Debug Interface"); console.log(" --RemoteDebug Specifies a port number for the Agent's Web Debug Interface");
@@ -172,19 +175,26 @@ if (process.argv.getParameter('help') != null)
console.log(''); console.log('');
process.exit(); process.exit();
} }
if (process.argv.getParameter('AgentsFolder') == null)
localOnly = process.argv.getParameter('LocalTests') != null;
if (!localOnly)
{ {
console.log('\nRequired parameter: AgentsFolder, was not specified.'); if (process.argv.getParameter('AgentsFolder') == null)
process.exit();
}
else
{
if(!require('fs').existsSync(process.argv.getParameter('AgentsFolder')))
{ {
console.log('\nThe specified folder does not exist: ' + process.argv.getParameter('AgentsFolder')); console.log('\nRequired parameter: AgentsFolder, was not specified.');
process.exit(); process.exit();
} }
else
{
if (!require('fs').existsSync(process.argv.getParameter('AgentsFolder')))
{
console.log('\nThe specified folder does not exist: ' + process.argv.getParameter('AgentsFolder'));
process.exit();
}
}
} }
if (process.argv.getParameter('LocalDebug') != null) if (process.argv.getParameter('LocalDebug') != null)
{ {
try try
@@ -286,16 +296,23 @@ if (remoteDebug > 0)
process.stdout.write('Remote WebDebug will listen on port: ' + remoteDebug + '\n'); process.stdout.write('Remote WebDebug will listen on port: ' + remoteDebug + '\n');
} }
process.stdout.write('Generating Certificate...'); var cert;
var cert = require('tls').generateCertificate('test', { certType: 2, noUsages: 1 }); var server = { on: function () { } };
var server = require('https').createServer({ pfx: cert, passphrase: 'test' }); var loadedCert;
server.listen(); var der;
process.stdout.write('\rGenerating Certificate... [DONE]\n'); if (!localOnly)
{
process.stdout.write('Generating Certificate...');
cert = require('tls').generateCertificate('test', { certType: 2, noUsages: 1 });
server = require('https').createServer({ pfx: cert, passphrase: 'test' });
server.listen();
process.stdout.write('\rGenerating Certificate... [DONE]\n');
var loadedCert = require('tls').loadCertificate({ pfx: cert, passphrase: 'test' }); loadedCert = require('tls').loadCertificate({ pfx: cert, passphrase: 'test' });
var der = loadedCert.toDER(); der = loadedCert.toDER();
global._test = []; global._test = [];
}
if (process.argv.getParameter('NoInstall') != null) if (process.argv.getParameter('NoInstall') != null)
{ {
@@ -956,7 +973,7 @@ function fs_test()
promises.fs_1.then(function (r) promises.fs_1.then(function (r)
{ {
process.stdout.write('\r fs.readFileSync().................................[OK] \n'); process.stdout.write('\r fs.readFileSync().................................[OK] \n');
process.stdout.write(' => Handle Counts (Pre/Post): ' + r.reason.pre + '/' + r.reason.post + '\n'); //process.stdout.write(' => Handle Counts (Pre/Post): ' + r.reason.pre + '/' + r.reason.post + '\n');
}).then(function () }).then(function ()
{ {
var tmp = generateRandomString(5); var tmp = generateRandomString(5);
@@ -975,7 +992,7 @@ function fs_test()
{ {
process.stdout.write('\r fs.writeFileSync()................................[OK] \n'); process.stdout.write('\r fs.writeFileSync()................................[OK] \n');
process.stdout.write(' fs.existsSync()...................................[OK] \n'); process.stdout.write(' fs.existsSync()...................................[OK] \n');
process.stdout.write(' => Handle Counts (Pre/Post): ' + r.reason.pre + '/' + r.reason.post + '\n'); //process.stdout.write(' => Handle Counts (Pre/Post): ' + r.reason.pre + '/' + r.reason.post + '\n');
}).then(function () }).then(function ()
{ {
process.stdout.write(' fs.renameSync()...................................[WAITING]'); process.stdout.write(' fs.renameSync()...................................[WAITING]');
@@ -990,7 +1007,7 @@ function fs_test()
promises.fs_3.then(function (r) promises.fs_3.then(function (r)
{ {
process.stdout.write('\r fs.renameSync()...................................[OK] \n'); process.stdout.write('\r fs.renameSync()...................................[OK] \n');
process.stdout.write(' => Handle Counts (Pre/Post): ' + r.reason.pre + '/' + r.reason.post + '\n'); //process.stdout.write(' => Handle Counts (Pre/Post): ' + r.reason.pre + '/' + r.reason.post + '\n');
ret.resolve(); ret.resolve();
}).catch(function () }).catch(function ()
@@ -1898,25 +1915,33 @@ if (process.argv.getParameter('verbose') != null)
console.setInfoLevel(parseInt(process.argv.getParameter('verbose'))); console.setInfoLevel(parseInt(process.argv.getParameter('verbose')));
} }
if (process.argv.getParameter('NoInstall') == null) if (!localOnly)
{ {
// if (process.argv.getParameter('NoInstall') == null)
// Start by installing agent as service {
// //
var params = ['--__skipExit=1', '--logUpdate=1', '--meshServiceName=TestAgent']; // Start by installing agent as service
var options = //
{ var params = ['--__skipExit=1', '--logUpdate=1', '--meshServiceName=TestAgent'];
files: var options =
[ {
{ files:
newName: (process.platform == 'win32' ? 'MeshAgent.msh' : 'meshagent.msh'), [
_buffer: 'enableILibRemoteLogging=5556\nlogUpdate=1\nMeshID=0x43FEF862BF941B2BBE5964CC7CA02573BBFB94D5A717C5AA3FC103558347D0BE26840ACBD30FFF981F7F5A2083D0DABC\nMeshServer=wss://127.0.0.1:' + server.address().port + '/agent.ashx\nmeshServiceName=TestAgent\nServerID=' + loadedCert.getKeyHash().toString('hex') {
} newName: (process.platform == 'win32' ? 'MeshAgent.msh' : 'meshagent.msh'),
], _buffer: 'enableILibRemoteLogging=5556\nlogUpdate=1\nMeshID=0x43FEF862BF941B2BBE5964CC7CA02573BBFB94D5A717C5AA3FC103558347D0BE26840ACBD30FFF981F7F5A2083D0DABC\nMeshServer=wss://127.0.0.1:' + server.address().port + '/agent.ashx\nmeshServiceName=TestAgent\nServerID=' + loadedCert.getKeyHash().toString('hex')
binary: updateSource, }
noParams: true ],
}; binary: updateSource,
require('agent-installer').fullInstallEx(params, options); noParams: true
console.setDestination(console.Destinations.STDOUT); };
require('agent-installer').fullInstallEx(params, options);
console.setDestination(console.Destinations.STDOUT);
}
console.log('\nWaiting for Agent Connection...');
} }
console.log('\nWaiting for Agent Connection...');
if (localOnly)
{
console.log('Running Local Tests');
}