1
0
Fork 0
arangodb/js/server/tests/aql-distinct.js

245 lines
9.2 KiB
JavaScript

/*jshint globalstrict:false, strict:false, maxlen: 500 */
/*global assertEqual, fail, AQL_EXECUTE */
////////////////////////////////////////////////////////////////////////////////
/// @brief tests for COLLECT w/ COUNT
///
/// @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 jsunity = require("jsunity");
var db = require("org/arangodb").db;
var internal = require("internal");
var errors = internal.errors;
////////////////////////////////////////////////////////////////////////////////
/// @brief test suite
////////////////////////////////////////////////////////////////////////////////
function ahuacatlDistinct () {
var c;
return {
setUp : function () {
db._drop("UnitTestsCollection");
c = db._create("UnitTestsCollection");
for (var i = 0; i < 100; ++i) {
c.save({ value1: i, value2: i % 10, value3: "test" + i, value4: "test" + (i % 10) });
}
},
tearDown : function () {
db._drop("UnitTestsCollection");
},
////////////////////////////////////////////////////////////////////////////////
/// @brief distinct usage in different places
////////////////////////////////////////////////////////////////////////////////
testDistinctUsageFail : function () {
var queries = [
"FOR i IN DISTINCT 1..10 RETURN i",
"FOR i IN 1..10 LET a = DISTINCT i RETURN a",
"RETURN DISTINCT 1",
"LET x = DISTINCT 1 RETURN x",
"LET a = 1 LET x = DISTINCT a RETURN x",
"LET a = 1 COLLECT x = a RETURN DISTINCT x",
"LET x = (FOR i IN 1..100 RETURN i) RETURN DISTINCT x",
"LET values = (RETURN DISTINCT 1) RETURN values"
];
queries.forEach(function(query) {
try {
AQL_EXECUTE(query);
fail();
}
catch (e) {
assertEqual(errors.ERROR_QUERY_PARSE.code, e.errorNum);
}
});
},
////////////////////////////////////////////////////////////////////////////////
/// @brief distinct usage
////////////////////////////////////////////////////////////////////////////////
testDistinctInTopLevelForWithInteger : function () {
var result = AQL_EXECUTE("FOR i IN " + c.name() + " RETURN DISTINCT i.value2").json;
result = result.sort();
assertEqual([ 0, 1, 2, 3, 4, 5, 6, 7, 8, 9 ], result);
},
////////////////////////////////////////////////////////////////////////////////
/// @brief distinct usage
////////////////////////////////////////////////////////////////////////////////
testDistinctInTopLevelForWithString : function () {
var result = AQL_EXECUTE("FOR i IN " + c.name() + " RETURN DISTINCT i.value4").json;
result = result.sort();
assertEqual([ "test0", "test1", "test2", "test3", "test4", "test5", "test6", "test7", "test8", "test9" ], result);
},
////////////////////////////////////////////////////////////////////////////////
/// @brief distinct usage
////////////////////////////////////////////////////////////////////////////////
testDistinctOnDistinctValuesWithInteger : function () {
var result = AQL_EXECUTE("FOR i IN " + c.name() + " RETURN DISTINCT i.value1").json;
assertEqual(100, result.length);
},
////////////////////////////////////////////////////////////////////////////////
/// @brief distinct usage
////////////////////////////////////////////////////////////////////////////////
testDistinctOnDistinctValuesWithString : function () {
var result = AQL_EXECUTE("FOR i IN " + c.name() + " RETURN DISTINCT i.value3").json;
assertEqual(100, result.length);
},
////////////////////////////////////////////////////////////////////////////////
/// @brief distinct usage
////////////////////////////////////////////////////////////////////////////////
testDistinctInTopLevelForWithExpression : function () {
var result = AQL_EXECUTE("FOR i IN " + c.name() + " RETURN DISTINCT i.value1 % 10").json;
result = result.sort();
assertEqual([ 0, 1, 2, 3, 4, 5, 6, 7, 8, 9 ], result);
},
////////////////////////////////////////////////////////////////////////////////
/// @brief distinct usage
////////////////////////////////////////////////////////////////////////////////
testDistinctReferringToLet : function () {
var result = AQL_EXECUTE("LET a = PASSTHRU(1) FOR i IN " + c.name() + " RETURN DISTINCT a").json;
assertEqual([ 1 ], result);
},
////////////////////////////////////////////////////////////////////////////////
/// @brief distinct usage
////////////////////////////////////////////////////////////////////////////////
testDistinctInSubquery : function () {
var result = AQL_EXECUTE("LET values = (FOR i IN " + c.name() + " RETURN DISTINCT i.value2) RETURN values").json[0];
result = result.sort();
assertEqual([ 0, 1, 2, 3, 4, 5, 6, 7, 8, 9 ], result);
},
////////////////////////////////////////////////////////////////////////////////
/// @brief distinct usage
////////////////////////////////////////////////////////////////////////////////
testDistinctInSubqueries : function () {
var result = AQL_EXECUTE("LET values1 = (FOR i IN " + c.name() + " RETURN DISTINCT i.value2) LET values2 = (FOR i IN " + c.name() + " RETURN DISTINCT i.value4) RETURN [ values1, values2 ]").json[0];
var integers = result[0].sort();
var strings = result[1].sort();
assertEqual([ 0, 1, 2, 3, 4, 5, 6, 7, 8, 9 ], integers);
assertEqual([ "test0", "test1", "test2", "test3", "test4", "test5", "test6", "test7", "test8", "test9" ], strings);
},
////////////////////////////////////////////////////////////////////////////////
/// @brief distinct usage
////////////////////////////////////////////////////////////////////////////////
testDistinctOnDistinctValuesInSubqueries : function () {
var result = AQL_EXECUTE("LET values1 = (FOR i IN " + c.name() + " RETURN DISTINCT i.value1) LET values2 = (FOR i IN " + c.name() + " RETURN DISTINCT i.value3) RETURN [ values1, values2 ]").json[0];
var integers = result[0];
var strings = result[1];
assertEqual(100, integers.length);
assertEqual(100, strings.length);
},
////////////////////////////////////////////////////////////////////////////////
/// @brief distinct usage
////////////////////////////////////////////////////////////////////////////////
testDistinctWithCollect1 : function () {
var result = AQL_EXECUTE("FOR i IN " + c.name() + " COLLECT x = i.value1 % 11 RETURN DISTINCT x").json;
result = result.sort(function(l, r) {
return l - r;
});
assertEqual([ 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10 ], result);
},
////////////////////////////////////////////////////////////////////////////////
/// @brief distinct usage
////////////////////////////////////////////////////////////////////////////////
testDistinctWithCollect2 : function () {
var result = AQL_EXECUTE("FOR i IN " + c.name() + " COLLECT x = i.value1 RETURN DISTINCT x % 11").json;
result = result.sort(function(l, r) {
return l - r;
});
assertEqual([ 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10 ], result);
},
////////////////////////////////////////////////////////////////////////////////
/// @brief distinct usage
////////////////////////////////////////////////////////////////////////////////
testDistinctWithArrays1 : function () {
var result = AQL_EXECUTE("FOR i IN 1..2 RETURN DISTINCT (FOR j IN [ 1, 2, 3, 4, 1, 3 ] RETURN j)").json;
assertEqual(1, result.length);
assertEqual([ 1, 2, 3, 4, 1, 3 ], result[0]);
},
////////////////////////////////////////////////////////////////////////////////
/// @brief distinct usage
////////////////////////////////////////////////////////////////////////////////
testDistinctWithArrays2 : function () {
var result = AQL_EXECUTE("FOR i IN 1..2 RETURN (FOR j IN [ 1, 2, 3, 4, 1, 3 ] RETURN DISTINCT j)").json;
assertEqual(2, result.length);
assertEqual([ 1, 2, 3, 4 ], result[0].sort(function (l, r) { return l - r; }));
assertEqual([ 1, 2, 3, 4 ], result[1].sort(function (l, r) { return l - r; }));
}
};
}
////////////////////////////////////////////////////////////////////////////////
/// @brief executes the test suite
////////////////////////////////////////////////////////////////////////////////
jsunity.run(ahuacatlDistinct);
return jsunity.done();
// Local Variables:
// mode: outline-minor
// outline-regexp: "^\\(/// @brief\\|/// @addtogroup\\|// --SECTION--\\|/// @page\\|/// @}\\)"
// End: