mirror of https://gitee.com/bigwinds/arangodb
Merge branch 'devel' of https://github.com/arangodb/arangodb into devel
This commit is contained in:
Kaveh Vahedipour
commit
39083cb3d0
|
|
@ -1,447 +0,0 @@
|
|||
/*jshint globalstrict:false, strict:false */
|
||||
/*global assertFalse, assertTrue, assertEqual, ArangoAgency, ArangoServerState, ArangoClusterInfo */
|
||||
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
/// @brief test the cluster helper functions
|
||||
///
|
||||
/// @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 Jan Steemann
|
||||
/// @author Copyright 2013, triAGENS GmbH, Cologne, Germany
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
|
||||
var cluster = require("@arangodb/cluster");
|
||||
var jsunity = require("jsunity");
|
||||
|
||||
var compareStringIds = function (l, r) {
|
||||
'use strict';
|
||||
var i;
|
||||
if (l.length !== r.length) {
|
||||
return l.length - r.length < 0 ? -1 : 1;
|
||||
}
|
||||
|
||||
// length is equal
|
||||
for (i = 0; i < l.length; ++i) {
|
||||
if (l[i] !== r[i]) {
|
||||
return l[i] < r[i] ? -1 : 1;
|
||||
}
|
||||
}
|
||||
|
||||
return 0;
|
||||
};
|
||||
|
||||
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
/// @brief test suite: cluster enabled
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
|
||||
function ClusterEnabledSuite () {
|
||||
'use strict';
|
||||
var agency = ArangoAgency;
|
||||
var ss = ArangoServerState;
|
||||
var ci = ArangoClusterInfo;
|
||||
var oldPrefix = agency.prefix(true);
|
||||
var oldId = ss.id();
|
||||
var oldRole = ss.role();
|
||||
|
||||
var cleanDirectories = function () {
|
||||
[ "Target", "Plan", "Current" ].forEach(function (d) {
|
||||
try {
|
||||
agency.remove(d, true);
|
||||
}
|
||||
catch (err) {
|
||||
// dir may not exist. this is not a problem
|
||||
}
|
||||
});
|
||||
};
|
||||
|
||||
var initValues = function () {
|
||||
[ "Target",
|
||||
"Plan",
|
||||
"Current",
|
||||
"Target/DBServers",
|
||||
"Target/Coordinators",
|
||||
"Target/Collections",
|
||||
"Plan/DBServers",
|
||||
"Plan/Coordinators",
|
||||
"Plan/Collections",
|
||||
"Current/ServersRegistered",
|
||||
"Current/DBServers",
|
||||
"Current/Coordinators",
|
||||
"Current/Collections",
|
||||
"Current/ShardsCopied"
|
||||
].forEach(function (d) {
|
||||
try {
|
||||
agency.createDirectory(d);
|
||||
}
|
||||
catch (err) {
|
||||
}
|
||||
});
|
||||
};
|
||||
|
||||
return {
|
||||
setUp : function () {
|
||||
assertTrue(agency.setPrefix("UnitTestsAgency"));
|
||||
assertEqual("UnitTestsAgency", agency.prefix(true));
|
||||
|
||||
cleanDirectories();
|
||||
initValues();
|
||||
},
|
||||
|
||||
tearDown : function () {
|
||||
cleanDirectories();
|
||||
|
||||
assertTrue(agency.setPrefix(oldPrefix));
|
||||
ss.setId(oldId);
|
||||
assertEqual(oldPrefix, agency.prefix(true));
|
||||
ss.setRole(oldRole);
|
||||
ci.flush(); // reload caches with proper agency data
|
||||
},
|
||||
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
/// @brief test id
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
|
||||
testId : function () {
|
||||
ss.setId("myself");
|
||||
assertEqual("myself", ss.id());
|
||||
},
|
||||
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
/// @brief test isCluster
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
|
||||
testIsCluster : function () {
|
||||
assertTrue(cluster.isCluster());
|
||||
},
|
||||
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
/// @brief test isCoordinator
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
|
||||
testIsCoordinatorTrue : function () {
|
||||
assertTrue(agency.set("Plan/Coordinators/myself", "none"));
|
||||
ss.setId("myself");
|
||||
ss.flush();
|
||||
|
||||
assertTrue(cluster.isCoordinator());
|
||||
},
|
||||
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
/// @brief test isCoordinator
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
|
||||
testIsCoordinatorFalse : function () {
|
||||
assertTrue(agency.set("Plan/Coordinators/DBServers/myself", "other"));
|
||||
ss.setId("myself");
|
||||
ss.flush();
|
||||
|
||||
assertFalse(cluster.isCoordinator());
|
||||
},
|
||||
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
/// @brief test role
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
|
||||
testRoleCoordinator : function () {
|
||||
assertTrue(agency.set("Plan/Coordinators/myself", "none"));
|
||||
ss.setId("myself");
|
||||
ss.flush();
|
||||
|
||||
assertTrue(cluster.role() !== undefined);
|
||||
assertTrue(cluster.role() === "COORDINATOR");
|
||||
},
|
||||
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
/// @brief test role
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
|
||||
testRoleDBServer : function () {
|
||||
assertTrue(agency.set("Plan/DBServers/myself", "other"));
|
||||
ss.setId("myself");
|
||||
ss.flush();
|
||||
|
||||
assertTrue(cluster.role() !== undefined);
|
||||
assertTrue(cluster.role() === "PRIMARY");
|
||||
},
|
||||
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
/// @brief test status
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
|
||||
testStatus : function () {
|
||||
assertTrue(cluster.status() !== undefined);
|
||||
},
|
||||
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
/// @brief test isCoordinatorRequest
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
|
||||
testIsCoordinatorRequest : function () {
|
||||
assertFalse(cluster.isCoordinatorRequest(null));
|
||||
assertFalse(cluster.isCoordinatorRequest(undefined));
|
||||
assertFalse(cluster.isCoordinatorRequest({ }));
|
||||
assertFalse(cluster.isCoordinatorRequest({ headers: { } }));
|
||||
assertFalse(cluster.isCoordinatorRequest({ headers: { test: "" } }));
|
||||
assertTrue(cluster.isCoordinatorRequest({ headers: { "x-arango-coordinator": "abc" } }));
|
||||
},
|
||||
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
/// @brief test doesDatabaseExist
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
|
||||
testDoesDatabaseExist : function () {
|
||||
var database = {
|
||||
name: "test"
|
||||
};
|
||||
|
||||
assertTrue(agency.set("Plan/Databases/" + database.name, database));
|
||||
assertTrue(agency.set("Current/DBServers/Foo", "Bar"));
|
||||
assertTrue(agency.set("Current/DBServers/Barz", "Bat"));
|
||||
assertTrue(agency.set("Current/Databases/test/Foo", database));
|
||||
assertTrue(agency.set("Current/Databases/test/Barz", database));
|
||||
|
||||
assertTrue(ci.doesDatabaseExist("test"));
|
||||
|
||||
assertFalse(ci.doesDatabaseExist("UnitTestsAgencyNonExisting"));
|
||||
},
|
||||
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
/// @brief test doesDatabaseExist
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
|
||||
testDoesDatabaseExistNotReady : function () {
|
||||
var database = {
|
||||
name: "test"
|
||||
};
|
||||
|
||||
assertTrue(agency.set("Plan/Databases/" + database.name, database));
|
||||
assertTrue(agency.set("Current/DBServers/Foo", "Bar"));
|
||||
assertTrue(agency.set("Current/DBServers/Barz", "Bat"));
|
||||
assertTrue(agency.set("Current/Databases/test/Foo", database));
|
||||
|
||||
assertFalse(ci.doesDatabaseExist("test"));
|
||||
},
|
||||
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
/// @brief test getCollectionInfo
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
|
||||
testGetCollectionInfo : function () {
|
||||
var collection = {
|
||||
id: "123",
|
||||
name: "mycollection",
|
||||
type: 2,
|
||||
status: 3, // LOADED
|
||||
shardKeys: [ "_key" ],
|
||||
shards: { "s1" : ["myself"], "s2" : ["other"] }
|
||||
};
|
||||
assertTrue(agency.set("Plan/Collections/test/" + collection.id, collection));
|
||||
|
||||
ci.flush();
|
||||
|
||||
var data = ci.getCollectionInfo("test", collection.id);
|
||||
|
||||
assertEqual(collection.id, data.id);
|
||||
assertEqual(collection.name, data.name);
|
||||
assertEqual(collection.type, data.type);
|
||||
assertEqual(collection.status, data.status);
|
||||
assertEqual(collection.shardKeys, data.shardKeys);
|
||||
assertEqual(collection.shards, data.shards);
|
||||
},
|
||||
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
/// @brief test getCollectionInfo
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
|
||||
testGetCollectionInfo2 : function () {
|
||||
var collection = {
|
||||
id: "12345868390663",
|
||||
name: "mycollection_test",
|
||||
type: 3,
|
||||
status: 2, // LOADED
|
||||
shardKeys: [ "_key", "a", "bc" ],
|
||||
shards: { "s1" : ["myself"], "s2" : ["other"], "s3" : ["foo"], "s4" : ["bar"] }
|
||||
};
|
||||
|
||||
assertTrue(agency.set("Plan/Collections/test/" + collection.id, collection));
|
||||
|
||||
ci.flush();
|
||||
|
||||
var data = ci.getCollectionInfo("test", collection.id);
|
||||
|
||||
assertEqual(collection.id, data.id);
|
||||
assertEqual(collection.name, data.name);
|
||||
assertEqual(collection.type, data.type);
|
||||
assertEqual(collection.status, data.status);
|
||||
assertEqual(collection.shardKeys, data.shardKeys);
|
||||
assertEqual(collection.shards, data.shards);
|
||||
},
|
||||
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
/// @brief test getResponsibleServer
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
|
||||
testGetResponsibleServer : function () {
|
||||
var id = "12345868390663";
|
||||
|
||||
var current = {
|
||||
"error": false,
|
||||
"errorNum": 0,
|
||||
"errorMessage": "",
|
||||
"index": [], // fake
|
||||
"servers": ["myself"]
|
||||
};
|
||||
|
||||
assertTrue(agency.set("Current/Collections/test/" + id +
|
||||
"/s1", current));
|
||||
ci.flush();
|
||||
|
||||
assertEqual(["myself"], ci.getResponsibleServer("s1"));
|
||||
assertEqual([], ci.getResponsibleServer("s9999"));
|
||||
|
||||
current.servers = ["other"];
|
||||
assertTrue(agency.set("Current/Collections/test/" + id +
|
||||
"/s1", current));
|
||||
current.servers = ["myself"];
|
||||
assertTrue(agency.set("Current/Collections/test/" + id +
|
||||
"/s2", current));
|
||||
ci.flush();
|
||||
|
||||
assertEqual(["other"], ci.getResponsibleServer("s1"));
|
||||
assertEqual(["myself"], ci.getResponsibleServer("s2"));
|
||||
},
|
||||
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
/// @brief test getServerEndpoint
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
|
||||
testGetServerEndpoint : function () {
|
||||
assertTrue(agency.set("Current/ServersRegistered/myself", {endpoint:"tcp://127.0.0.1:8529"}));
|
||||
ci.flush();
|
||||
|
||||
assertEqual("tcp://127.0.0.1:8529", ci.getServerEndpoint("myself"));
|
||||
|
||||
assertTrue(agency.set("Current/ServersRegistered/myself", {endpoint:"tcp://127.0.0.1:8530"}));
|
||||
ci.flush();
|
||||
|
||||
assertEqual("tcp://127.0.0.1:8530", ci.getServerEndpoint("myself"));
|
||||
|
||||
assertTrue(agency.remove("Current/ServersRegistered/myself"));
|
||||
assertTrue(agency.set("Current/ServersRegistered/other", {endpoint:"tcp://127.0.0.1:8529"}));
|
||||
ci.flush();
|
||||
|
||||
assertEqual("", ci.getServerEndpoint("myself"));
|
||||
assertEqual("tcp://127.0.0.1:8529", ci.getServerEndpoint("other"));
|
||||
},
|
||||
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
/// @brief test uniqid
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
|
||||
testUniqid : function () {
|
||||
var last = "0";
|
||||
|
||||
for (var i = 0; i < 1005; ++i) {
|
||||
var id = ci.uniqid();
|
||||
assertEqual(-1, compareStringIds(last, id));
|
||||
last = id;
|
||||
}
|
||||
},
|
||||
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
/// @brief test uniqid
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
|
||||
testUniqidRange : function () {
|
||||
var last = "0";
|
||||
|
||||
for (var i = 0; i < 100; ++i) {
|
||||
var id = ci.uniqid(10);
|
||||
assertEqual(-1, compareStringIds(last, id));
|
||||
last = id;
|
||||
}
|
||||
}
|
||||
|
||||
};
|
||||
}
|
||||
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
/// @brief test suite: cluster disabled
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
|
||||
function ClusterDisabledSuite () {
|
||||
'use strict';
|
||||
return {
|
||||
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
/// @brief test isCluster
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
|
||||
testIsCluster : function () {
|
||||
assertFalse(cluster.isCluster());
|
||||
},
|
||||
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
/// @brief test role
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
|
||||
testRole : function () {
|
||||
assertTrue(cluster.role() === undefined);
|
||||
},
|
||||
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
/// @brief test status
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
|
||||
testStatus : function () {
|
||||
assertTrue(cluster.status() === undefined);
|
||||
},
|
||||
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
/// @brief test isCoordinatorRequest
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
|
||||
testIsCoordinatorRequest : function () {
|
||||
assertFalse(cluster.isCoordinatorRequest(null));
|
||||
assertFalse(cluster.isCoordinatorRequest(undefined));
|
||||
assertFalse(cluster.isCoordinatorRequest({ }));
|
||||
assertFalse(cluster.isCoordinatorRequest({ headers: { } }));
|
||||
assertFalse(cluster.isCoordinatorRequest({ headers: { test: "" } }));
|
||||
assertTrue(cluster.isCoordinatorRequest({ headers: { "x-arango-coordinator": "abc" } }));
|
||||
}
|
||||
};
|
||||
}
|
||||
|
||||
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
/// @brief executes the test suites
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
|
||||
if (cluster.isCluster()) {
|
||||
jsunity.run(ClusterEnabledSuite);
|
||||
}
|
||||
else {
|
||||
jsunity.run(ClusterDisabledSuite);
|
||||
}
|
||||
|
||||
return jsunity.done();
|
||||
|
||||
Loading…
Reference in New Issue