@sterblue/sterblue-sdk/lib/utils/__tests__/stream-query.js

Sterblue Graph SDK for graphile.sterblue.com

"use strict";

var _fp = require("lodash/fp");

var _streamingIterables = require("streaming-iterables");

var _streamQuery = require("../stream-query");

function ownKeys(object, enumerableOnly) { var keys = Object.keys(object); if (Object.getOwnPropertySymbols) { var symbols = Object.getOwnPropertySymbols(object); if (enumerableOnly) symbols = symbols.filter(function (sym) { return Object.getOwnPropertyDescriptor(object, sym).enumerable; }); keys.push.apply(keys, symbols); } return keys; }

function _objectSpread(target) { for (var i = 1; i < arguments.length; i++) { var source = arguments[i] != null ? arguments[i] : {}; if (i % 2) { ownKeys(Object(source), true).forEach(function (key) { _defineProperty(target, key, source[key]); }); } else if (Object.getOwnPropertyDescriptors) { Object.defineProperties(target, Object.getOwnPropertyDescriptors(source)); } else { ownKeys(Object(source)).forEach(function (key) { Object.defineProperty(target, key, Object.getOwnPropertyDescriptor(source, key)); }); } } return target; }

function _defineProperty(obj, key, value) { if (key in obj) { Object.defineProperty(obj, key, { value: value, enumerable: true, configurable: true, writable: true }); } else { obj[key] = value; } return obj; }

function _objectWithoutProperties(source, excluded) { if (source == null) return {}; var target = _objectWithoutPropertiesLoose(source, excluded); var key, i; if (Object.getOwnPropertySymbols) { var sourceSymbolKeys = Object.getOwnPropertySymbols(source); for (i = 0; i < sourceSymbolKeys.length; i++) { key = sourceSymbolKeys[i]; if (excluded.indexOf(key) >= 0) continue; if (!Object.prototype.propertyIsEnumerable.call(source, key)) continue; target[key] = source[key]; } } return target; }

function _objectWithoutPropertiesLoose(source, excluded) { if (source == null) return {}; var target = {}; var sourceKeys = Object.keys(source); var key, i; for (i = 0; i < sourceKeys.length; i++) { key = sourceKeys[i]; if (excluded.indexOf(key) >= 0) continue; target[key] = source[key]; } return target; }

test("streamQuery", async () => {
  const mockListQuery = async (_ref) => {
    let {
      first,
      offset
    } = _ref,
        rest = _objectWithoutProperties(_ref, ["first", "offset"]);

    return {
      list: (0, _fp.map)(index => _objectSpread({
        index,
        first,
        offset
      }, rest), (0, _fp.range)(offset, offset + first))
    };
  };

  const mockCountQuery = async (_ref2) => {
    let {
      first,
      offset
    } = _ref2,
        rest = _objectWithoutProperties(_ref2, ["first", "offset"]);

    return {
      connection: {
        totalCount: 10
      }
    };
  };

  expect(await (0, _streamingIterables.collect)(await (0, _streamQuery.streamQuery)(mockListQuery)({
    first: 17,
    offset: 11,
    batchSize: 3,
    batchConcurrency: 4
  }))).toEqual([{
    first: 3,
    index: 11,
    offset: 11
  }, {
    first: 3,
    index: 12,
    offset: 11
  }, {
    first: 3,
    index: 13,
    offset: 11
  }, {
    first: 3,
    index: 14,
    offset: 14
  }, {
    first: 3,
    index: 15,
    offset: 14
  }, {
    first: 3,
    index: 16,
    offset: 14
  }, {
    first: 3,
    index: 17,
    offset: 17
  }, {
    first: 3,
    index: 18,
    offset: 17
  }, {
    first: 3,
    index: 19,
    offset: 17
  }, {
    first: 3,
    index: 20,
    offset: 20
  }, {
    first: 3,
    index: 21,
    offset: 20
  }, {
    first: 3,
    index: 22,
    offset: 20
  }, {
    first: 3,
    index: 23,
    offset: 23
  }, {
    first: 3,
    index: 24,
    offset: 23
  }, {
    first: 3,
    index: 25,
    offset: 23
  }, {
    first: 2,
    index: 26,
    offset: 26
  }, {
    first: 2,
    index: 27,
    offset: 26
  }]);
  expect(await (0, _streamingIterables.collect)(await (0, _streamQuery.streamQuery)(mockListQuery)({
    first: 3,
    offset: 4,
    batchSize: 17,
    batchConcurrency: 11
  }))).toEqual([{
    first: 3,
    index: 4,
    offset: 4
  }, {
    first: 3,
    index: 5,
    offset: 4
  }, {
    first: 3,
    index: 6,
    offset: 4
  }]);
  expect(await (0, _streamingIterables.collect)(await (0, _streamQuery.streamQuery)(mockListQuery, mockCountQuery)({
    first: 17,
    offset: 7,
    batchSize: 11,
    batchConcurrency: 4
  }))).toEqual([{
    first: 3,
    index: 7,
    offset: 7
  }, {
    first: 3,
    index: 8,
    offset: 7
  }, {
    first: 3,
    index: 9,
    offset: 7
  }]);
});