Added tests for unique constraints.

arangodb · maierlars · Mar 12, 2021 · Mar 12, 2021 · Mar 12, 2021 · Mar 12, 2021
commit 2372f61e80070c211397ff7f5b85b4d9bed7759c
diff --git a/tests/js/server/aql/aql-optimizer-zkdindex-unique.js b/tests/js/server/aql/aql-optimizer-zkdindex-unique.js
@@ -0,0 +1,110 @@
+/* global AQL_EXPLAIN, AQL_EXECUTE, fail */
+////////////////////////////////////////////////////////////////////////////////
+/// DISCLAIMER
+///
+/// Copyright 2021-2021 ArangoDB 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 ArangoDB GmbH, Cologne, Germany
+///
+/// @author Tobias Gödderz
+////////////////////////////////////////////////////////////////////////////////
+
+'use strict';
+
+const jsunity = require("jsunity");
+const arangodb = require("@arangodb");
+const internal = require("internal");
+const db = arangodb.db;
+const aql = arangodb.aql;
+const {assertTrue, assertFalse, assertEqual} = jsunity.jsUnity.assertions;
+const _ = require("lodash");
+
+const useIndexes = 'use-indexes';
+const removeFilterCoveredByIndex = "remove-filter-covered-by-index";
+const moveFiltersIntoEnumerate = "move-filters-into-enumerate";
+
+function optimizerRuleZkd2dIndexTestSuite() {
+    const colName = 'UnitTestZkdIndexCollection';
+    let col;
+
+    return {
+        setUpAll: function () {
+            col = db._create(colName);
+            col.ensureIndex({type: 'zkd', name: 'zkdIndex', fields: ['x', 'y'], unique: true});
+            // Insert 1001 points
+            // (-500, -499.5), (-499.1, -499.4), ..., (0, 0.5), ..., (499.9, 500.4), (500, 500.5)
+            db._query(aql`
+        FOR i IN 0..1000
+          LET x = (i - 500) / 10
+          LET y = x + 0.5
+          INSERT {x, y, i} INTO ${col}
+      `);
+        },
+
+        tearDownAll: function () {
+            col.drop();
+        },
+
+        testIndexAccess: function () {
+            const query = aql`
+        FOR d IN ${col}
+          FILTER 0 <= d.x && d.x <= 1
+          RETURN d.x
+      `;
+            const explainRes = AQL_EXPLAIN(query.query, query.bindVars);
+            const appliedRules = explainRes.plan.rules;
+            const nodeTypes = explainRes.plan.nodes.map(n => n.type).filter(n => !["GatherNode", "RemoteNode"].includes(n));
+            assertEqual(["SingletonNode", "IndexNode", "CalculationNode", "ReturnNode"], nodeTypes);
+            assertTrue(appliedRules.includes(useIndexes));
+            assertTrue(appliedRules.includes(removeFilterCoveredByIndex));
+            const executeRes = AQL_EXECUTE(
8BEE
query.query, query.bindVars);
+            const res = executeRes.json;
+            res.sort();
+            assertEqual([0, 0.1, 0.2, 0.3, 0.4, 0.5, 0.6, 0.7, 0.8, 0.9, 1], res);
+        },
+
+        testIndexAccess2: function () {
+            const query = aql`
+        FOR d IN ${col}
+          FILTER 0 <= d.x && d.y <= 1
+          RETURN d.x
+      `;
+            const explainRes = AQL_EXPLAIN(query.query, query.bindVars);
+            const appliedRules = explainRes.plan.rules;
+            const nodeTypes = explainRes.plan.nodes.map(n => n.type).filter(n => !["GatherNode", "RemoteNode"].includes(n));
+            assertEqual(["SingletonNode", "IndexNode", "CalculationNode", "ReturnNode"], nodeTypes);
+            assertTrue(appliedRules.includes(useIndexes));
+            assertTrue(appliedRules.includes(removeFilterCoveredByIndex));
+            const executeRes = AQL_EXECUTE(query.query, query.bindVars);
+            const res = executeRes.json;
+            res.sort();
+            assertEqual([0, 0.1, 0.2, 0.3, 0.4, 0.5], res);
+        },
+
+        testUniqueConstraint: function () {
+            col.save({x: 0, y: 0.50001});
+            try {
+                col.save({x: 0, y: 0.5});
+                fail();
+            } catch (e) {
+                assertEqual(e.errorNum, internal.errors.ERROR_ARANGO_UNIQUE_CONSTRAINT_VIOLATED.code);
+            }
+        }
+    };
+}
+
+jsunity.run(optimizerRuleZkd2dIndexTestSuite);
+
+return jsunity.done();