mirror of https://gitee.com/bigwinds/arangodb
379 lines
16 KiB
JavaScript
379 lines
16 KiB
JavaScript
////////////////////////////////////////////////////////////////////////////////
|
|
/// @brief tests for query language, escaping
|
|
///
|
|
/// @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 ArangoError = require("org/arangodb").ArangoError;
|
|
var QUERY = require("internal").AQL_QUERY;
|
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
/// @brief test suite
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
|
|
function ahuacatlEscapingTestSuite () {
|
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
/// @brief execute a given query
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
|
|
function executeQuery (query) {
|
|
var cursor = QUERY(query, undefined);
|
|
if (cursor instanceof ArangoError) {
|
|
print(query, cursor.errorMessage);
|
|
}
|
|
assertFalse(cursor instanceof ArangoError);
|
|
return cursor;
|
|
}
|
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
/// @brief execute a given query and return the results as an array
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
|
|
function getQueryResults (query, isFlat) {
|
|
var result = executeQuery(query).getRows();
|
|
var results = [ ];
|
|
|
|
for (var i in result) {
|
|
if (!result.hasOwnProperty(i)) {
|
|
continue;
|
|
}
|
|
|
|
var row = result[i];
|
|
if (isFlat) {
|
|
results.push(row);
|
|
}
|
|
else {
|
|
var keys = [ ];
|
|
for (var k in row) {
|
|
if (row.hasOwnProperty(k) && k != '_id' && k != '_rev' && k != '_key') {
|
|
keys.push(k);
|
|
}
|
|
}
|
|
|
|
keys.sort();
|
|
var resultRow = { };
|
|
for (var k in keys) {
|
|
if (keys.hasOwnProperty(k)) {
|
|
resultRow[keys[k]] = row[keys[k]];
|
|
}
|
|
}
|
|
results.push(resultRow);
|
|
}
|
|
}
|
|
|
|
return results;
|
|
}
|
|
|
|
|
|
return {
|
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
/// @brief set up
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
|
|
setUp : function () {
|
|
},
|
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
/// @brief tear down
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
|
|
tearDown : function () {
|
|
},
|
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
/// @brief test simple name handling
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
|
|
testSimpleName : function () {
|
|
var expected = [ 1, 2, 3 ];
|
|
var actual = getQueryResults("FOR `x` IN [ 1, 2, 3 ] RETURN `x`", true);
|
|
assertEqual(expected, actual);
|
|
},
|
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
/// @brief test reserved name handling
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
|
|
testReservedNames1 : function () {
|
|
var expected = [ 1, 2, 3 ];
|
|
var names = [ "for", "let", "return", "sort", "collect", "limit", "filter", "asc", "desc", "in", "into" ];
|
|
|
|
for (var i in names) {
|
|
if (!names.hasOwnProperty(i)) {
|
|
continue;
|
|
}
|
|
|
|
var actual = getQueryResults("FOR `" + names[i] + "` IN [ 1, 2, 3 ] RETURN `" + names[i] + "`", true);
|
|
assertEqual(expected, actual);
|
|
}
|
|
},
|
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
/// @brief test reserved names handling
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
|
|
testReservedNames2 : function () {
|
|
var expected = [ { "let" : 1 }, { "collect" : 2 }, { "sort" : 3 } ];
|
|
var actual = getQueryResults("FOR `for` IN [ { \"let\" : 1 }, { \"collect\" : 2 }, { \"sort\" : 3 } ] RETURN `for`", false);
|
|
assertEqual(expected, actual);
|
|
},
|
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
/// @brief test punctuation names escaping
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
|
|
testPunctuationBackticks1 : function () {
|
|
var expected = [ 1, 2, 3 ];
|
|
var actual = getQueryResults("FOR `brown_fox` IN [ 1, 2, 3 ] RETURN `brown_fox`", true);
|
|
assertEqual(expected, actual);
|
|
},
|
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
/// @brief test punctuation names escaping
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
|
|
testPunctuationBackticks2 : function () {
|
|
var expected = [ 1, 2, 3 ];
|
|
var actual = getQueryResults("FOR `brown_fox__1234_` IN [ 1, 2, 3 ] RETURN `brown_fox__1234_`", true);
|
|
assertEqual(expected, actual);
|
|
},
|
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
/// @brief test punctuation names escaping
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
|
|
testPunctuationBackticks3 : function () {
|
|
var expected = [ 1, 2, 3 ];
|
|
var actual = getQueryResults("FOR `brown fox 1234_` IN [ 1, 2, 3 ] RETURN `brown fox 1234_`", true);
|
|
assertEqual(expected, actual);
|
|
},
|
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
/// @brief test punctuation names escaping
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
|
|
testPunctuationBackticks4 : function () {
|
|
var expected = [ 1, 3 ];
|
|
var actual = getQueryResults("FOR r IN [ { \"a\" : 1, \"b\" : 1 }, { \"a\" : 2, \"b\" : 2 }, { \"a\" : 1, \"b\" : 3 } ] FILTER r.`a` == 1 RETURN r.`b`", true);
|
|
assertEqual(expected, actual);
|
|
},
|
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
/// @brief test punctuation names escaping
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
|
|
testPunctuationBackticks5 : function () {
|
|
var expected = [ 1, 3 ];
|
|
var actual = getQueryResults("FOR r IN [ { \"a fox\" : 1, \"b fox\" : 1 }, { \"a fox\" : 2, \"b fox\" : 2 }, { \"a fox\" : 1, \"b fox\" : 3 } ] FILTER r.`a fox` == 1 RETURN r.`b fox`", true);
|
|
assertEqual(expected, actual);
|
|
},
|
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
/// @brief test punctuation names escaping
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
|
|
testPunctuationDoubleQuotes1 : function () {
|
|
var expected = [ { '"a"' : 1 }, { '"a"' : '"b"' } ];
|
|
var actual = getQueryResults("FOR r IN [ { '\"a\"' : 1 }, { '\"a\"' : '\"b\"' } ] RETURN r", true);
|
|
assertEqual(expected, actual);
|
|
},
|
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
/// @brief test punctuation names escaping
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
|
|
testPunctuationDoubleQuotes2 : function () {
|
|
var expected = [ { '"a fox"' : 1 }, { '"a fox "' : '"b fox"' } ];
|
|
var actual = getQueryResults("FOR r IN [ { '\"a fox\"' : 1 }, { '\"a fox \"' : '\"b fox\"' } ] RETURN r", true);
|
|
assertEqual(expected, actual);
|
|
},
|
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
/// @brief test punctuation names escaping
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
|
|
testPunctuationDoubleQuotes3 : function () {
|
|
var expected = [ { '"a" \\ " fox"' : 1 }, { '"a" \\ " fox "' : '"b fox"' } ];
|
|
var actual = getQueryResults("FOR r IN [ { '\"a\" \\\\ \\\" fox\"' : 1 }, { '\"a\" \\\\ \\\" fox \"' : '\"b fox\"' } ] RETURN r", true);
|
|
assertEqual(expected, actual);
|
|
},
|
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
/// @brief test punctuation names escaping
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
|
|
testPunctuationSingleQuotes1 : function () {
|
|
var expected = [ { "'a'" : 1 }, { "'a'" : "'b'" } ];
|
|
var actual = getQueryResults("FOR r IN [ { \"'a'\" : 1 }, { \"'a'\" : \"'b'\" } ] RETURN r", true);
|
|
assertEqual(expected, actual);
|
|
},
|
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
/// @brief test punctuation names escaping
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
|
|
testPunctuationSingleQuotes2 : function () {
|
|
var expected = [ { "'a fox'" : 1 }, { "'a fox'" : "'b fox'" } ];
|
|
var actual = getQueryResults("FOR r IN [ { \"'a fox'\" : 1 }, { \"'a fox'\" : \"'b fox'\" } ] RETURN r", true);
|
|
assertEqual(expected, actual);
|
|
},
|
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
/// @brief test punctuation names escaping
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
|
|
testPunctuationSingleQuotes3 : function () {
|
|
var expected = [ { "'a \\ ' fox'" : 1 }, { "'a \\ ' fox'" : "'b fox'" } ];
|
|
var actual = getQueryResults("FOR r IN [ { \"'a \\\\ \\' fox'\" : 1 }, { \"'a \\\\ \\' fox'\" : \"'b fox'\" } ] RETURN r", true);
|
|
assertEqual(expected, actual);
|
|
},
|
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
/// @brief test punctuation unquoted names handling
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
|
|
testPunctuationNoQuotes1 : function () {
|
|
var expected = [ { "r" : 1 }, { "r" : 2 }, { "r" : 3 } ];
|
|
var actual = getQueryResults("FOR r IN [ 1, 2, 3 ] return { r : r }", true);
|
|
assertEqual(expected, actual);
|
|
},
|
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
/// @brief test punctuation unquoted names handling
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
|
|
testPunctuationNoQuotes2 : function () {
|
|
var expected = [ { "alias" : 1 }, { "alias" : 2 }, { "alias" : 3 } ];
|
|
var actual = getQueryResults("FOR r IN [ 1, 2, 3 ] return { alias : r }", true);
|
|
assertEqual(expected, actual);
|
|
},
|
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
/// @brief test punctuation unquoted names handling
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
|
|
testPunctuationNoQuotes3 : function () {
|
|
var expected = [ { "v" : { "value" : 1 } }, { "v" : { "value" : 2 } } ];
|
|
var actual = getQueryResults("FOR r IN [ 1, 2 ] LET fx = { value : r } RETURN { v : fx }", false);
|
|
assertEqual(expected, actual);
|
|
},
|
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
/// @brief test punctuation unquoted names handling
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
|
|
testPunctuationNoQuotes4 : function () {
|
|
var expected = [ { "fx" : [ { "someval" : 99 }, { "someval" : 1 } ] }, { "fx" : [ { "someval" : 99 }, { "someval" : 2 } ] } ];
|
|
var actual = getQueryResults("FOR r IN [ 1, 2 ] LET fx = ( FOR someval IN [ 99, r ] RETURN { someval : someval } ) RETURN { fx : fx }", false);
|
|
assertEqual(expected, actual);
|
|
},
|
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
/// @brief test UTF8 names escaping
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
|
|
testUtf8Names1 : function () {
|
|
var expected = [ "wälder", "hänsel", "grätel", "fraß", "kloß" ];
|
|
var actual = getQueryResults("FOR r IN [ { \"äöüÄÖÜß\" : \"wälder\" }, { \"äöüÄÖÜß\" : \"hänsel\" }, { \"äöüÄÖÜß\" : \"grätel\" }, { \"äöüÄÖÜß\" : \"fraß\" }, { \"äöüÄÖÜß\" : \"kloß\" } ] RETURN r.`äöüÄÖÜß`", true);
|
|
assertEqual(expected, actual);
|
|
},
|
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
/// @brief test UTF8 names escaping
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
|
|
testUtf8Names2 : function () {
|
|
var expected = [ "中央アメリカ", "熱帯", "亜熱帯" ];
|
|
var actual = getQueryResults("FOR r IN [ { \"アボカド\" : \"中央アメリカ\" }, { \"アボカド\" : \"熱帯\" }, { \"アボカド\" : \"亜熱帯\" } ] RETURN r.`アボカド`", true);
|
|
assertEqual(expected, actual);
|
|
},
|
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
/// @brief line breaks
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
|
|
testLineBreaks1 : function () {
|
|
var expected = [ "the\nquick\nbrown\r\nfox\rjumped\n" ];
|
|
var actual = getQueryResults("RETURN \"the\nquick\nbrown\r\nfox\rjumped\n\"", true);
|
|
assertEqual(expected, actual);
|
|
},
|
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
/// @brief line breaks2
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
|
|
testLineBreaks2 : function () {
|
|
var expected = [ { "the\nquick\nbrown\r\nfox\rjumped\n" : "over\nthis\r\nattribute" } ];
|
|
var actual = getQueryResults("RETURN { \"the\nquick\nbrown\r\nfox\rjumped\n\" : \"over\nthis\r\nattribute\" }", true);
|
|
assertEqual(expected, actual);
|
|
},
|
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
/// @brief test comments
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
|
|
testComments1 : function () {
|
|
var expected = [ "jumped" ];
|
|
var actual = getQueryResults("RETURN /* \"the quick fox\" */ \"jumped\"", true);
|
|
assertEqual(expected, actual);
|
|
},
|
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
/// @brief test comments
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
|
|
testComments2 : function () {
|
|
var expected = [ 1 ];
|
|
var actual = getQueryResults("RETURN /* \"the\n \" \\qui\\ck\\ \"\"\"\" \" \"\"f\\ox\" */ 1", true);
|
|
assertEqual(expected, actual);
|
|
},
|
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
/// @brief test comments
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
|
|
testComments3 : function () {
|
|
var expected = [ 1 ];
|
|
var actual = getQueryResults("RETURN /*'the ' \\qui\\ck\\ ''''' '' 'f\\ox'*/ 1", true);
|
|
assertEqual(expected, actual);
|
|
}
|
|
};
|
|
}
|
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
/// @brief executes the test suite
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
|
|
jsunity.run(ahuacatlEscapingTestSuite);
|
|
|
|
return jsunity.done();
|
|
|
|
// Local Variables:
|
|
// mode: outline-minor
|
|
// outline-regexp: "^\\(/// @brief\\|/// @addtogroup\\|// --SECTION--\\|/// @page\\|/// @}\\)"
|
|
// End:
|