1
0
Fork 0
arangodb/js/server/tests/aql-join-cluster.js

307 lines
12 KiB
JavaScript

/*jshint globalstrict:false, strict:false, maxlen: 500 */
/*global assertEqual, AQL_EXECUTE */
////////////////////////////////////////////////////////////////////////////////
/// @brief tests for query language, bind parameters
///
/// @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 2012, triAGENS GmbH, Cologne, Germany
////////////////////////////////////////////////////////////////////////////////
var db = require("org/arangodb").db;
var jsunity = require("jsunity");
////////////////////////////////////////////////////////////////////////////////
/// @brief test suite
////////////////////////////////////////////////////////////////////////////////
function ahuacatlClusterJoinKeySuite () {
var cn1 = "UnitTestsAhuacatlSubquery1";
var cn2 = "UnitTestsAhuacatlSubquery2";
var c1, c2;
var sorter = function (l, r) {
if (l !== r) {
return l - r;
}
return 0;
};
var sorter2 = function (l, r) {
if (l[0] !== r[0]) {
return l[0] - r[0];
}
return 0;
};
return {
////////////////////////////////////////////////////////////////////////////////
/// @brief set up
////////////////////////////////////////////////////////////////////////////////
setUp : function () {
db._drop(cn1);
db._drop(cn2);
c1 = db._create(cn1, { numberOfShards: 5 });
c2 = db._create(cn2, { numberOfShards: 4 });
for (var i = 0; i < 20; ++i) {
c1.save({ _key: "test" + i, value: i });
c2.save({ _key: "test" + i, value: i });
}
},
////////////////////////////////////////////////////////////////////////////////
/// @brief tear down
////////////////////////////////////////////////////////////////////////////////
tearDown : function () {
db._drop(cn1);
db._drop(cn2);
c1 = null;
c2 = null;
},
////////////////////////////////////////////////////////////////////////////////
/// @brief test subquery
////////////////////////////////////////////////////////////////////////////////
testNestedLoopsDifferentCollectionsNonKey : function () {
var actual = AQL_EXECUTE("FOR c1 IN @@cn1 FOR c2 IN @@cn2 FILTER c1.value == c2.value RETURN c2.value", { "@cn1": cn1, "@cn2": cn2 });
actual.json.sort(sorter);
assertEqual([ 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19 ], actual.json);
},
////////////////////////////////////////////////////////////////////////////////
/// @brief test subquery
////////////////////////////////////////////////////////////////////////////////
testNestedLoopsDifferentCollectionsNonKeyFilter : function () {
var actual = AQL_EXECUTE("FOR c1 IN @@cn1 FILTER c1.value IN [ 1, 2, 4 ] FOR c2 IN @@cn2 FILTER c1.value == c2.value RETURN c2.value", { "@cn1": cn1, "@cn2": cn2 });
actual.json.sort(sorter);
assertEqual([ 1, 2, 4 ], actual.json);
},
////////////////////////////////////////////////////////////////////////////////
/// @brief test subquery
////////////////////////////////////////////////////////////////////////////////
testNestedLoopsDifferentCollectionsKey : function () {
var actual = AQL_EXECUTE("FOR c1 IN @@cn1 FOR c2 IN @@cn2 FILTER c1._key == c2._key RETURN c2.value", { "@cn1": cn1, "@cn2": cn2 });
actual.json.sort(sorter);
assertEqual([ 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19 ], actual.json);
},
////////////////////////////////////////////////////////////////////////////////
/// @brief test subquery
////////////////////////////////////////////////////////////////////////////////
testNestedLoopsDifferentCollectionsKeyFilter : function () {
var actual = AQL_EXECUTE("FOR c1 IN @@cn1 FILTER c1.value IN [ 1, 2, 4 ] FOR c2 IN @@cn2 FILTER c1._key == c2._key RETURN c2.value", { "@cn1": cn1, "@cn2": cn2 });
actual.json.sort(sorter);
assertEqual([ 1, 2, 4 ], actual.json);
},
////////////////////////////////////////////////////////////////////////////////
/// @brief test subquery
////////////////////////////////////////////////////////////////////////////////
testNestedLoopsSameCollection : function () {
var actual = AQL_EXECUTE("FOR c1 IN @@cn1 FILTER c1.value IN [ 1, 2, 4 ] FOR c2 IN @@cn2 FILTER c1._key == c2._key RETURN c2.value", { "@cn1": cn1, "@cn2": cn1 }); // same collection
actual.json.sort(sorter);
assertEqual([ 1, 2, 4 ], actual.json);
},
////////////////////////////////////////////////////////////////////////////////
/// @brief test subquery
////////////////////////////////////////////////////////////////////////////////
testNestedLoopsSameCollectionNonShardKey : function () {
var actual = AQL_EXECUTE("FOR c1 IN @@cn1 FILTER c1.value IN [ 1, 2, 4 ] FOR c2 IN @@cn2 FILTER c1.value == c2.value RETURN c2.value", { "@cn1": cn1, "@cn2": cn1 }); // same collection
actual.json.sort(sorter);
assertEqual([ 1, 2, 4 ], actual.json);
},
////////////////////////////////////////////////////////////////////////////////
/// @brief test subquery
////////////////////////////////////////////////////////////////////////////////
testSubqueryDifferentCollectionsKey : function () {
var actual = AQL_EXECUTE("FOR c1 IN @@cn1 FILTER c1.value IN [ 1, 2, 4 ] LET match = (FOR c2 IN @@cn2 FILTER c1._key == c2._key RETURN c2.value) RETURN match", { "@cn1": cn1, "@cn2": cn2 });
actual.json.sort(sorter2);
assertEqual([ [ 1 ], [ 2 ], [ 4 ] ], actual.json);
},
////////////////////////////////////////////////////////////////////////////////
/// @brief test subquery
////////////////////////////////////////////////////////////////////////////////
testSubqueryDifferentCollectionsNonKey : function () {
var actual = AQL_EXECUTE("FOR c1 IN @@cn1 FILTER c1.value IN [ 1, 2, 4 ] LET match = (FOR c2 IN @@cn2 FILTER c1.value == c2.value RETURN c2.value) RETURN match", { "@cn1": cn1, "@cn2": cn2 });
actual.json.sort(sorter2);
assertEqual([ [ 1 ], [ 2 ], [ 4 ] ], actual.json);
},
////////////////////////////////////////////////////////////////////////////////
/// @brief test subquery
////////////////////////////////////////////////////////////////////////////////
testSubquerySameCollectionKey : function () {
var actual = AQL_EXECUTE("FOR c1 IN @@cn1 FILTER c1.value IN [ 1, 2, 4 ] LET match = (FOR c2 IN @@cn2 FILTER c1._key == c2._key RETURN c2.value) RETURN match", { "@cn1": cn1, "@cn2": cn1 }); // same collection
actual.json.sort(sorter2);
assertEqual([ [ 1 ], [ 2 ], [ 4 ] ], actual.json);
},
////////////////////////////////////////////////////////////////////////////////
/// @brief test subquery
////////////////////////////////////////////////////////////////////////////////
testSubquerySameCollectionNonKey : function () {
var actual = AQL_EXECUTE("FOR c1 IN @@cn1 FILTER c1.value IN [ 1, 2, 4 ] LET match = (FOR c2 IN @@cn2 FILTER c1.value == c2.value RETURN c2.value) RETURN match", { "@cn1": cn1, "@cn2": cn1 }); // same collection
actual.json.sort(sorter2);
assertEqual([ [ 1 ], [ 2 ], [ 4 ] ], actual.json);
}
};
}
////////////////////////////////////////////////////////////////////////////////
/// @brief test suite
////////////////////////////////////////////////////////////////////////////////
function ahuacatlClusterJoinNonKeySuite () {
var cn1 = "UnitTestsAhuacatlSubquery1";
var cn2 = "UnitTestsAhuacatlSubquery2";
var c1, c2;
var sorter = function (l, r) {
if (l !== r) {
return l - r;
}
return 0;
};
var sorter2 = function (l, r) {
if (l[0] !== r[0]) {
return l[0] - r[0];
}
return 0;
};
return {
////////////////////////////////////////////////////////////////////////////////
/// @brief set up
////////////////////////////////////////////////////////////////////////////////
setUp : function () {
db._drop(cn1);
db._drop(cn2);
c1 = db._create(cn1, { numberOfShards: 5, shardKeys: [ "value" ] });
c2 = db._create(cn2, { numberOfShards: 4, shardKeys: [ "value" ] });
for (var i = 0; i < 20; ++i) {
c1.save({ value: i });
c2.save({ value: i });
}
},
////////////////////////////////////////////////////////////////////////////////
/// @brief tear down
////////////////////////////////////////////////////////////////////////////////
tearDown : function () {
db._drop(cn1);
db._drop(cn2);
c1 = null;
c2 = null;
},
////////////////////////////////////////////////////////////////////////////////
/// @brief test subquery
////////////////////////////////////////////////////////////////////////////////
testNestedLoopsDifferentCollectionsNonKey : function () {
var actual = AQL_EXECUTE("FOR c1 IN @@cn1 FOR c2 IN @@cn2 FILTER c1.value == c2.value RETURN c2.value", { "@cn1": cn1, "@cn2": cn2 });
actual.json.sort(sorter);
assertEqual([ 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19 ], actual.json);
},
////////////////////////////////////////////////////////////////////////////////
/// @brief test subquery
////////////////////////////////////////////////////////////////////////////////
testNestedLoopsDifferentCollectionsNonKeyFilter : function () {
var actual = AQL_EXECUTE("FOR c1 IN @@cn1 FILTER c1.value IN [ 1, 2, 4 ] FOR c2 IN @@cn2 FILTER c1.value == c2.value RETURN c2.value", { "@cn1": cn1, "@cn2": cn2 });
actual.json.sort(sorter);
assertEqual([ 1, 2, 4 ], actual.json);
},
////////////////////////////////////////////////////////////////////////////////
/// @brief test subquery
////////////////////////////////////////////////////////////////////////////////
testNestedLoopsSameCollectionNonShardKey : function () {
var actual = AQL_EXECUTE("FOR c1 IN @@cn1 FILTER c1.value IN [ 1, 2, 4 ] FOR c2 IN @@cn2 FILTER c1.value == c2.value RETURN c2.value", { "@cn1": cn1, "@cn2": cn1 }); // same collection
actual.json.sort(sorter);
assertEqual([ 1, 2, 4 ], actual.json);
},
////////////////////////////////////////////////////////////////////////////////
/// @brief test subquery
////////////////////////////////////////////////////////////////////////////////
testSubqueryDifferentCollectionsNonKey : function () {
var actual = AQL_EXECUTE("FOR c1 IN @@cn1 FILTER c1.value IN [ 1, 2, 4 ] LET match = (FOR c2 IN @@cn2 FILTER c1.value == c2.value RETURN c2.value) RETURN match", { "@cn1": cn1, "@cn2": cn2 });
actual.json.sort(sorter2);
assertEqual([ [ 1 ], [ 2 ], [ 4 ] ], actual.json);
},
////////////////////////////////////////////////////////////////////////////////
/// @brief test subquery
////////////////////////////////////////////////////////////////////////////////
testSubquerySameCollectionNonKey : function () {
var actual = AQL_EXECUTE("FOR c1 IN @@cn1 FILTER c1.value IN [ 1, 2, 4 ] LET match = (FOR c2 IN @@cn2 FILTER c1.value == c2.value RETURN c2.value) RETURN match", { "@cn1": cn1, "@cn2": cn1 }); // same collection
actual.json.sort(sorter2);
assertEqual([ [ 1 ], [ 2 ], [ 4 ] ], actual.json);
}
};
}
////////////////////////////////////////////////////////////////////////////////
/// @brief executes the test suite
////////////////////////////////////////////////////////////////////////////////
jsunity.run(ahuacatlClusterJoinKeySuite);
jsunity.run(ahuacatlClusterJoinNonKeySuite);
return jsunity.done();
// Local Variables:
// mode: outline-minor
// outline-regexp: "^\\(/// @brief\\|/// @addtogroup\\|// --SECTION--\\|/// @page\\|/// @}\\)"
// End: