1
0
Fork 0
arangodb/tests/js/server/shell/shell-collection-properties...

269 lines
9.3 KiB
JavaScript

/*global describe, it, ArangoAgency, beforeEach, afterEach */
////////////////////////////////////////////////////////////////////////////////
/// @brief cluster collection creation tests
///
/// @file
///
/// DISCLAIMER
///
/// Copyright 2010-2012 triagens GmbH, Cologne, Germany
///
/// Licensed under the Apache License, Version 2.0 (the "License");
/// you may not use this file except in compliance with the License.
/// You may obtain a copy of the License at
///
/// http://www.apache.org/licenses/LICENSE-2.0
///
/// Unless required by applicable law or agreed to in writing, software
/// distributed under the License is distributed on an "AS IS" BASIS,
/// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
/// See the License for the specific language governing permissions and
/// limitations under the License.
///
/// Copyright holder is triAGENS GmbH, Cologne, Germany
///
/// @author Andreas Streichardt
/// @author Copyright 2017, ArangoDB GmbH, Cologne, Germany
////////////////////////////////////////////////////////////////////////////////
'use strict';
const expect = require('chai').expect;
const internal = require("internal");
const errors = require('@arangodb').errors;
const db = require("@arangodb").db;
const cn1 = "UnitTestPropertiesLeader";
const cn2 = "UnitTestPropertiesFollower";
// check whether all shards have the right amount of followers
function checkReplicationFactor(name, fac) {
// first we need the plan id of the collection
let plan = ArangoAgency.get('Plan/Collections/_system');
let collectionId = Object.values(plan.arango.Plan.Collections['_system']).reduce((result, collectionDef) => {
if (result) {
return result;
}
if (collectionDef.name === name) {
return collectionDef.id;
}
}, undefined);
for (let i = 0; i < 120; i++) {
let current = ArangoAgency.get('Current/Collections/_system');
let shards = Object.values(current.arango.Current.Collections['_system'][collectionId]);
let finished = 0;
shards.forEach(entry => {
finished += entry.servers.length === fac ? 1 : 0;
});
if (shards.length > 0 && finished === shards.length) {
return;
}
internal.sleep(0.5);
}
let current = ArangoAgency.get('Current/Collections/_system');
let val = current.arango.Current.Collections['_system'][collectionId];
expect(true).to.equal(false, "Expected replicationFactor of " + fac + " in collection "
+ name + " is not reflected properly in " +
"/Current/Collections/_system/" + collectionId + ": "+ JSON.stringify(val));
};
describe('Update collection properties', function() {
beforeEach(function() {
db._useDatabase("_system");
});
afterEach(function() {
db._useDatabase("_system");
try {
db._drop(cn1);
} catch (e) {}
});
it('increase replication factor ', function() {
db._create(cn1, {replicationFactor: 1, numberOfShards: 2}, {waitForSyncReplication: true});
checkReplicationFactor(cn1, 1);
const coll = db._collection(cn1);
let props = coll.properties({replicationFactor: 2});
expect(props.replicationFactor).to.equal(2);
checkReplicationFactor(cn1, 2);
});
it('decrease replication factor ', function() {
db._create(cn1, {replicationFactor: 2, numberOfShards: 2}, {waitForSyncReplication: true});
checkReplicationFactor(cn1, 2);
const coll = db._collection(cn1);
let props = coll.properties({replicationFactor: 1});
expect(props.replicationFactor).to.equal(1);
checkReplicationFactor(cn1, 1);
});
it('invalid replication factor', function() {
db._create(cn1, {replicationFactor: 2, numberOfShards: 2}, {waitForSyncReplication: true});
checkReplicationFactor(cn1, 2);
try {
const coll = db._collection(cn1);
coll.properties({replicationFactor: -1});
expect(false.replicationFactor).to.equal(true,
"Was able to update replicationFactor of follower");
} catch(e) {
expect(e.errorNum).to.equal(errors.ERROR_BAD_PARAMETER.code);
}
try {
const coll = db._collection(cn1);
coll.properties({replicationFactor: 100});
expect(false.replicationFactor).to.equal(true,
"Was able to update replicationFactor of follower");
} catch(e) {
expect(e.errorNum).to.equal(errors.ERROR_CLUSTER_INSUFFICIENT_DBSERVERS.code);
}
try {
const coll = db._collection(cn1);
coll.properties({replicationFactor: "satellite"});
expect(false.replicationFactor).to.equal(true,
"Was able to update replicationFactor of follower");
} catch(e) {
expect(e.errorNum).to.equal(errors.ERROR_FORBIDDEN.code);
}
});
});
describe('Update collection properties with distributeShardsLike, ', function() {
beforeEach(function() {
db._useDatabase("_system");
});
afterEach(function() {
db._useDatabase("_system");
try {
db._drop(cn2);
} catch (e) {}
try {
db._drop(cn1);
} catch (e) {}
});
it('increase replication factor', function() {
db._create(cn1, {replicationFactor: 1, numberOfShards: 2}, {waitForSyncReplication: true});
db._create(cn2, {distributeShardsLike: cn1}, {waitForSyncReplication: true});
checkReplicationFactor(cn1, 1);
checkReplicationFactor(cn2, 1);
const leader = db._collection(cn1);
let props = leader.properties({replicationFactor: 2});
expect(props.replicationFactor).to.equal(2);
checkReplicationFactor(cn1, 2);
checkReplicationFactor(cn2, 2);
});
it('decrease replication factor', function() {
db._create(cn1, {replicationFactor: 2, numberOfShards: 2}, {waitForSyncReplication: true});
db._create(cn2, {distributeShardsLike: cn1}, {waitForSyncReplication: true});
checkReplicationFactor(cn1, 2);
checkReplicationFactor(cn2, 2);
const leader = db._collection(cn1);
let props = leader.properties({replicationFactor: 1});
expect(props.replicationFactor).to.equal(1);
checkReplicationFactor(cn1, 1);
checkReplicationFactor(cn2, 1);
});
it('change replicationFactor of follower', function() {
db._create(cn1, {replicationFactor: 2, numberOfShards: 2}, {waitForSyncReplication: true});
db._create(cn2, {distributeShardsLike: cn1}, {waitForSyncReplication: true});
checkReplicationFactor(cn1, 2);
checkReplicationFactor(cn2, 2);
try {
const follower = db._collection(cn2);
follower.properties({replicationFactor: 1});
expect(false.replicationFactor).to.equal(true,
"Was able to update replicationFactor of follower");
} catch(e) {
expect(e.errorNum).to.equal(errors.ERROR_FORBIDDEN.code);
}
});
});
describe('Replication factor constraints', function() {
beforeEach(function() {
db._useDatabase("_system");
});
afterEach(function() {
db._useDatabase("_system");
try {
// must be dropped first because cn1 is prototype for this collection
// and can only be dropped if all dependent collections are dropped first.
db._drop(cn2);
} catch (e) {}
try {
db._drop(cn1);
} catch (e) {}
});
it('should not allow to create a collection with more replicas than dbservers available', function() {
try {
db._create(cn1, {replicationFactor: 5});
throw new Error('Should not reach this');
} catch (e) {
expect(e.errorNum).to.equal(errors.ERROR_CLUSTER_INSUFFICIENT_DBSERVERS.code);
}
});
it('should allow to create a collection with more replicas than dbservers when explicitly requested', function() {
db._create(cn1, {replicationFactor: 5}, {enforceReplicationFactor: false});
});
it('check replication factor of system collections', function() {
["_appbundles", "_apps", "_aqlfunctions", "_frontend", "_graphs",
"_jobs", "_modules", "_queues", "_routing",
"_statistics" , "_statistics15" , "_statisticsRaw" ,"_users"
].forEach(name => {
if(name === "_users"){
expect(db[name].properties()['replicationFactor']).to.equal(2);
} else if(db[name]){
expect(db[name].properties()['replicationFactor']).to.equal(2);
expect(db[name].properties()['distributeShardsLike']).to.equal("_users");
}
});
});
it('distributeShardsLike should ignore additional parameters', function() {
db._create(cn1, {replicationFactor: 2, numberOfShards: 2}, {waitForSyncReplication: true});
db._create(cn2, {distributeShardsLike: cn1, replicationFactor: 5, numberOfShards: 99}, {waitForSyncReplication: true});
expect(db[cn1].properties()['replicationFactor']).to.equal(db[cn2].properties()['replicationFactor']);
expect(db[cn1].properties()['numberOfShards']).to.equal(db[cn2].properties()['numberOfShards']);
expect(db[cn2].properties()['distributeShardsLike']).to.equal(cn1);
});
});