133 lines
4.4 KiB
JavaScript
133 lines
4.4 KiB
JavaScript
/*
|
|
* This Source Code Form is subject to the terms of the Mozilla Public
|
|
* License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
* file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
|
*/
|
|
|
|
/*
|
|
* Copyright 2016, Joyent, Inc.
|
|
*/
|
|
|
|
/*
|
|
* Test affinity/locality hints with `triton create -a RULE ...`.
|
|
*
|
|
* This is really only testable against a DC with multiple target CNs (i.e.
|
|
* COAL is out), and even then it is hard to test more than just basic cases
|
|
* without knowing some details about the CN provisioning pool.
|
|
*/
|
|
|
|
var format = require('util').format;
|
|
var os = require('os');
|
|
var test = require('tape');
|
|
var vasync = require('vasync');
|
|
|
|
var common = require('../../lib/common');
|
|
var h = require('./helpers');
|
|
|
|
|
|
// --- globals
|
|
|
|
var ALIAS_PREFIX = format('nodetritontest-affinity-%s', os.hostname());
|
|
|
|
var testOpts = {
|
|
skip: !h.CONFIG.allowWriteActions || h.CONFIG.skipAffinityTests
|
|
};
|
|
|
|
|
|
// --- Tests
|
|
|
|
test('affinity (triton create -a RULE ...)', testOpts, function (tt) {
|
|
tt.comment('Add \'"skipAffinityTests":true\' to test/config.json if ' +
|
|
'this target DC does not have multiple provisionable CNs (e.g. COAL).');
|
|
|
|
// TODO: `triton rm -f` would be helpful for this
|
|
tt.test(' setup: rm existing insts ' + ALIAS_PREFIX + '*', function (t) {
|
|
// Cheat and use the current SNAFU behaviour that 'name=foo' matches
|
|
// all VMs *prefixed* with "foo".
|
|
h.safeTriton(t, ['inst', 'list', '-j', 'name='+ALIAS_PREFIX],
|
|
function (err, stdout) {
|
|
var instsToRm = h.jsonStreamParse(stdout);
|
|
if (instsToRm.length === 0) {
|
|
t.end();
|
|
return;
|
|
}
|
|
var rmCmd = ['inst', 'rm', '-w'].concat(
|
|
instsToRm.map(function (i) { return i.id; }));
|
|
h.safeTriton(t, rmCmd, function () {
|
|
t.ok(true, rmCmd.join(' '));
|
|
t.end();
|
|
});
|
|
});
|
|
});
|
|
|
|
var imgId;
|
|
tt.test(' setup: find test image', function (t) {
|
|
h.getTestImg(t, function (err, imgId_) {
|
|
t.ifError(err, 'getTestImg' + (err ? ': ' + err : ''));
|
|
imgId = imgId_;
|
|
t.end();
|
|
});
|
|
});
|
|
|
|
var pkgId;
|
|
tt.test(' setup: find test package', function (t) {
|
|
h.getTestPkg(t, function (err, pkgId_) {
|
|
t.ifError(err, 'getTestPkg' + (err ? ': ' + err : ''));
|
|
pkgId = pkgId_;
|
|
t.end();
|
|
});
|
|
});
|
|
|
|
var db0Alias = ALIAS_PREFIX + '-db0';
|
|
var db0;
|
|
tt.test(' setup: triton create -n db0', function (t) {
|
|
var argv = ['create', '-wj', '-n', db0Alias, imgId, pkgId];
|
|
h.safeTriton(t, argv, function (err, stdout) {
|
|
var lines = h.jsonStreamParse(stdout);
|
|
db0 = lines[1];
|
|
t.end();
|
|
});
|
|
});
|
|
|
|
// Test db1 being put on same server as db0.
|
|
var db1Alias = ALIAS_PREFIX + '-db1';
|
|
var db1;
|
|
tt.test(' setup: triton create -n db1 -a db0', function (t) {
|
|
var argv = ['create', '-wj', '-n', db1Alias, '-a', db0Alias,
|
|
imgId, pkgId];
|
|
h.safeTriton(t, argv, function (err, stdout) {
|
|
var lines = h.jsonStreamParse(stdout);
|
|
db1 = lines[1];
|
|
t.equal(db0.compute_node, db1.compute_node,
|
|
format('inst %s landed on same CN as inst %s: %s',
|
|
db1Alias, db0Alias, db1.compute_node));
|
|
t.end();
|
|
});
|
|
});
|
|
|
|
// Test db2 being put on server *other* than db0.
|
|
var db2Alias = ALIAS_PREFIX + '-db2';
|
|
var db2;
|
|
tt.test(' setup: triton create -n db2 -a \'inst!=db0\'', function (t) {
|
|
var argv = ['create', '-wj', '-n', db2Alias, '-a', 'inst!='+db0Alias,
|
|
imgId, pkgId];
|
|
h.safeTriton(t, argv, function (err, stdout) {
|
|
var lines = h.jsonStreamParse(stdout);
|
|
db2 = lines[1];
|
|
t.notEqual(db0.compute_node, db2.compute_node,
|
|
format('inst %s landed on different CN (%s) as inst %s (%s)',
|
|
db2Alias, db2.compute_node, db0Alias, db0.compute_node));
|
|
t.end();
|
|
});
|
|
});
|
|
|
|
// Remove instances. Add a test timeout, because '-w' on delete doesn't
|
|
// have a way to know if the attempt failed or if it is just taking a
|
|
// really long time.
|
|
tt.test(' cleanup: triton rm', {timeout: 10 * 60 * 1000}, function (t) {
|
|
h.safeTriton(t, ['rm', '-w', db0.id, db1.id, db2.id], function () {
|
|
t.end();
|
|
});
|
|
});
|
|
});
|