qa-588: add vector index test

test_data/checkdata.js

@@ -50,6 +50,12 @@ let {
   mainTestLoop
 } = require(fs.join(PWD, 'common'));
 
+const {
+  assertTrue,
+  assertFalse,
+  assertEqual
+} = require("jsunity").jsUnity.assertions;
+
 const {
   createAnalyzerSet,
   checkAnalyzerSet,

test_data/cleardata.js

@@ -48,6 +48,12 @@ let {
   mainTestLoop
 } = require(fs.join(PWD, 'common'));
 
+const {
+  assertTrue,
+  assertFalse,
+  assertEqual
+} = require("jsunity").jsUnity.assertions;
+
 const {
   createAnalyzerSet,
   checkAnalyzerSet,

test_data/common.js

@@ -366,6 +366,13 @@ function makeRandomDoc() {
   }
   return {
     Type: makeRandomNumber(1000, 65535),
+    TypeVec: [
+      makeRandomNumber(1000, 65535),
+      makeRandomNumber(1000, 65535),
+      makeRandomNumber(1000, 65535),
+      makeRandomNumber(1000, 65535),
+      makeRandomNumber(1000, 65535),
+    ],
     ID: makeRandomString(40),
     OptOut: rand() > 0 ? 1 : 0,
     Source: makeRandomString(14),

test_data/makedata.js

@@ -63,6 +63,12 @@ let {
   resetRCount,
 } = require(fs.join(PWD, 'common'));
 
+const {
+  assertTrue,
+  assertFalse,
+  assertEqual
+} = require("jsunity").jsUnity.assertions;
+
 const {
   createAnalyzerSet,
   checkAnalyzerSet,

test_data/makedata_suites/070_foxx.js

@@ -1,17 +1,11 @@
-/* global print, fs, db, internal, arango, assertTrue */
+/* global print, fs, db, internal, arango, assertTrue, assertEqual, assertFalse */
 
 // inspired by shell-foxx-api-spec.js
 
 const utils = require('@arangodb/foxx/manager-utils');
 const download = internal.download;
 const path = require('path');
 
-const {
-  assertTrue,
-  assertFalse,
-  assertEqual
-} = require("jsunity").jsUnity.assertions;
-
 function loadFoxxIntoZip (path) {
   let zip = utils.zipDirectory(path);
   let content = fs.readFileSync(zip);

test_data/makedata_suites/107_vector.js

@@ -0,0 +1,89 @@
+/* global print,  db, progress, createCollectionSafe, createIndexSafe, time, runAqlQueryResultCount, aql, semver, resetRCount, writeData */
+
+(function () {
+  return {
+    isSupported: function (currentVersion, oldVersion, options, enterprise, cluster) {
+      let currentVersionSemver = semver.parse(semver.coerce(currentVersion));
+      let oldVersionSemver = semver.parse(semver.coerce(oldVersion));
+      return semver.gt(oldVersionSemver, "3.12.4");
+    },
+    makeDataDB: function (options, isCluster, isEnterprise, database, dbCount) {
+      progress('107: createCollection');
+      let c_vector = createCollectionSafe(`c_vector_${dbCount}`, 3, 2);
+      progress('107: createIndexHash');
+    },
+    makeData: function (options, isCluster, isEnterprise, dbCount, loopCount) {
+      progress(`107: Makedata ${dbCount} ${loopCount}`);
+      let c_vector = db[`c_vector_${dbCount}`];
+
+      // Now the actual data writing:
+      resetRCount();
+      writeData(c_vector, 1000);
+      if (c_vector.indexes().length === 1) {
+        createIndexSafe({
+          col: c_vector,
+          name: `i_vector_dbcount`,
+          type: "vector",
+          fields: ["TypeVec"],
+          inBackground: false,
+          params: {
+            metric: "l2",
+            dimension: 5,
+            nLists: 1
+          },
+        });
+      }
+      progress('107: writeData1');
+    },
+    checkDataDB: function (options, isCluster, isEnterprise, database, dbCount, readOnly) {
+      print(`${Date()} 107: checking data ${dbCount}`);
+      let cols = db._collections();
+      let allFound = true;
+      [`c_vector_${dbCount}`].forEach(colname => {
+         let foundOne = false;
+         cols.forEach(oneCol => {
+           if (oneCol.name() === colname) {
+             foundOne = true;
+           }
+         });
+         if (!foundOne) {
+           print(`${Date()} 107: Didn't find this collection: ${colname}`);
+           allFound = false;
+         }
+       });
+      if (!allFound) {
+        throw new Error("107: not all collections were present on the system!");
+      }
+
+      let c_vector = db._collection(`c_vector_${dbCount}`);
+
+      // Check indexes:
+      progress("107: checking indices");
+
+      if (c_vector.getIndexes().length !== 2 || c_vector.getIndexes()[1].type !== "vector") {
+        throw new Error(`Banana ${c_vector.getIndexes().length} `);
+      }
+
+      // Check data:
+      progress("107: checking data");
+      if (c_vector.count() !== 1000 * options.dataMultiplier) { throw new Error(`Audi ${c_vector.count()} !== 1000`); }
+
+      // Check a few queries:
+      progress("107: query 1");
+      runAqlQueryResultCount(aql`
+           FOR d IN ${c_vector}
+               SORT APPROX_NEAR_L2(d.TypeVec,  [1,2,3,4,5], {nProbe: 5})
+                 LIMIT 5 RETURN d`, 5);
+      progress("107: queries done");
+      progress("107: done");
+    },
+    clearData: function (options, isCluster, isEnterprise, dbCount, loopCount, readOnly) {
+      print(`${Date()} 107: clearing data ${dbCount} ${loopCount}`);
+      progress("107: drop 1");
+      try {
+        db._drop(`c_vector_${loopCount}`);
+      } catch (ex) {}
+      progress("107: drop done");
+    }
+  };
+}());