/*jshint globalstrict:false, strict:false, maxlen:1000*/
/*global assertEqual, assertTrue, assertUndefined, fail, more */

////////////////////////////////////////////////////////////////////////////////
/// @brief test the statement class
///
/// @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
////////////////////////////////////////////////////////////////////////////////

const jsunity = require("jsunity");
const db = require("internal").db;
const aqlfunctions = require("@arangodb/aql/functions");

////////////////////////////////////////////////////////////////////////////////
/// @brief test suite: stream cursors
////////////////////////////////////////////////////////////////////////////////

function StreamCursorSuite () {
  'use strict';

  const cn = "StreamCursorCollection";
  let c = null;
  const queries = [
    `FOR doc IN ${cn} RETURN doc`,
    `FOR doc IN ${cn} FILTER doc.value1 == 1 UPDATE doc WITH { value1: 1 } INTO ${cn}`,
    `FOR doc IN ${cn} FILTER doc.value1 == 1 REPLACE doc WITH { value1: 2 } INTO ${cn}`,
    `FOR doc IN ${cn} FILTER doc.value1 == 2 INSERT { value2: doc.value1 } INTO ${cn}`,
    `FOR doc IN ${cn} FILTER doc.value1 == 2 INSERT { value2: doc.value1 } INTO ${cn}`,
    `FOR doc IN ${cn} FILTER doc.value3 >= 4 RETURN doc`,
    `FOR doc IN ${cn} COLLECT a = doc.value1 INTO groups = doc RETURN { val: a, doc: groups }`
  ];

  return {

////////////////////////////////////////////////////////////////////////////////
/// @brief set up
////////////////////////////////////////////////////////////////////////////////

    setUp : function () {
      c = db._create(cn);
      c.ensureIndex({ type: 'skiplist', fields: ["value1"]});
      c.ensureIndex({ type: 'skiplist', fields: ["value2"]});

      for (let i = 0; i < 5000; i++) {
        c.insert({value1: i % 10, value2: i % 25 , value3: i % 25 });
      }
 
      try {
        aqlfunctions.unregister("my::test");
      } catch (err) {}

      aqlfunctions.register("my::test", function() { return 42; });
    },

////////////////////////////////////////////////////////////////////////////////
/// @brief tear down
////////////////////////////////////////////////////////////////////////////////

    tearDown : function () {
      try {
        aqlfunctions.unregister("my::test");
      } catch (err) {}

      c.drop();
    },

////////////////////////////////////////////////////////////////////////////////
/// @brief test cursor
////////////////////////////////////////////////////////////////////////////////

    testQueries : function () {
      queries.forEach(q => {
        var stmt = db._createStatement({ query: q,
          options: { stream: true },
          batchSize: 1000});
        var cursor = stmt.execute();

        assertEqual(undefined, cursor.count());
        while (cursor.hasNext()) {
          cursor.next();
        }
      });      
    },

    testInfiniteAQL : function() {
      var stmt = db._createStatement({ query: "FOR i IN 1..100000000000 RETURN i",
        options: { stream: true },
        batchSize: 1000});
      var cursor = stmt.execute();

      assertUndefined(cursor.count());
      let i = 10;
      while (cursor.hasNext() && i-- > 0) {
        cursor.next();
      }
    },

    testUserDefinedFunction : function () {
      let stmt = db._createStatement({ 
        query: "FOR i IN 1..10000 RETURN my::test()",
        options: { stream: true },
        batchSize: 1000
      });
      let cursor = stmt.execute();

      assertEqual(undefined, cursor.count());
      let count = 0;
      while (cursor.hasNext()) {
        assertEqual(42, cursor.next());
        ++count;
      }
      assertEqual(10000, count);
    },
  };
}

jsunity.run(StreamCursorSuite);
return jsunity.done();